Home
last modified time | relevance | path

Searched refs:gpu_ring_order_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.cc34 gpu_ring_order_( in CollectiveExecutorMgr()
66 &gpu_ring_order_, work_queue_); in Create()
Dbase_collective_executor.h106 gpu_ring_order_(gpu_ring_order), in BaseCollectiveExecutor()
142 const string* gpu_ring_order_; // Not owned. variable
Dpermuter_test.cc165 if (!gpu_ring_order_) { in Init()
166 gpu_ring_order_ = absl::make_unique<string>(); in Init()
174 gpu_ring_order_.get(), work_queue_); in Init()
439 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::__anon153c086f0111::PermuterTest
Dcollective_executor_mgr.h73 string gpu_ring_order_; variable
Dring_gatherer_test.cc175 if (!gpu_ring_order_) { in Init()
176 gpu_ring_order_ = absl::make_unique<string>(); in Init()
184 gpu_ring_order_.get(), work_queue_); in Init()
533 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingGathererTest
Dring_reducer_test.cc198 if (!gpu_ring_order_) { in Init()
199 gpu_ring_order_ = absl::make_unique<string>(); in Init()
207 gpu_ring_order_.get(), work_queue_); in Init()
565 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingReducerTest
Dhierarchical_tree_broadcaster_test.cc258 if (!gpu_ring_order_) { in Init()
259 gpu_ring_order_ = absl::make_unique<string>(); in Init()
267 gpu_ring_order_.get(), work_queue_); in Init()
722 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::__anonb5878a120111::HierarchicalTreeBroadcasterTest
Dbase_collective_executor.cc349 cp->group.gpu_ring_order = *gpu_ring_order_; in CompleteParamsAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpc_collective_executor_mgr.cc56 &gpu_ring_order_, work_queue_); in Create()