Searched refs:OutfeedThunk (Results 1 – 3 of 3) sorted by relevance
31 class OutfeedThunk : public Thunk {35 OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices,38 OutfeedThunk(const OutfeedThunk&) = delete;39 OutfeedThunk& operator=(const OutfeedThunk&) = delete;
26 OutfeedThunk::OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices, in OutfeedThunk() function in xla::gpu::OutfeedThunk31 Status OutfeedThunk::ExecuteOnStream( in ExecuteOnStream()
1756 return absl::make_unique<OutfeedThunk>(std::move(slices), inst); in BuildOutfeedThunk()