Searched refs:DeviceResolverDistributed (Results 1 – 10 of 10) 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()
82 DeviceResolverDistributed* dres, bool is_failed, in FakeWorker()166 DeviceResolverDistributed* device_resolver_;281 DeviceResolverDistributed* dev_resolver = dev_resolver_item.second; in ResolveDeviceAttributes()304 DeviceResolverDistributed* dev_res = new DeviceResolverDistributed(dev_mgr); in DefineWorker()354 std::unordered_map<string, DeviceResolverDistributed*> dev_resolvers_;
30 class DeviceResolverDistributed : public DeviceResolverInterface {32 explicit DeviceResolverDistributed(const DeviceMgr* dev_mgr);
26 class DeviceResolverDistributed; variable33 DeviceResolverDistributed* dev_resolver,
25 class DeviceResolverDistributed; variable39 std::unique_ptr<DeviceResolverDistributed> dev_resolver,
61 std::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()176 auto dev_resolver = std::make_unique<DeviceResolverDistributed>(device_mgr); in CreateProdRpcCollectiveExecutorMgr()
144 dev_resolvers_[worker_name] = std::make_unique<DeviceResolverDistributed>( in DefineWorker()306 absl::flat_hash_map<string, std::unique_ptr<DeviceResolverDistributed>>
91 DeviceResolverDistributed* dev_resolver, in CollectiveParamResolverDistributed()