Searched refs:output_grad_offset (Results 1 – 3 of 3) sorted by relevance
221 const BufferAllocation::Slice& output_grad_offset, in CudnnBatchNormBackwardThunk() argument233 output_grad_offset_(output_grad_offset), in CudnnBatchNormBackwardThunk()263 se::DeviceMemory<float> output_grad_offset( in ExecuteOnStream() local275 &output_grad_scale, &output_grad_offset); in ExecuteOnStream()279 output_grad_offset.opaque()}; in ExecuteOnStream()
120 const BufferAllocation::Slice& output_grad_offset,
449 auto output_grad_offset = in HandleCustomCall() local461 /*output_grad_offset=*/output_grad_offset, in HandleCustomCall()