Searched refs:ConvolutionThunk (Results 1 – 3 of 3) sorted by relevance
40 class ConvolutionThunk : public Thunk {45 ConvolutionThunk(ThunkInfo thunk_info, GpuConvConfig config,50 ConvolutionThunk(const ConvolutionThunk&) = delete;51 ConvolutionThunk& operator=(const ConvolutionThunk&) = delete;
33 ConvolutionThunk::ConvolutionThunk( in ConvolutionThunk() function in xla::gpu::ConvolutionThunk43 Status ConvolutionThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
1217 AddThunkToThunkSequence(absl::make_unique<ConvolutionThunk>( in EmitConvolutionThunkFromMlir()