Searched refs:CudnnBatchNormForwardTrainingThunk (Results 1 – 3 of 3) sorted by relevance
76 class CudnnBatchNormForwardTrainingThunk : public Thunk {78 CudnnBatchNormForwardTrainingThunk(87 CudnnBatchNormForwardTrainingThunk(88 const CudnnBatchNormForwardTrainingThunk&) = delete;89 CudnnBatchNormForwardTrainingThunk& operator=(90 const CudnnBatchNormForwardTrainingThunk&) = delete;
73 CudnnBatchNormForwardTrainingThunk::CudnnBatchNormForwardTrainingThunk( in CudnnBatchNormForwardTrainingThunk() function in xla::gpu::CudnnBatchNormForwardTrainingThunk89 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream()
1379 absl::make_unique<CudnnBatchNormForwardTrainingThunk>( in EmitBatchNormThunkFromMlir()