Searched refs:gpu_ring_order_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 33 gpu_ring_order_( in CollectiveExecutorMgr()
62 &gpu_ring_order_); in Create()
|
| D | base_collective_executor.h | 104 gpu_ring_order_(gpu_ring_order) {} in BaseCollectiveExecutor()
158 const string* gpu_ring_order_; // Not owned. variable
|
| D | collective_executor_mgr.h | 65 string gpu_ring_order_; variable
|
| D | ring_gatherer_test.cc | 169 if (!gpu_ring_order_) gpu_ring_order_.reset(new string()); in Init()
174 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
524 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 191 if (!gpu_ring_order_) gpu_ring_order_.reset(new string()); in Init()
196 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
551 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::RingReducerTest
|
| D | hierarchical_tree_broadcaster_test.cc | 249 if (!gpu_ring_order_) gpu_ring_order_.reset(new string()); in Init()
254 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
720 std::unique_ptr<string> gpu_ring_order_; member in tensorflow::__anonb5878a120111::HierarchicalTreeBroadcasterTest
|
| D | base_collective_executor.cc | 279 cp->instance.gpu_ring_order = *gpu_ring_order_; in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 156 &gpu_ring_order_); in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 53 &gpu_ring_order_); in Create()
|