| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 30 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
62 dev_mgr_, dev_resolver_.get(), work_queue_, step_id); in Create()
63 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()
47 const DeviceMgr* dev_mgr_;
|
| D | device_resolver_local.cc | 27 Status s = dev_mgr_->LookupDevice(device_name, &dev); in GetAllDeviceAttributesAsync()
42 Status s = dev_mgr_->LookupDevice(device, &dev); in GetDeviceAttributesAsync()
|
| D | buf_rendezvous.h | 44 : step_id_(step_id), dev_mgr_(dev_mgr) {} in BufRendezvous()
104 const DeviceMgr* const dev_mgr_; // Not owned. variable
|
| D | ring_gatherer_test.cc | 167 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
170 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
175 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
177 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), work_queue_, in Init()
180 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
391 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
393 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
487 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoGather()
528 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingGathererTest
|
| D | buf_rendezvous_test.cc | 50 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in InitializeDevice()
51 br_ = absl::make_unique<BufRendezvous>(123, dev_mgr_.get()); in InitializeDevice()
59 TF_CHECK_OK(dev_mgr_->LookupDevice(*kDefaultDeviceName, &default_device_));
67 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::__anon3d4b4ed70111::BufRendezvousTest
|
| D | ring_reducer_test.cc | 189 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
192 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
197 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
199 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), work_queue_, in Init()
202 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
416 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
418 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
518 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoReduce()
556 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingReducerTest
|
| D | collective_rma_local.h | 33 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
97 const DeviceMgr* dev_mgr_; // not owned
|
| D | base_collective_executor.h | 103 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
161 const DeviceMgr* dev_mgr_; // Not owned. variable
|
| D | hierarchical_tree_broadcaster_test.cc | 249 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
250 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
255 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
257 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), work_queue_, in Init()
260 &col_exec_mgr_, rma_, kStepId, dev_mgr_.get(), gpu_ring_order_.get()); in Init()
568 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)); in DeviceInstance()
689 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in DoBroadcast()
725 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::__anonfb751b9d0111::HierarchicalTreeBroadcasterTest
|
| D | collective_executor_mgr.h | 63 const DeviceMgr* dev_mgr_; variable
|
| D | collective_rma_local.cc | 45 Status status = dev_mgr_->LookupDevice(peer_device, &from_device); in RecvFromPeer()
|
| D | buf_rendezvous.cc | 124 Status consumebuf_status = dev_mgr_->LookupDevice(device_name, &device); in ConsumeBuf()
|
| D | collective_param_resolver_local.h | 231 const DeviceMgr* dev_mgr_; variable
|
| D | base_collective_executor.cc | 257 new CollectiveContext(this, dev_mgr_, ctx, CtxParams(ctx), col_params, in ExecuteAsync()
|
| D | collective_param_resolver_local.cc | 46 dev_mgr_(dev_mgr), in CollectiveParamResolverLocal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 117 if (!dev_mgr_) in Init()
118 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
120 &col_exec_mgr_, /*remote_access=*/nullptr, kStepId, dev_mgr_.get(), in Init()
249 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(device_name_, &device_)) in DeviceInstance()
251 << parent_->dev_mgr_->DebugString(); in DeviceInstance()
327 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in RunReduce()
358 parent_->col_exec_, parent_->dev_mgr_.get(), in RunBroadcast()
396 CollectiveContext col_ctx(parent_->col_exec_, parent_->dev_mgr_.get(), in RunGather()
427 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::NcclTestBase
|
| /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()
32 Status s = dev_mgr_->LookupDevice(device, &dev); in GetDeviceAttributesAsync()
|
| D | rpc_collective_executor_mgr.cc | 51 dev_mgr_, dev_resolver_.get(), work_queue_, worker_cache_, step_id); in Create()
52 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| D | device_resolver_distributed.h | 62 const DeviceMgr* dev_mgr_; // Not owned variable
|
| D | collective_rma_distributed.cc | 124 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()
|