HomeSort by relevance Sort by last modified time
    Searched refs:output_grad_scale (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc 220 const BufferAllocation::Slice& output_grad_scale,
232 output_grad_scale_(output_grad_scale),
261 se::DeviceMemory<float> output_grad_scale(
275 &output_grad_scale, &output_grad_offset); local
278 void* ptrs[] = {output_grad_data.opaque(), output_grad_scale.opaque(),
cudnn_batchnorm_thunk.h 119 const BufferAllocation::Slice& output_grad_scale,
ir_emitter_unnested.cc 448 auto output_grad_scale = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); local
460 /*output_grad_scale=*/output_grad_scale,
    [all...]

Completed in 1380 milliseconds