Searched defs:output_grad_data (Results 1 – 2 of 2) sorted by relevance
118 const BufferAllocation::Slice& output_grad_data, in CudnnBatchNormBackwardThunk()136 se::DeviceMemoryBase output_grad_data = in ExecuteOnStream() local
57 se::DeviceMemoryBase output_grad_data; member194 auto output_grad_data = se::DeviceMemory<ElemType>(params->output_grad_data); in RunCudnnBatchNormBackwardImpl() local287 se::DeviceMemoryBase output_grad_data, se::DeviceMemoryBase grad_output, in RunCudnnBatchNormBackward()