Searched refs:output_grad_offset_ (Results 1 – 2 of 2) sorted by relevance
233 output_grad_offset_(output_grad_offset), in CudnnBatchNormBackwardThunk()264 buffer_allocations.GetDeviceAddress(output_grad_offset_)); in ExecuteOnStream()
142 BufferAllocation::Slice output_grad_offset_; variable