Searched refs:ConvolutionThunk (Results 1 – 3 of 3) sorted by relevance
40 class ConvolutionThunk : public Thunk {46 ConvolutionThunk(const HloCustomCallInstruction* cudnn_call,52 ConvolutionThunk(const ConvolutionThunk&) = delete;53 ConvolutionThunk& operator=(const ConvolutionThunk&) = delete;
32 ConvolutionThunk::ConvolutionThunk( in ConvolutionThunk() function in xla::gpu::ConvolutionThunk44 Status ConvolutionThunk::ExecuteOnStream( in ExecuteOnStream()
478 AddThunkToThunkSequence(absl::make_unique<ConvolutionThunk>( in HandleCustomCall()