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.cc120 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.cc67 cp->subdiv_rank = {R}; \
283 col_params_->subdiv_rank.resize(num_subdivs); in Init()
527 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
535 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
577 col_params_->subdiv_rank = parent_->col_params_->subdiv_rank; in DeviceInstance()
587 col_params_->subdiv_rank.resize(num_subdivs); in DeviceInstance()
596 col_params_->subdiv_rank[si] = perm_rank; in DeviceInstance()
599 for (int r : col_params_->subdiv_rank) { in DeviceInstance()
605 col_params_->subdiv_rank[0] == impl.subdiv_source_rank[0]; in DeviceInstance()
Dring_gatherer_test.cc198 col_params_->subdiv_rank.resize(num_subdivs); in Init()
203 col_params_->subdiv_rank[sdi] = sdi * subdiv_stride; in Init()
376 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
383 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
402 col_params_->subdiv_rank = parent_->col_params_->subdiv_rank; in DeviceInstance()
404 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in DeviceInstance()
420 col_params_->subdiv_rank[sdi] = r; in DeviceInstance()
Dring_reducer_test.cc221 col_params_->subdiv_rank.resize(num_subdivs); in Init()
226 col_params_->subdiv_rank[sdi] = sdi * subdiv_stride; in Init()
401 cp->subdiv_rank.clear(); in RunSubdivPermsTest()
408 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank); in RunSubdivPermsTest()
428 col_params_->subdiv_rank = parent_->col_params_->subdiv_rank; in DeviceInstance()
430 int num_subdivs = static_cast<int>(col_params_->subdiv_rank.size()); in DeviceInstance()
446 col_params_->subdiv_rank[sdi] = r; in DeviceInstance()
Dring_alg.cc204 col_params->subdiv_rank.resize( in InitializeCollectiveParams()
232 col_params->subdiv_rank[sdi] = rank; in InitializeCollectiveParams()
324 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.cc105 Tensor alias_chunk(ca_->ChunkAlias(col_params_->subdiv_rank[0])); in Run()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc157 for (const auto& r : subdiv_rank) { in ToString()
Dcollective.h145 std::vector<int> subdiv_rank; member