Searched refs:outfeed_manager (Results 1 – 4 of 4) sorted by relevance
37 OutfeedManager* outfeed_manager = GetOrCreateOutfeedManager(); in ExecuteOnStream() local39 outfeed_manager->BlockingGetNextDestination(); in ExecuteOnStream()
170 gpu::OutfeedManager* outfeed_manager = gpu::GetOrCreateOutfeedManager(); in TransferLiteralFromOutfeed() local171 outfeed_manager->EnqueueDestination(&outfeed_buffers); in TransferLiteralFromOutfeed()
376 ":outfeed_manager",784 ":outfeed_manager",923 name = "outfeed_manager",924 srcs = ["outfeed_manager.cc"],925 hdrs = ["outfeed_manager.h"],
71 "//tensorflow/compiler/xla/service/gpu:outfeed_manager",