Searched refs:CudnnBatchNormForwardInferenceThunk (Results 1 – 3 of 3) sorted by relevance
48 class CudnnBatchNormForwardInferenceThunk : public Thunk {50 CudnnBatchNormForwardInferenceThunk(ThunkInfo thunk_info,59 CudnnBatchNormForwardInferenceThunk(60 const CudnnBatchNormForwardInferenceThunk&) = delete;61 CudnnBatchNormForwardInferenceThunk& operator=(62 const CudnnBatchNormForwardInferenceThunk&) = delete;
34 CudnnBatchNormForwardInferenceThunk::CudnnBatchNormForwardInferenceThunk( in CudnnBatchNormForwardInferenceThunk() function in xla::gpu::CudnnBatchNormForwardInferenceThunk50 Status CudnnBatchNormForwardInferenceThunk::ExecuteOnStream( in ExecuteOnStream()
1469 CudnnBatchNormForwardInferenceThunk>( in EmitBatchNormThunkFromMlir()