Searched refs:output_grad_data (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 57 se::DeviceMemoryBase output_grad_data; member
193 auto output_grad_data = se::DeviceMemory<ElemType>(params->output_grad_data); in RunCudnnBatchNormBackwardImpl() local
203 &output_grad_data, // in RunCudnnBatchNormBackwardImpl()
286 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output, in RunCudnnBatchNormBackward() argument
293 backward_params.output_grad_data = output_grad_data; in RunCudnnBatchNormBackward()
|
| D | cudnn_batchnorm_thunk.cc | 123 const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk() argument
133 output_grad_data_(output_grad_data), in CudnnBatchNormBackwardThunk()
141 se::DeviceMemoryBase output_grad_data = in ExecuteOnStream() local
154 config_, operand, output_grad_data, grad_output, output_grad_scale, in ExecuteOnStream()
|
| D | cudnn_batchnorm_runner.h | 56 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output,
|
| D | cudnn_batchnorm_thunk.h | 112 const BufferAllocation::Slice& output_grad_data,
|
| D | ir_emitter_unnested.cc | 1416 TF_ASSIGN_OR_RETURN(auto output_grad_data, in EmitBatchNormThunkFromMlir()
1437 /*output_grad_data=*/output_grad_data, in EmitBatchNormThunkFromMlir()
|