Searched refs:gpu_ring_order_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 33 gpu_ring_order_( in CollectiveExecutorMgr()
64 &gpu_ring_order_); in Create()
|
| D | base_collective_executor.h | 105 gpu_ring_order_(gpu_ring_order) {} in BaseCollectiveExecutor()
163 const string* gpu_ring_order_; // Not owned. variable
|
| D | collective_executor_mgr.h | 66 string gpu_ring_order_; variable
|
| D | ring_gatherer_test.cc | 172 if (!gpu_ring_order_) { in Init()
173 gpu_ring_order_ = absl::make_unique<string>(); in Init()
180 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
529 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 194 if (!gpu_ring_order_) { in Init()
195 gpu_ring_order_ = absl::make_unique<string>(); in Init()
202 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
557 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingReducerTest
|
| D | hierarchical_tree_broadcaster_test.cc | 252 if (!gpu_ring_order_) { in Init()
253 gpu_ring_order_ = absl::make_unique<string>(); in Init()
260 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
726 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::__anonfb751b9d0111::HierarchicalTreeBroadcasterTest
|
| D | base_collective_executor.cc | 287 cp->instance.gpu_ring_order = *gpu_ring_order_; in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 53 &gpu_ring_order_); in Create()
|