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