Searched defs:output_grad_offset (Results 1 – 2 of 2) sorted by relevance
221 const BufferAllocation::Slice& output_grad_offset, in CudnnBatchNormBackwardThunk()263 se::DeviceMemory<float> output_grad_offset( in ExecuteOnStream() local
449 auto output_grad_offset = in HandleCustomCall() local