Searched refs:output_grad_offset (Results 1 – 5 of 5) sorted by relevance
125 const BufferAllocation::Slice& output_grad_offset) in CudnnBatchNormBackwardThunk() argument135 output_grad_offset_(output_grad_offset) {} in CudnnBatchNormBackwardThunk()147 se::DeviceMemory<float> output_grad_offset( in ExecuteOnStream() local155 output_grad_offset, in ExecuteOnStream()
60 se::DeviceMemory<float> output_grad_offset; member205 ¶ms->output_grad_offset, // in RunCudnnBatchNormBackwardImpl()288 se::DeviceMemory<float> output_grad_offset, se::DeviceMemory<float> scale, in RunCudnnBatchNormBackward() argument296 backward_params.output_grad_offset = output_grad_offset; in RunCudnnBatchNormBackward()
58 se::DeviceMemory<float> output_grad_offset, se::DeviceMemory<float> scale,
114 const BufferAllocation::Slice& output_grad_offset);
1420 TF_ASSIGN_OR_RETURN(auto output_grad_offset, in EmitBatchNormThunkFromMlir()1439 /*output_grad_offset=*/output_grad_offset)); in EmitBatchNormThunkFromMlir()