Home
last modified time | relevance | path

Searched refs:GetOrCreateOutfeedManager (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_manager.cc26 OutfeedManager* GetOrCreateOutfeedManager() { in GetOrCreateOutfeedManager() function
Doutfeed_manager.h61 OutfeedManager* GetOrCreateOutfeedManager();
Doutfeed_thunk.cc37 OutfeedManager* outfeed_manager = GetOrCreateOutfeedManager(); in ExecuteOnStream()
Dgpu_transfer_manager.cc170 gpu::OutfeedManager* outfeed_manager = gpu::GetOrCreateOutfeedManager(); in TransferLiteralFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc259 xla::gpu::GetOrCreateOutfeedManager() in ReplayComputation()