Searched defs:output_grad_offset (Results 1 – 2 of 2) sorted by relevance
125 const BufferAllocation::Slice& output_grad_offset) in CudnnBatchNormBackwardThunk()147 se::DeviceMemory<float> output_grad_offset( in ExecuteOnStream() local
60 se::DeviceMemory<float> output_grad_offset; member288 se::DeviceMemory<float> output_grad_offset, se::DeviceMemory<float> scale, in RunCudnnBatchNormBackward()