Searched refs:CudnnBatchNormBackwardThunk (Results 1 – 3 of 3) sorted by relevance
110 class CudnnBatchNormBackwardThunk : public Thunk {112 CudnnBatchNormBackwardThunk(const BufferAllocation::Slice& operand,124 CudnnBatchNormBackwardThunk(const CudnnBatchNormBackwardThunk&) = delete;125 CudnnBatchNormBackwardThunk& operator=(const CudnnBatchNormBackwardThunk&) =
214 CudnnBatchNormBackwardThunk::CudnnBatchNormBackwardThunk( in CudnnBatchNormBackwardThunk() function in xla::gpu::CudnnBatchNormBackwardThunk247 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream()
451 AddThunkToThunkSequence(absl::make_unique<CudnnBatchNormBackwardThunk>( in HandleCustomCall()