Searched refs:GemmThunk (Results 1 – 3 of 3) sorted by relevance
36 class GemmThunk : public Thunk {40 GemmThunk(const BufferAllocation::Slice& lhs_buffer,48 GemmThunk(const GemmThunk&) = delete;49 GemmThunk& operator=(const GemmThunk&) = delete;
293 GemmThunk::GemmThunk(const BufferAllocation::Slice& lhs_buffer, in GemmThunk() function in xla::gpu::GemmThunk311 Status GemmThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
1779 return absl::make_unique<GemmThunk>( in BuildGemmThunk()1846 thunks.push_back(absl::make_unique<GemmThunk>( in BuildGemmThunk()1858 return absl::make_unique<GemmThunk>( in BuildGemmThunk()