Searched refs:GemmThunk (Results 1 – 4 of 4) sorted by relevance
47 class GemmThunk : public Thunk {51 GemmThunk(ThunkInfo thunk_info, GpuGemmConfig config,57 GemmThunk(const GemmThunk&) = delete;58 GemmThunk& operator=(const GemmThunk&) = delete;
48 thunks.push_back(absl::make_unique<GemmThunk>( in BuildGemmThunk()59 return absl::make_unique<GemmThunk>( in BuildGemmThunk()
46 GemmThunk::GemmThunk(ThunkInfo thunk_info, GpuGemmConfig config, in GemmThunk() function in xla::gpu::GemmThunk58 Status GemmThunk::ExecuteOnStream(const ExecuteParams ¶ms) { in ExecuteOnStream()
1284 thunks.push_back(absl::make_unique<GemmThunk>( in EmitGemmThunkFromMlir()1293 AddThunkToThunkSequence(absl::make_unique<GemmThunk>( in EmitGemmThunkFromMlir()