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