| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.h | 31 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()
58 dev_resolver_->GetDeviceLocalitiesAsync(ci_params, localities, done); in GetDeviceLocalitiesAsync()
64 dev_resolver_->GetLocalityAsync(device, task, locality, done); in GetLocalityAsync()
68 dev_resolver_->ClearTask(task); in ClearTask()
86 DeviceResolverInterface* dev_resolver_; // not owned variable
|
| D | collective_executor_mgr.h | 42 return dev_resolver_.get(); in GetDeviceResolver()
63 std::unique_ptr<DeviceResolverInterface> dev_resolver_; variable
|
| D | collective_executor_mgr.cc | 31 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()
60 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
|
| D | ring_gatherer_test.cc | 170 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
171 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
519 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 192 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
193 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
546 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::RingReducerTest
|
| D | collective_param_resolver_local.h | 233 DeviceResolverInterface* dev_resolver_; // Not owned. variable
|
| D | hierarchical_tree_broadcaster_test.cc | 250 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
251 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
715 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::__anonf709145b0111::HierarchicalTreeBroadcasterTest
|
| D | collective_param_resolver_local.cc | 45 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()
434 dev_resolver_->GetDeviceLocalitiesAsync( in InitInstanceSharedParams()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 110 dev_resolver_->ClearTask(peer_task); in RecvFromPeer()
133 dev_resolver_->GetLocalityAsync( in RecvFromPeer()
152 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 153 dev_resolver_->ClearTask(peer_task); in RecvFromPeer()
177 dev_resolver_->GetLocalityAsync( in RecvFromPeer()
|
| D | rpc_collective_executor_mgr.cc | 50 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
|