Searched defs:output_grad_scale (Results 1 – 2 of 2) sorted by relevance
124 const BufferAllocation::Slice& output_grad_scale, in CudnnBatchNormBackwardThunk()145 se::DeviceMemory<float> output_grad_scale( in ExecuteOnStream() local
59 se::DeviceMemory<float> output_grad_scale; member287 se::DeviceMemory<float> output_grad_scale, in RunCudnnBatchNormBackward()