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

  /external/tensorflow/tensorflow/core/common_runtime/
hierarchical_tree_broadcaster.cc 120 col_params->instance.impl_details.subdiv_source_rank.reserve(num_subdivs);
150 col_params->instance.impl_details.subdiv_source_rank.push_back(source_task);
175 col_params->instance.impl_details.subdiv_source_rank.push_back(
180 CHECK_GE(col_params->instance.impl_details.subdiv_source_rank[sri], 0);
224 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size()));
225 int source_rank = impl.subdiv_source_rank[subdiv];
244 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size()));
245 int source_rank = impl.subdiv_source_rank[subdiv];
301 int source_rank = col_params_->instance.impl_details.subdiv_source_rank[si];
collective_util.cc 73 strings::StrAppend(&buf, " subdiv_source_rank: ");
74 for (auto src : col_params.instance.impl_details.subdiv_source_rank)
hierarchical_tree_broadcaster_test.cc 61 cp.instance.impl_details.subdiv_source_rank = {S}; \
431 [0][cp.instance.impl_details.subdiv_source_rank[0]];
523 cp->instance.impl_details.subdiv_source_rank.clear();
531 cp->instance.impl_details.subdiv_source_rank);
584 impl.subdiv_source_rank.resize(num_subdivs, 0);
603 col_params_.subdiv_rank[0] == impl.subdiv_source_rank[0];
    [all...]
collective_param_resolver_local_test.cc 196 EXPECT_EQ(cps[i].instance.impl_details.subdiv_source_rank.size(), 0);
  /external/tensorflow/tensorflow/core/framework/
collective.cc 79 impl_details.subdiv_source_rank.assign(
80 other.impl_details.subdiv_source_rank.begin(),
81 other.impl_details.subdiv_source_rank.end());
117 if (!impl_details.subdiv_source_rank.empty()) {
118 strings::StrAppend(&v, " subdiv_source_rank={");
119 for (const auto& r : impl_details.subdiv_source_rank) {
collective.h 73 std::vector<int> subdiv_source_rank; // rank of source in each subdiv member in struct:tensorflow::CollImplDetails

Completed in 56 milliseconds