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