Home
last modified time | relevance | path

Searched refs:subdiv_rank (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.cc119 col_params->subdiv_rank.reserve(num_subdivs); in InitializeCollectiveParams()
146 if (participate) col_params->subdiv_rank.push_back(ti); in InitializeCollectiveParams()
149 if (col_params->subdiv_rank.empty()) col_params->subdiv_rank.push_back(-1); in InitializeCollectiveParams()
169 col_params->subdiv_rank.push_back(di); in InitializeCollectiveParams()
174 if (!participate) col_params->subdiv_rank.push_back(-1); in InitializeCollectiveParams()
219 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeRecvFrom()
220 int my_rank = cp.subdiv_rank[subdiv]; in TreeRecvFrom()
239 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeSendTo()
240 int my_rank = cp.subdiv_rank[subdiv]; in TreeSendTo()
291 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in RunTree()
[all …]
Dhierarchical_tree_broadcaster_test.cc64 cp.subdiv_rank = {R}; \
269 col_params_.subdiv_rank.resize(num_subdivs); in Init()
522 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
529 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
575 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()
585 col_params_.subdiv_rank.resize(num_subdivs); in DeviceInstance()
594 col_params_.subdiv_rank[si] = perm_rank; in DeviceInstance()
597 for (int r : col_params_.subdiv_rank) { in DeviceInstance()
603 col_params_.subdiv_rank[0] == impl.subdiv_source_rank[0]; in DeviceInstance()
Dring_gatherer_test.cc187 col_params_.subdiv_rank.resize(num_subdivs); in Init()
192 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()
367 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
373 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
393 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()
395 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance()
411 col_params_.subdiv_rank[sdi] = r; in DeviceInstance()
Dring_reducer_test.cc209 col_params_.subdiv_rank.resize(num_subdivs); in Init()
214 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride; in Init()
390 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
396 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
417 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank; in DeviceInstance()
419 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size()); in DeviceInstance()
435 col_params_.subdiv_rank[sdi] = r; in DeviceInstance()
Dring_alg.cc203 col_params->subdiv_rank.resize( in InitializeCollectiveParams()
231 col_params->subdiv_rank[sdi] = rank; in InitializeCollectiveParams()
322 rf->rank = col_params_->subdiv_rank[subdiv_idx]; in InitRingField()
Dcollective_util.cc71 for (auto d : col_params.subdiv_rank) strings::StrAppend(&buf, d, " "); in SubdivPermDebugString()
Dring_gatherer.cc101 Tensor alias_chunk(ca_->ChunkAlias(col_params_->subdiv_rank[0])); in Run()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc144 for (const auto& r : subdiv_rank) { in ToString()
Dcollective.h121 std::vector<int> subdiv_rank; member