| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 30 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
60 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
61 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| D | device_resolver_local.h | 29 DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
44 const DeviceMgr* dev_mgr_;
|
| D | device_resolver_local.cc | 27 Status s = dev_mgr_->LookupDevice(device_name, &dev); in GetDeviceLocalitiesAsync()
42 Status s = dev_mgr_->LookupDevice(device, &dev); in GetLocalityAsync()
|
| D | ring_gatherer_test.cc | 164 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
167 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
170 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
171 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
174 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
384 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
386 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
483 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather()
523 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 186 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
189 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
192 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
193 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
196 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
408 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
410 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
513 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce()
550 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingReducerTest
|
| D | collective_rma_local.h | 30 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
85 const DeviceMgr* dev_mgr_; // not owned
|
| D | base_collective_executor.h | 102 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
156 const DeviceMgr* dev_mgr_; // Not owned. variable
|
| D | hierarchical_tree_broadcaster_test.cc | 246 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
247 dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
250 dev_resolver_.reset(new DeviceResolverLocal(dev_mgr_.get())); in Init()
251 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
254 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
560 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)); in DeviceInstance()
684 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast()
719 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::__anonf709145b0111::HierarchicalTreeBroadcasterTest
|
| D | collective_executor_mgr.h | 62 const DeviceMgr* dev_mgr_; variable
|
| D | collective_param_resolver_local.h | 232 const DeviceMgr* dev_mgr_; variable
|
| D | base_collective_executor.cc | 254 new CollectiveContext(this, dev_mgr_, ctx, CtxParams(ctx), col_params, in ExecuteAsync()
|
| D | collective_param_resolver_local.cc | 44 dev_mgr_(dev_mgr), in CollectiveParamResolverLocal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 110 if (!dev_mgr_) dev_mgr_.reset(new DeviceMgr(std::move(local_devices))); in Init()
112 &col_exec_mgr_, /*remote_access=*/nullptr, kStepId, dev_mgr_.get(), in Init()
219 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(device_name_, &device_)) in DeviceInstance()
221 << parent_->dev_mgr_->DebugString(); in DeviceInstance()
291 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce()
317 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::NcclReducerTest
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed.cc | 24 : dev_mgr_(dev_mgr), worker_cache_(worker_cache), task_name_(task_name) {} in DeviceResolverDistributed()
33 Status s = dev_mgr_->LookupDevice(device, &dev); in GetLocalityAsync()
|
| D | rpc_collective_executor_mgr.cc | 50 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
52 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| D | device_resolver_distributed.h | 59 const DeviceMgr* dev_mgr_; // Not owned variable
|
| D | collective_rma_distributed.cc | 123 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 152 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
155 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|