Home
last modified time | relevance | path

Searched defs:output_grad_data (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc118 const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk()
136 se::DeviceMemoryBase output_grad_data = in ExecuteOnStream() local
Dcudnn_batchnorm_runner.cc57 se::DeviceMemoryBase output_grad_data; member
194 auto output_grad_data = se::DeviceMemory<ElemType>(params->output_grad_data); in RunCudnnBatchNormBackwardImpl() local
287 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output, in RunCudnnBatchNormBackward()