Searched refs:output_grad_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 219 int64 feature_index, const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk() argument
231 output_grad_data_(output_grad_data), in CudnnBatchNormBackwardThunk()
259 se::DeviceMemory<float> output_grad_data( in ExecuteOnStream() local
274 operand_desc, scale_offset_desc, epsilon_, &output_grad_data, in ExecuteOnStream()
278 void* ptrs[] = {output_grad_data.opaque(), output_grad_scale.opaque(), in ExecuteOnStream()
|
| D | cudnn_batchnorm_thunk.h | 118 const BufferAllocation::Slice& output_grad_data,
|
| D | ir_emitter_unnested.cc | 447 auto output_grad_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie(); in HandleCustomCall() local
459 /*output_grad_data=*/output_grad_data, in HandleCustomCall()
|