Searched refs:outfeed_manager (Results 1 – 4 of 4) sorted by relevance
41 OutfeedManager* outfeed_manager = GetOrCreateOutfeedManager(); in ExecuteOnStream() local43 outfeed_manager->BlockingGetNextDestination(); in ExecuteOnStream()
132 gpu::OutfeedManager* outfeed_manager = gpu::GetOrCreateOutfeedManager(); in TransferLiteralFromOutfeed() local133 outfeed_manager->EnqueueDestination(&outfeed_buffers); in TransferLiteralFromOutfeed()
614 ":outfeed_manager",1209 ":outfeed_manager",1488 name = "outfeed_manager",1489 srcs = ["outfeed_manager.cc"],1490 hdrs = ["outfeed_manager.h"],
79 "//tensorflow/compiler/xla/service/gpu:outfeed_manager",