Searched refs:dev_resolvers_ (Results 1 – 2 of 2) sorted by relevance
235 for (auto it : dev_resolvers_) { in ~CollRMADistTest()257 device_mgrs_[0], dev_resolvers_[dev0_worker_name], work_queue_, &wc_, in SetUp()280 for (auto& dev_resolver_item : dev_resolvers_) { in ResolveDeviceAttributes()305 dev_resolvers_[worker_name] = dev_res; in DefineWorker()318 auto it = dev_resolvers_.find(worker_name); in RestartWorker()319 if (it != dev_resolvers_.end()) { in RestartWorker()321 dev_resolvers_.erase(it); in RestartWorker()354 std::unordered_map<string, DeviceResolverDistributed*> dev_resolvers_; member in tensorflow::__anon0d8d35760111::CollRMADistTest
144 dev_resolvers_[worker_name] = std::make_unique<DeviceResolverDistributed>( in DefineWorker()149 dev_resolvers_[worker_name].get(), &nccl_communicator_, &wc_, in DefineWorker()277 TF_ASSERT_OK(dev_resolvers_[task]->GetDeviceAttributes( in ValidateDeviceResolver()307 dev_resolvers_; member in tensorflow::__anon1d35d2f80111::DeviceResDistTest