Home
last modified time | relevance | path

Searched refs:DeviceResolverLocal (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_resolver_local.cc22 Status DeviceResolverLocal::GetDeviceAttributes(const string& device, in GetDeviceAttributes()
36 Status DeviceResolverLocal::GetAllDeviceAttributes( in GetAllDeviceAttributes()
42 Status DeviceResolverLocal::UpdateDeviceAttributes( in UpdateDeviceAttributes()
Ddevice_resolver_local.h29 class DeviceResolverLocal : public DeviceResolverInterface {
31 explicit DeviceResolverLocal(const DeviceMgr* dev_mgr) : dev_mgr_(dev_mgr) {} in DeviceResolverLocal() function
Ddevice_resolver_local_test.cc40 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in DeviceResolverLocalTest()
44 std::unique_ptr<DeviceResolverLocal> drl_;
Dcollective_rma_local_test.cc50 drl_ = absl::make_unique<DeviceResolverLocal>(device_mgr_.get()); in CollectiveRemoteAccessLocalTest()
63 std::unique_ptr<DeviceResolverLocal> drl_;
Dcollective_executor_mgr.cc100 auto device_resolver = absl::make_unique<DeviceResolverLocal>(device_mgr); in CreateProdLocalCollectiveExecutorMgr()
Dcollective_executor_mgr_test.cc52 new DeviceResolverLocal(device_mgr_.get())); in CollectiveExecutorMgrTest()
Dcollective_param_resolver_local_test.cc52 drl_.reset(new DeviceResolverLocal(device_mgr_.get())); in CollectiveParamResolverLocalTest()
88 std::unique_ptr<DeviceResolverLocal> drl_;
Dcollective_test_util.cc169 absl::make_unique<DeviceResolverLocal>(test_env->device_mgr.get()); in CreateCollectiveTestEnv()