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