Searched refs:outfeed_buffers (Results 1 – 2 of 2) sorted by relevance
150 ShapeTree<std::unique_ptr<gpu::OutfeedBuffer>> outfeed_buffers( in TransferLiteralFromOutfeed() local154 outfeed_buffers.ForEachMutableElement( in TransferLiteralFromOutfeed()171 outfeed_manager->EnqueueDestination(&outfeed_buffers); in TransferLiteralFromOutfeed()174 ShapeTreeToLiteral(&outfeed_buffers); in TransferLiteralFromOutfeed()
38 ShapeTree<std::unique_ptr<OutfeedBuffer>>* outfeed_buffers = in ExecuteOnStream() local46 outfeed_buffers->shape())); in ExecuteOnStream()48 TF_RETURN_IF_ERROR(outfeed_buffers->ForEachMutableElementWithStatus( in ExecuteOnStream()