| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 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::__anone296589f0111::DeviceResDistTest
79 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 …]
|
| D | collective_rma_distributed.cc | 165 Status s = dev_resolver_->GetDeviceAttributes(peer_device, in RecvFromPeer()
220 s = dev_resolver_->GetAllDeviceAttributes(peer_task, &cached_attrs); in CheckPeerHealth()
|
| D | rpc_collective_executor_mgr.cc | 52 new CollectiveRemoteAccessDistributed(dev_mgr_, dev_resolver_.get(), in Create()
|
| D | collective_param_resolver_distributed.cc | 121 s = dev_resolver_->UpdateDeviceAttributes(attributes); in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.h | 32 dev_resolver_(dev_resolver), in CollectiveRemoteAccessLocal()
76 DeviceResolverInterface* dev_resolver_; // not owned variable
|
| D | collective_executor_mgr.h | 46 return dev_resolver_.get(); in GetDeviceResolver()
71 std::unique_ptr<DeviceResolverInterface> dev_resolver_; variable
|
| D | collective_executor_mgr.cc | 32 dev_resolver_(std::move(dev_resolver)), in CollectiveExecutorMgr()
64 new CollectiveRemoteAccessLocal(dev_mgr_, dev_resolver_.get(), step_id); in Create()
|
| D | permuter_test.cc | 168 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
170 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
433 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::__anonfb3b1ff60111::PermuterTest
|
| D | collective_param_resolver_local.h | 190 DeviceResolverInterface* dev_resolver_; // Not owned. variable
|
| D | ring_gatherer_test.cc | 178 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
180 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
527 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 201 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
203 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
559 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::RingReducerTest
|
| D | hierarchical_tree_broadcaster_test.cc | 261 dev_resolver_ = absl::make_unique<DeviceResolverLocal>(dev_mgr_.get()); in Init()
263 rma_ = new FailTestRMA(dev_mgr_.get(), dev_resolver_.get(), kStepId, in Init()
716 std::unique_ptr<DeviceResolverLocal> dev_resolver_; member in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
|
| D | collective_param_resolver_local.cc | 46 dev_resolver_(dev_resolver), in CollectiveParamResolverLocal()
|