Searched refs:dev_resolver_ (Results 1 – 9 of 9) sorted by relevance
60 dev_resolver_ = in SetUp()70 TF_ASSERT_OK(dev_resolver_->UpdateDeviceAttributes(attributes)); in SetUp()74 std::unique_ptr<DeviceResolverDistributed> dev_resolver_; member in tensorflow::__anon7e53058e0111::DeviceResDistTest79 TF_ASSERT_OK(dev_resolver_->GetDeviceAttributes( in TEST_F()86 EXPECT_TRUE(errors::IsNotFound(dev_resolver_->GetDeviceAttributes( in TEST_F()92 TF_ASSERT_OK(dev_resolver_->GetAllDeviceAttributes( in TEST_F()100 TF_ASSERT_OK(dev_resolver_->GetAllDeviceAttributes( in TEST_F()112 EXPECT_TRUE(errors::IsNotFound(dev_resolver_->GetAllDeviceAttributes( in TEST_F()124 TF_ASSERT_OK(dev_resolver_->UpdateDeviceAttributes(attributes)); in TEST_F()126 TF_ASSERT_OK(dev_resolver_->GetAllDeviceAttributes( in TEST_F()[all …]
132 Status s = dev_resolver_->GetDeviceAttributes(peer_device, in RecvFromPeer()264 s = dev_resolver_->GetAllDeviceAttributes(peer_task, &cached_attrs); in CheckPeerHealth()
52 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
122 s = dev_resolver_->UpdateDeviceAttributes(devices); in CompleteParamsAsync()
32 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()76 DeviceResolverInterface* dev_resolver_; // not owned variable
45 return dev_resolver_.get(); in GetDeviceResolver()70 std::unique_ptr<DeviceResolverInterface> dev_resolver_; variable
35 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()67 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
190 DeviceResolverInterface* dev_resolver_; // Not owned. variable
51 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()