Searched refs:CudnnBatchNormForwardTrainingThunk (Results 1 – 3 of 3) sorted by relevance
78 class CudnnBatchNormForwardTrainingThunk : public Thunk {80 CudnnBatchNormForwardTrainingThunk(89 CudnnBatchNormForwardTrainingThunk(90 const CudnnBatchNormForwardTrainingThunk&) = delete;91 CudnnBatchNormForwardTrainingThunk& operator=(92 const CudnnBatchNormForwardTrainingThunk&) = delete;
135 CudnnBatchNormForwardTrainingThunk::CudnnBatchNormForwardTrainingThunk( in CudnnBatchNormForwardTrainingThunk() function in xla::gpu::CudnnBatchNormForwardTrainingThunk163 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream()
421 absl::make_unique<CudnnBatchNormForwardTrainingThunk>( in HandleCustomCall()