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

  /external/tensorflow/tensorflow/core/common_runtime/
hierarchical_tree_broadcaster.cc 118 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs);
128 col_params->instance.impl_details.subdiv_permutations[sdi];
161 col_params->instance.impl_details.subdiv_permutations[sdi];
248 for (int i = 0; i < impl.subdiv_permutations[subdiv].size(); i++) {
249 if (impl.subdiv_permutations[subdiv][i] >= 0) {
304 for (int r : col_params_->instance.impl_details.subdiv_permutations[si]) {
405 col_params_->instance.impl_details.subdiv_permutations[subdiv][dst_rank];
424 col_params_->instance.impl_details.subdiv_permutations[subdiv][src_rank];
hierarchical_tree_broadcaster_test.cc 62 cp.instance.impl_details.subdiv_permutations.push_back( \
268 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
274 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize(
279 .subdiv_permutations[subdiv_i][rank] = i;
286 col_params_.instance.impl_details.subdiv_permutations[subdiv_i])
294 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize(
299 << col_params_.instance.impl_details.subdiv_permutations.size();
303 col_params_.instance.impl_details.subdiv_permutations[subdiv_i][rank] =
309 col_params_.instance.impl_details.subdiv_permutations[subdiv_i])
430 cp.instance.impl_details.subdiv_permutations
    [all...]
ring_alg.cc 199 << " subdiv_permutations "
200 << &col_params->instance.impl_details.subdiv_permutations;
201 col_params->instance.impl_details.subdiv_permutations.resize(
208 col_params->instance.impl_details.subdiv_permutations[sdi];
329 .subdiv_permutations[subdiv_idx][recv_from_rank];
331 .subdiv_permutations[subdiv_idx][send_to_rank];
389 .subdiv_permutations[rf->subdiv_idx][send_to_rank];
ring_gatherer.cc 67 col_params_->instance.impl_details.subdiv_permutations.size());
77 sd < col_params_->instance.impl_details.subdiv_permutations.size();
81 col_params_->instance.impl_details.subdiv_permutations[sd]) {
ring_reducer.cc 59 col_params_->instance.impl_details.subdiv_permutations.size());
69 sd < col_params_->instance.impl_details.subdiv_permutations.size();
73 col_params_->instance.impl_details.subdiv_permutations[sd]) {
collective_util.cc 56 col_params.instance.impl_details.subdiv_permutations;
ring_gatherer_test.cc 186 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
229 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back(
366 cp->instance.impl_details.subdiv_permutations.clear();
372 cp->instance.impl_details.subdiv_permutations);
401 col_params_.instance.impl_details.subdiv_permutations[0][rank];
406 .subdiv_permutations[sdi]
410 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) {
ring_reducer_test.cc 208 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
251 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back(
389 cp->instance.impl_details.subdiv_permutations.clear();
395 cp->instance.impl_details.subdiv_permutations);
425 col_params_.instance.impl_details.subdiv_permutations[0][rank];
430 .subdiv_permutations[sdi]
434 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) {
  /external/tensorflow/tensorflow/core/framework/
collective.cc 74 impl_details.subdiv_permutations.clear();
75 for (auto p : other.impl_details.subdiv_permutations) {
76 impl_details.subdiv_permutations.push_back(
110 for (const auto& p : impl_details.subdiv_permutations) {
collective.h 71 std::vector<std::vector<int>> subdiv_permutations; member in struct:tensorflow::CollImplDetails

Completed in 212 milliseconds