Searched refs:output_grad_data_ (Results 1 – 2 of 2) sorted by relevance
231 output_grad_data_(output_grad_data), in CudnnBatchNormBackwardThunk()260 buffer_allocations.GetDeviceAddress(output_grad_data_)); in ExecuteOnStream()
140 BufferAllocation::Slice output_grad_data_; variable