Searched refs:OutfeedBuffer (Results 1 – 3 of 3) sorted by relevance
34 class OutfeedBuffer {36 OutfeedBuffer(int64 length) : length_(length) {} in OutfeedBuffer() function58 using OutfeedManager = XfeedQueue<ShapeTree<std::unique_ptr<OutfeedBuffer>>*>;
42 ShapeTree<std::unique_ptr<OutfeedBuffer>>* output_buffers = in ExecuteOnStream()61 std::unique_ptr<OutfeedBuffer>& buffer = output_leaf_it->second; in ExecuteOnStream()
118 ShapeTree<std::unique_ptr<gpu::OutfeedBuffer>> outfeed_buffers( in TransferLiteralFromOutfeed()125 absl::make_unique<gpu::OutfeedBuffer>(GetByteSizeRequirement(shape)); in TransferLiteralFromOutfeed()