Searched refs:output_grad_scale_ (Results 1 – 2 of 2) sorted by relevance
134 output_grad_scale_(output_grad_scale), in CudnnBatchNormBackwardThunk()146 buffer_allocations.GetDeviceAddress(output_grad_scale_)); in ExecuteOnStream()
130 BufferAllocation::Slice output_grad_scale_; variable