Home
last modified time | relevance | path

Searched refs:device_mgr_ (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc47 device_mgr_.reset(new DeviceMgr(std::move(devices))); in CollectiveRemoteAccessLocalTest()
48 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveRemoteAccessLocalTest()
49 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in CollectiveRemoteAccessLocalTest()
51 rma_.reset(new CollectiveRemoteAccessLocal(device_mgr_.get(), drl_.get(), in CollectiveRemoteAccessLocalTest()
55 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon962ef3b70111::CollectiveRemoteAccessLocalTest
65 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
108 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:2", &cpu2)); in TEST_F()
127 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:1", &cpu1)); in TEST_F()
Dcollective_executor_mgr_test.cc43 device_mgr_.reset(new DeviceMgr(std::move(devices))); in CollectiveExecutorMgrTest()
45 new DeviceResolverLocal(device_mgr_.get())); in CollectiveExecutorMgrTest()
47 new CollectiveParamResolverLocal(cp, device_mgr_.get(), drl.get(), in CollectiveExecutorMgrTest()
49 cme_.reset(new CollectiveExecutorMgr(cp, device_mgr_.get(), std::move(drl), in CollectiveExecutorMgrTest()
54 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon0f9ea3e10111::CollectiveExecutorMgrTest
Ddevice_resolver_local_test.cc40 device_mgr_.reset(new DeviceMgr(std::move(devices))); in DeviceResolverLocalTest()
41 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in DeviceResolverLocalTest()
44 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon0d30c4ef0111::DeviceResolverLocalTest
Drendezvous_mgr.cc35 : device_mgr_(device_mgr), local_(NewLocalRendezvous()) {} in IntraProcessRendezvous()
88 Status s = device_mgr_->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
94 s = device_mgr_->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
Dcollective_param_resolver_local_test.cc42 device_mgr_.reset(new DeviceMgr(std::move(devices))); in CollectiveParamResolverLocalTest()
43 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveParamResolverLocalTest()
44 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in CollectiveParamResolverLocalTest()
77 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::CollectiveParamResolverLocalTest
Ddevice.h179 DeviceMgr* device_mgr() const { return device_mgr_; } in device_mgr()
209 DeviceMgr* device_mgr_ = nullptr; variable
Dprocess_function_library_runtime_test.cc75 device_mgr_.reset(new DeviceMgr(std::move(devices))); in Init()
76 TF_CHECK_OK(device_mgr_->LookupDevice( in Init()
78 TF_CHECK_OK(device_mgr_->LookupDevice( in Init()
81 Status status = device_mgr_->LookupDevice( in Init()
93 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
95 rendezvous_ = new IntraProcessRendezvous(device_mgr_.get()); in Init()
209 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anonbd125a7f0111::ProcessFunctionLibraryRuntimeTest
Dpartitioning_utils_test.cc50 device_mgr_.reset(new DeviceMgr(std::move(devices))); in SetUp()
52 for (auto d : device_mgr_->ListDevices()) { in SetUp()
105 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anone4792f6c0111::PartitioningUtilsTest
Ddevice_mgr.cc31 CHECK(d->device_mgr_ == nullptr); in DeviceMgr()
32 d->device_mgr_ = this; in DeviceMgr()
Ddirect_session.cc278 device_mgr_(device_mgr), in DirectSession()
310 const string mapping_str = device_mgr_->DeviceMappingString(); in DirectSession()
318 for (auto d : device_mgr_->ListDevices()) { in DirectSession()
341 for (auto d : device_mgr_->ListDevices()) { in ~DirectSession()
344 for (auto d : device_mgr_->ListDevices()) { in ~DirectSession()
482 run_state.rendez = new IntraProcessRendezvous(device_mgr_.get()); in RunInternal()
502 new DeviceResolverLocal(device_mgr_.get())); in RunInternal()
504 new CollectiveParamResolverLocal(options_.config, device_mgr_.get(), in RunInternal()
508 options_.config, device_mgr_.get(), std::move(drl), std::move(cprl))); in RunInternal()
859 run_state->rendez = new IntraProcessRendezvous(device_mgr_.get()); in PRunSetup()
[all …]
Dfunction_threadpool_test.cc65 device_mgr_.reset(new DeviceMgr(std::move(devices))); in Init()
67 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
193 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anonaa65071b0111::FunctionLibraryRuntimeTest
Drendezvous_mgr.h59 const DeviceMgr* device_mgr_;
Dprocess_function_library_runtime.h140 const DeviceMgr* device_mgr() { return device_mgr_; } in device_mgr()
291 const DeviceMgr* const device_mgr_; variable
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpc_collective_executor_mgr_test.cc47 device_mgr_.reset(new DeviceMgr(std::move(devices))); in RpcCollectiveExecutorMgrTest()
49 device_mgr_.get(), worker_cache, task_name)); in RpcCollectiveExecutorMgrTest()
52 device_mgr_.get(), dr.get(), in RpcCollectiveExecutorMgrTest()
55 cme_.reset(new RpcCollectiveExecutorMgr(options.config, device_mgr_.get(), in RpcCollectiveExecutorMgrTest()
61 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::RpcCollectiveExecutorMgrTest
Dworker_session.h49 return device_mgr_ ? device_mgr_.get() : borrowed_device_mgr_; in device_mgr()
78 const std::unique_ptr<DeviceMgr> device_mgr_; member
Dsession_mgr_test.cc49 device_mgr_ = absl::make_unique<DeviceMgr>( in SessionMgrTest()
51 env_.local_devices = device_mgr_->ListDevices(); in SessionMgrTest()
52 env_.device_mgr = device_mgr_.get(); in SessionMgrTest()
55 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::SessionMgrTest
Ddevice_resolver_distributed_test.cc66 : name_(name), device_mgr_(dev_mgr), device_resolver_(dres) {} in FakeWorker()
72 device_mgr_->ListDeviceAttributes(&dev_attr); in GetStatusAsync()
81 DeviceMgr* device_mgr_; member in tensorflow::__anon0903cdc10111::FakeWorker
Dgraph_mgr.cc57 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
142 device_mgr_, worker_env_->env, gdef.versions().producer(), in InitItem()
163 Status s = device_mgr_->LookupDevice(name, &device); in InitItem()
209 Status s = device_mgr_->LookupDevice(device_name, &unit->device); in InitItem()
479 [this](const string& name) { device_mgr_->ClearContainers({name}); }); in StartParallelExecutors()
Dworker_session.cc106 device_mgr_(std::move(device_mgr)), in WorkerSession()
130 device_mgr_(nullptr), in WorkerSession()
Dcollective_rma_distributed_test.cc66 device_mgr_(dev_mgr), in FakeWorker()
78 device_mgr_->ListDeviceAttributes(&dev_attr); in GetStatusAsync()
120 DeviceMgr* device_mgr_; member in tensorflow::__anon7e5c43490111::FakeWorker
Dcollective_param_resolver_distributed_test.cc51 : name_(name), device_mgr_(dev_mgr), param_resolver_(cpres) {} in FakeWorker()
57 device_mgr_->ListDeviceAttributes(&dev_attr); in GetStatusAsync()
80 DeviceMgr* device_mgr_; member in tensorflow::__anonc2f8b6eb0111::FakeWorker
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op_test.cc73 device_mgr_ = absl::make_unique<DeviceMgr>(std::move(devices)); in Init()
75 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
81 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::CreateXlaLaunchOpTest
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc48 device_mgr_ = absl::make_unique<DeviceMgr>(std::move(devices));
51 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, &flib_def_,
64 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anone5cc3cd90111::TestEnv
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc71 device_mgr_ = absl::make_unique<DeviceMgr>(DeviceFactory::NewDevice(
73 worker_env_.local_devices = device_mgr_->ListDevices();
74 worker_env_.device_mgr = device_mgr_.get();
81 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::eager::__anon4b5abff60111::EagerServiceImplTest
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc94 op_params.device = device_mgr_->ListDevices()[0]; in MakeIterator()
115 device_mgr_(device_mgr), in Dataset()

12