HomeSort by relevance Sort by last modified time
    Searched refs:subdiv_rank (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
hierarchical_tree_broadcaster.cc 119 col_params->subdiv_rank.reserve(num_subdivs);
124 // does not participate in the subdiv, set subdiv_rank to -1.
146 if (participate) col_params->subdiv_rank.push_back(ti);
149 if (col_params->subdiv_rank.empty()) col_params->subdiv_rank.push_back(-1);
156 // the subdiv, set subdiv_rank to -1;
169 col_params->subdiv_rank.push_back(di);
174 if (!participate) col_params->subdiv_rank.push_back(-1);
219 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size()));
220 int my_rank = cp.subdiv_rank[subdiv]
    [all...]
hierarchical_tree_broadcaster_test.cc 64 cp.subdiv_rank = {R}; \
269 col_params_.subdiv_rank.resize(num_subdivs);
522 cp->subdiv_rank.clear();
529 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank);
575 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank;
585 col_params_.subdiv_rank.resize(num_subdivs);
594 col_params_.subdiv_rank[si] = perm_rank;
597 for (int r : col_params_.subdiv_rank) {
603 col_params_.subdiv_rank[0] == impl.subdiv_source_rank[0]
    [all...]
collective_util.cc 71 for (auto d : col_params.subdiv_rank) strings::StrAppend(&buf, d, " ");
ring_gatherer_test.cc 187 col_params_.subdiv_rank.resize(num_subdivs);
192 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride;
367 cp->subdiv_rank.clear();
373 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank);
393 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank;
395 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size());
411 col_params_.subdiv_rank[sdi] = r;
ring_reducer_test.cc 209 col_params_.subdiv_rank.resize(num_subdivs);
214 col_params_.subdiv_rank[sdi] = sdi * subdiv_stride;
390 cp->subdiv_rank.clear();
396 EXPECT_EQ(expected_subdiv_rank, cp->subdiv_rank);
417 col_params_.subdiv_rank = parent_->col_params_.subdiv_rank;
419 int num_subdivs = static_cast<int>(col_params_.subdiv_rank.size());
435 col_params_.subdiv_rank[sdi] = r;
ring_alg.cc 203 col_params->subdiv_rank.resize(
231 col_params->subdiv_rank[sdi] = rank;
322 rf->rank = col_params_->subdiv_rank[subdiv_idx];
ring_gatherer.cc 101 Tensor alias_chunk(ca_->ChunkAlias(col_params_->subdiv_rank[0]));
  /external/tensorflow/tensorflow/core/framework/
collective.cc 143 " subdiv_rank={");
144 for (const auto& r : subdiv_rank) {
collective.h 121 std::vector<int> subdiv_rank; member in struct:tensorflow::CollectiveParams

Completed in 9070 milliseconds