Searched refs:output_grad_offset_ (Results 1 – 2 of 2) sorted by relevance
135 output_grad_offset_(output_grad_offset) {} in CudnnBatchNormBackwardThunk()148 buffer_allocations.GetDeviceAddress(output_grad_offset_)); in ExecuteOnStream()
131 BufferAllocation::Slice output_grad_offset_; variable