Home
last modified time | relevance | path

Searched refs:subdiv_permutations (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dhierarchical_tree_broadcaster.cc118 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs); in InitializeCollectiveParams()
128 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
161 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
248 for (int i = 0; i < impl.subdiv_permutations[subdiv].size(); i++) { in TreeSendTo()
249 if (impl.subdiv_permutations[subdiv][i] >= 0) { in TreeSendTo()
304 for (int r : col_params_->instance.impl_details.subdiv_permutations[si]) { in RunTree()
405 col_params_->instance.impl_details.subdiv_permutations[subdiv][dst_rank]; in DispatchSend()
424 col_params_->instance.impl_details.subdiv_permutations[subdiv][src_rank]; in DispatchRecv()
Dhierarchical_tree_broadcaster_test.cc62 cp.instance.impl_details.subdiv_permutations.push_back( \
268 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
274 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()
279 .subdiv_permutations[subdiv_i][rank] = i; in Init()
286 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()
294 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()
299 << col_params_.instance.impl_details.subdiv_permutations.size(); in Init()
303 col_params_.instance.impl_details.subdiv_permutations[subdiv_i][rank] = in Init()
309 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()
430 cp.instance.impl_details.subdiv_permutations in RunTest()
[all …]
Dring_alg.cc200 << &col_params->instance.impl_details.subdiv_permutations; in InitializeCollectiveParams()
201 col_params->instance.impl_details.subdiv_permutations.resize( in InitializeCollectiveParams()
208 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
329 .subdiv_permutations[subdiv_idx][recv_from_rank]; in InitRingField()
331 .subdiv_permutations[subdiv_idx][send_to_rank]; in InitRingField()
389 .subdiv_permutations[rf->subdiv_idx][send_to_rank]; in DispatchSend()
Dring_gatherer.cc67 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()
77 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()
81 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
Dring_gatherer_test.cc186 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
229 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()
366 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()
372 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()
401 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()
406 .subdiv_permutations[sdi] in DeviceInstance()
410 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
Dring_reducer.cc59 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()
69 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()
73 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
Dring_reducer_test.cc208 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
251 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()
389 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()
395 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()
425 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()
430 .subdiv_permutations[sdi] in DeviceInstance()
434 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
Dcollective_util.cc56 col_params.instance.impl_details.subdiv_permutations; in SubdivPermDebugString()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc74 impl_details.subdiv_permutations.clear(); in operator =()
75 for (auto p : other.impl_details.subdiv_permutations) { in operator =()
76 impl_details.subdiv_permutations.push_back( in operator =()
110 for (const auto& p : impl_details.subdiv_permutations) { in ToString()
Dcollective.h71 std::vector<std::vector<int>> subdiv_permutations; member