Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc49 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in CollectiveRemoteAccessLocalTest()
50 drl_ = absl::make_unique<DeviceResolverLocal>(device_mgr_.get()); in CollectiveRemoteAccessLocalTest()
52 cp, device_mgr_.get(), drl_.get(), kTaskName); in CollectiveRemoteAccessLocalTest()
53 rma_ = absl::make_unique<CollectiveRemoteAccessLocal>(device_mgr_.get(), in CollectiveRemoteAccessLocalTest()
61 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anon4ebbc1950111::CollectiveRemoteAccessLocalTest
72 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
115 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:2", &cpu2)); in TEST_F()
134 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:1", &cpu1)); in TEST_F()
172 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
194 TF_ASSERT_OK(device_mgr_->LookupDevice(kTaskName + "/device:CPU:0", &cpu0)); in TEST_F()
[all …]
Dcollective_executor_mgr_test.cc44 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in CollectiveExecutorMgrTest()
46 new DeviceResolverLocal(device_mgr_.get())); in CollectiveExecutorMgrTest()
48 new CollectiveParamResolverLocal(cp, device_mgr_.get(), drl.get(), in CollectiveExecutorMgrTest()
50 cme_.reset(new CollectiveExecutorMgr(cp, device_mgr_.get(), std::move(drl), in CollectiveExecutorMgrTest()
56 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anonfd8e0cff0111::CollectiveExecutorMgrTest
Ddevice_resolver_local_test.cc39 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in DeviceResolverLocalTest()
40 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in DeviceResolverLocalTest()
43 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anond7574e8d0111::DeviceResolverLocalTest
Drendezvous_mgr.cc154 : device_mgr_(device_mgr), local_(this) {} in RefCountedIntraProcessRendezvous()
170 IntraProcessRecvAsyncImpl(device_mgr_, &local_, key, args, std::move(done)); in RecvAsync()
179 : device_mgr_(device_mgr), local_(nullptr) {} in PrivateIntraProcessRendezvous()
196 IntraProcessRecvAsyncImpl(device_mgr_, &local_, key, args, std::move(done)); in RecvAsync()
Drendezvous_mgr.h55 const DeviceMgr* device_mgr_;
84 const DeviceMgr* device_mgr_;
Dprocess_function_library_runtime_test.cc53 explicit TestClusterFLR(DeviceMgr* device_mgr) : device_mgr_(device_mgr) {} in TestClusterFLR()
81 DeviceMgr* remote_device_mgr() const override { return device_mgr_; } in remote_device_mgr()
86 DeviceMgr* device_mgr_; member in tensorflow::__anon96a4e55d0111::TestClusterFLR
112 device_mgr_ = std::make_unique<DynamicDeviceMgr>(); in ProcessFunctionLibraryRuntimeTest()
113 TF_CHECK_OK(device_mgr_->AddDevices(std::move(created_devices))); in ProcessFunctionLibraryRuntimeTest()
114 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
116 TF_CHECK_OK(device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
121 device_mgr_ in ProcessFunctionLibraryRuntimeTest()
125 Status status = device_mgr_->LookupDevice( in ProcessFunctionLibraryRuntimeTest()
138 cluster_flr_.reset(new TestClusterFLR(device_mgr_.get())); in Init()
[all …]
Dkernel_benchmark_testlib.cc70 device_mgr_ = absl::make_unique<StaticDeviceMgr>( in Benchmark()
72 device_ = device_mgr_->ListDevices()[0]; in Benchmark()
94 device_mgr_.get(), Env::Default(), nullptr, graph_def_version, in Benchmark()
136 device_mgr_.reset(); in ~Benchmark()
Dfunction_threadpool_test.cc64 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in Init()
66 device_mgr_.get(), Env::Default(), /*config=*/nullptr, in Init()
197 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anonbf8091390111::FunctionLibraryRuntimeTest
Dcollective_param_resolver_local_test.cc50 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in CollectiveParamResolverLocalTest()
51 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveParamResolverLocalTest()
57 prl_.reset(new CollectiveParamResolverLocal(cp, device_mgr_.get(), in ResetParamResolver()
81 TF_CHECK_OK(device_mgr_->LookupDevice(device_name, &device)); in GetDeviceAttributes()
86 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::CollectiveParamResolverLocalTest
Dpartitioning_utils_test.cc51 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in SetUp()
53 for (auto d : device_mgr_->ListDevices()) { in SetUp()
112 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anone51cedca0111::PartitioningUtilsTest
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpc_collective_executor_mgr_test.cc50 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in RpcCollectiveExecutorMgrTest()
52 new DeviceResolverDistributed(device_mgr_.get())); in RpcCollectiveExecutorMgrTest()
55 device_mgr_.get(), dr.get(), in RpcCollectiveExecutorMgrTest()
59 options.config, device_mgr_.get(), std::move(dr), std::move(cpr), in RpcCollectiveExecutorMgrTest()
64 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::RpcCollectiveExecutorMgrTest
Dworker_session.h41 return device_mgr_ ? device_mgr_.get() : borrowed_device_mgr_; in device_mgr()
116 const std::unique_ptr<const DeviceMgr> device_mgr_; variable
Dsession_mgr_test.cc49 device_mgr_ = absl::make_unique<StaticDeviceMgr>( 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
Dgraph_mgr.cc60 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
140 device_mgr_, worker_env_->env, /*config=*/&config_proto, in InitItem()
175 Status s = device_mgr_->LookupDevice(name, &device); in InitItem()
221 Status s = device_mgr_->LookupDevice(device_name, &unit->device); in InitItem()
520 [this](const string& name) { device_mgr_->ClearContainers({name}); }); in StartParallelExecutors()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops_test.cc43 : device_mgr_(testing::CreateTestingDeviceMgr()), in VariableOpsTest()
44 ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} in VariableOpsTest()
49 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon4afb60000111::VariableOpsTest
Drestore_ops_test.cc41 : device_mgr_(testing::CreateTestingDeviceMgr()), in RestoreOpsTest()
42 ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} in RestoreOpsTest()
47 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anonf86beb7e0111::RestoreOpsTest
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dconstant_loading_test.cc40 : device_mgr_(testing::CreateTestingDeviceMgr()), in ConstantTest()
41 ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} in ConstantTest()
46 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anon2bedbe4f0111::ConstantTest
Dsaved_variable_loading_test.cc48 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in SavedVariableLoadingTest()
49 ctx_ = testing::CreateTestingEagerContext(device_mgr_.get()); in SavedVariableLoadingTest()
55 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anonb295f13d0111::SavedVariableLoadingTest
Dtf_concrete_function_loading_test.cc40 : device_mgr_(testing::CreateTestingDeviceMgr()), in SavedConcreteFunctionLoadingTest()
41 ctx_(testing::CreateTestingEagerContext(device_mgr_.get())) {} in SavedConcreteFunctionLoadingTest()
46 std::unique_ptr<StaticDeviceMgr> device_mgr_; member in tensorflow::__anonde6fafd60111::SavedConcreteFunctionLoadingTest
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_test.cc103 device_mgr_ = new StaticDeviceMgr(std::move(devices)); in PackedTensorHandleTest()
108 /* async= */ false, device_mgr_, in PackedTensorHandleTest()
114 delete device_mgr_; in ~PackedTensorHandleTest()
121 return device_mgr_->ListDevices(); in ListDevices()
135 StaticDeviceMgr* device_mgr_; member in tensorflow::PackedTensorHandleTest
339 device_mgr_ = new StaticDeviceMgr(std::move(devices)); in RemoteTensorHandleTest()
344 /* async= */ false, device_mgr_, in RemoteTensorHandleTest()
350 delete device_mgr_; in ~RemoteTensorHandleTest()
357 return device_mgr_->ListDevices(); in ListDevices()
366 StaticDeviceMgr* device_mgr_; member in tensorflow::RemoteTensorHandleTest
Dkernel_and_device_test.cc49 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices));
52 device_mgr_.get(), Env::Default(), /*config=*/nullptr,
66 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::__anond3bba5f70111::TestEnv
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator_test.cc76 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in Init()
78 device_mgr_.get(), Env::Default(), /*config=*/nullptr, in Init()
85 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::XlaKernelCreatorTest
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.cc79 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(device)); in OpsTestBase()
86 device_mgr_.get(), Env::Default(), /*config=*/nullptr, in OpsTestBase()
108 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(device)); in SetDevice()
110 device_mgr_.get(), Env::Default(), /*config=*/nullptr, in SetDevice()
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc112 op_params.device = device_mgr_->ListDevices()[0]; in MakeIterator()
148 device_mgr_(device_mgr), in Dataset()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc158 device_mgr_ = absl::make_unique<StaticDeviceMgr>(
160 worker_env_.local_devices = device_mgr_->ListDevices();
161 worker_env_.device_mgr = device_mgr_.get();
168 std::unique_ptr<DeviceMgr> device_mgr_; member in tensorflow::eager::__anon58c807940111::EagerServiceImplTest
775 context_id_, ctx, device_mgr_.get()); in Init()
944 TF_ASSERT_OK(device_mgr_->LookupDevice(local_device_, &local_device)); in TEST_F()
992 TF_ASSERT_OK(device_mgr_->LookupDevice(local_device_, &local_device)); in TEST_F()
1215 new tensorflow::IntraProcessRendezvous(device_mgr_.get()); in TEST_F()
1220 /*async=*/false, device_mgr_.get(), false, rendezvous); in TEST_F()

12