Home
last modified time | relevance | path

Searched defs:output_grad_offset (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc221 const BufferAllocation::Slice& output_grad_offset, in CudnnBatchNormBackwardThunk()
263 se::DeviceMemory<float> output_grad_offset( in ExecuteOnStream() local
Dir_emitter_unnested.cc449 auto output_grad_offset = in HandleCustomCall() local