Searched refs:CudnnBatchNormBackwardThunk (Results 1 – 3 of 3) sorted by relevance
104 class CudnnBatchNormBackwardThunk : public Thunk {106 CudnnBatchNormBackwardThunk(116 CudnnBatchNormBackwardThunk(const CudnnBatchNormBackwardThunk&) = delete;117 CudnnBatchNormBackwardThunk& operator=(const CudnnBatchNormBackwardThunk&) =
117 CudnnBatchNormBackwardThunk::CudnnBatchNormBackwardThunk( in CudnnBatchNormBackwardThunk() function in xla::gpu::CudnnBatchNormBackwardThunk137 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream()
1429 AddThunkToThunkSequence(absl::make_unique<CudnnBatchNormBackwardThunk>( in EmitBatchNormThunkFromMlir()