Home
last modified time | relevance | path

Searched refs:dev_mgr_ (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_executor_mgr.cc31 : dev_mgr_(dev_mgr), in CollectiveExecutorMgr()
64 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
65 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
Ddevice_resolver_local.h31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal()
43 const DeviceMgr* dev_mgr_;
Dpermuter_test.cc162 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
163 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
168 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
170 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
173 dev_mgr_.get(), in Init()
328 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)); in DeviceInstance()
403 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoPermute()
438 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::__anonfb3b1ff60111::PermuterTest
Dcollective_rma_local.h31 : dev_mgr_(dev_mgr), in CollectiveRemoteAccessLocal()
75 const DeviceMgr* dev_mgr_; // not owned
Dbuf_rendezvous.h45 : step_id_(step_id), dev_mgr_(dev_mgr) {} in BufRendezvous()
124 const DeviceMgr* const dev_mgr_; // Not owned. variable
Dring_gatherer_test.cc170 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
173 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
178 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
180 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
183 dev_mgr_.get(), in Init()
395 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
397 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
492 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoGather()
532 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingGathererTest
Dring_reducer_test.cc193 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
196 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
201 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
203 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
206 dev_mgr_.get(), in Init()
421 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)) in DeviceInstance()
423 << " existing devices: " << parent_->dev_mgr_->DebugString(); in DeviceInstance()
525 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoReduce()
564 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::RingReducerTest
Dbase_collective_executor.h104 dev_mgr_(dev_mgr), in BaseCollectiveExecutor()
140 const DeviceMgr* dev_mgr_; // Not owned. variable
Dbuf_rendezvous_test.cc52 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in InitializeDevice()
53 br_ = absl::make_unique<BufRendezvous>(123, dev_mgr_.get()); in InitializeDevice()
61 TF_CHECK_OK(dev_mgr_->LookupDevice(*kDefaultDeviceName, &default_device_));
69 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::__anon5d7062b30111::BufRendezvousTest
Dhierarchical_tree_broadcaster_test.cc255 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
256 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
261 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
263 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
266 dev_mgr_.get(), in Init()
567 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(dev_name, &device_)); in DeviceInstance()
686 parent_->dev_mgr_.get(), &ctx, &op_params, col_params_, exec_key, in DoBroadcast()
721 std::unique_ptr<tensorflow::DeviceMgr> dev_mgr_; member in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
Ddevice_resolver_local.cc26 Status s = dev_mgr_->LookupDevice(device, &dev); in GetDeviceAttributes()
Dcollective_executor_mgr.h70 const DeviceMgr* dev_mgr_; variable
Dcollective_rma_local.cc42 Status status = dev_mgr_->LookupDevice(peer_device, &from_device); in RecvFromPeer()
Dcollective_param_resolver_local.h189 const DeviceMgr* dev_mgr_; variable
Dbuf_rendezvous.cc156 Status consumebuf_status = dev_mgr_->LookupDevice(device_name, &device); in ConsumeBuf()
Dbase_collective_executor.cc316 this, cem_->GetNcclCommunicator(), dev_mgr_, ctx, CtxParams(ctx), in ExecuteAsync()
Dcollective_param_resolver_local.cc45 dev_mgr_(dev_mgr), in CollectiveParamResolverLocal()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc124 if (!dev_mgr_) in Init()
125 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(local_devices)); in Init()
127 &col_exec_mgr_, /*remote_access=*/nullptr, kStepId, dev_mgr_.get(), in Init()
254 TF_CHECK_OK(parent_->dev_mgr_->LookupDevice(device_name_, &device_)) in DeviceInstance()
256 << parent_->dev_mgr_->DebugString(); in DeviceInstance()
331 parent_->dev_mgr_.get(), in RunReduce()
363 parent_->dev_mgr_.get(), in RunBroadcast()
404 parent_->dev_mgr_.get(), in RunGather()
439 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::NcclTestBase
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc59 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in SetUp()
61 absl::make_unique<DeviceResolverDistributed>(dev_mgr_.get()); in SetUp()
73 std::unique_ptr<DeviceMgr> dev_mgr_; member in tensorflow::__anone296589f0111::DeviceResDistTest
Drpc_collective_executor_mgr.cc52 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
55 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
Dcollective_rma_distributed.cc127 Status status = dev_mgr_->LookupDevice("CPU:0", &cpu_dev); in RecvFromPeer()