Searched refs:DeviceResolverDistributed (Results 1 – 11 of 11) sorted by relevance
23 DeviceResolverDistributed::DeviceResolverDistributed(const DeviceMgr* dev_mgr) { in DeviceResolverDistributed() function in tensorflow::DeviceResolverDistributed30 Status DeviceResolverDistributed::GetDeviceAttributes( in GetDeviceAttributes()41 Status DeviceResolverDistributed::GetAllDeviceAttributes( in GetAllDeviceAttributes()57 Status DeviceResolverDistributed::UpdateDeviceAttributes( in UpdateDeviceAttributes()
30 class DeviceResolverDistributed : public DeviceResolverInterface {32 explicit DeviceResolverDistributed(const DeviceMgr* dev_mgr);
25 class DeviceResolverDistributed; variable39 std::unique_ptr<DeviceResolverDistributed> dev_resolver,
26 class DeviceResolverDistributed; variable33 DeviceResolverDistributed* dev_resolver,
66 DeviceResolverDistributed* dres, bool is_failed) in FakeWorker()136 DeviceResolverDistributed* device_resolver_;233 DeviceResolverDistributed* dev_resolver = dev_resolver_item.second; in ResolveDeviceAttributes()255 DeviceResolverDistributed* dev_res = new DeviceResolverDistributed(dev_mgr); in DefineWorker()286 std::unordered_map<string, DeviceResolverDistributed*> dev_resolvers_;
61 absl::make_unique<DeviceResolverDistributed>(dev_mgr_.get()); in SetUp()74 std::unique_ptr<DeviceResolverDistributed> dev_resolver_;
51 std::unique_ptr<DeviceResolverDistributed> dr( in RpcCollectiveExecutorMgrTest()52 new DeviceResolverDistributed(device_mgr_.get())); in RpcCollectiveExecutorMgrTest()
30 std::unique_ptr<DeviceResolverDistributed> dev_resolver, in RpcCollectiveExecutorMgr()
166 dev_resolvers_[worker_name] = absl::make_unique<DeviceResolverDistributed>( in DefineWorker()310 absl::flat_hash_map<string, std::unique_ptr<DeviceResolverDistributed>>
93 DeviceResolverDistributed* dev_resolver, WorkerCacheInterface* worker_cache, in CollectiveParamResolverDistributed()
282 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in Init()283 new DeviceResolverDistributed(worker_env_.device_mgr)); in Init()451 std::unique_ptr<DeviceResolverDistributed> dev_resolver( in UpdateServerDef()452 new DeviceResolverDistributed(worker_env_.device_mgr)); in UpdateServerDef()