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>>*>;
121 ShapeTree<std::unique_ptr<gpu::OutfeedBuffer>>* shape_tree) { in ShapeTreeToLiteral()125 ShapeTree<std::unique_ptr<gpu::OutfeedBuffer>>* shape_tree, in ShapeTreeToLiteral()150 ShapeTree<std::unique_ptr<gpu::OutfeedBuffer>> outfeed_buffers( in TransferLiteralFromOutfeed()156 std::unique_ptr<gpu::OutfeedBuffer>* buffer) { in TransferLiteralFromOutfeed()162 *buffer = absl::make_unique<gpu::OutfeedBuffer>( in TransferLiteralFromOutfeed()
38 ShapeTree<std::unique_ptr<OutfeedBuffer>>* outfeed_buffers = in ExecuteOnStream()49 [&](const ShapeIndex& index, std::unique_ptr<OutfeedBuffer>* buffer) { in ExecuteOnStream()