Searched refs:output_grad_scale (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 124 const BufferAllocation::Slice& output_grad_scale, in CudnnBatchNormBackwardThunk() argument
134 output_grad_scale_(output_grad_scale), in CudnnBatchNormBackwardThunk()
145 se::DeviceMemory<float> output_grad_scale( in ExecuteOnStream() local
154 config_, operand, output_grad_data, grad_output, output_grad_scale, in ExecuteOnStream()
|
| D | cudnn_batchnorm_runner.cc | 59 se::DeviceMemory<float> output_grad_scale; member
204 ¶ms->output_grad_scale, // in RunCudnnBatchNormBackwardImpl()
287 se::DeviceMemory<float> output_grad_scale, in RunCudnnBatchNormBackward() argument
295 backward_params.output_grad_scale = output_grad_scale; in RunCudnnBatchNormBackward()
|
| D | cudnn_batchnorm_runner.h | 57 se::DeviceMemory<float> output_grad_scale,
|
| D | cudnn_batchnorm_thunk.h | 113 const BufferAllocation::Slice& output_grad_scale,
|
| D | ir_emitter_unnested.cc | 1418 TF_ASSIGN_OR_RETURN(auto output_grad_scale, in EmitBatchNormThunkFromMlir()
1438 /*output_grad_scale=*/output_grad_scale, in EmitBatchNormThunkFromMlir()
|