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_test.cc64 cp.instance.impl_details.subdiv_permutations.push_back( \
274 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
280 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()
285 .subdiv_permutations[subdiv_i][rank] = i; in Init()
292 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()
300 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize( in Init()
305 << col_params_.instance.impl_details.subdiv_permutations.size(); in Init()
309 col_params_.instance.impl_details.subdiv_permutations[subdiv_i][rank] = in Init()
315 col_params_.instance.impl_details.subdiv_permutations[subdiv_i]) in Init()
436 cp.instance.impl_details.subdiv_permutations in RunTest()
[all …]
Dhierarchical_tree_broadcaster.cc120 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs); in InitializeCollectiveParams()
130 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
163 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
250 for (int i = 0; i < impl.subdiv_permutations[subdiv].size(); i++) { in TreeSendTo()
251 if (impl.subdiv_permutations[subdiv][i] >= 0) { in TreeSendTo()
306 for (int r : col_params_->instance.impl_details.subdiv_permutations[si]) { in RunTree()
415 col_params_->instance.impl_details.subdiv_permutations[subdiv][dst_rank]; in DispatchSend()
434 col_params_->instance.impl_details.subdiv_permutations[subdiv][src_rank]; in DispatchRecv()
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.cc70 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()
80 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()
84 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
Dring_gatherer_test.cc192 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
235 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()
373 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()
379 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()
408 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()
413 .subdiv_permutations[sdi] in DeviceInstance()
417 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
Dring_reducer_test.cc214 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
258 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()
397 cp->instance.impl_details.subdiv_permutations.clear(); in RunSubdivPermsTest()
403 cp->instance.impl_details.subdiv_permutations); in RunSubdivPermsTest()
433 col_params_.instance.impl_details.subdiv_permutations[0][rank]; in DeviceInstance()
438 .subdiv_permutations[sdi] in DeviceInstance()
442 col_params_.instance.impl_details.subdiv_permutations[sdi][r]) { in DeviceInstance()
Dring_reducer.cc66 col_params_->instance.impl_details.subdiv_permutations.size()); in Run()
76 sd < col_params_->instance.impl_details.subdiv_permutations.size(); in Run()
80 col_params_->instance.impl_details.subdiv_permutations[sd]) { in Run()
Dcollective_util.cc56 col_params.instance.impl_details.subdiv_permutations; in SubdivPermDebugString()
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc79 impl_details.subdiv_permutations.clear(); in operator =()
80 for (auto p : other.impl_details.subdiv_permutations) { in operator =()
81 impl_details.subdiv_permutations.push_back( in operator =()
115 for (const auto& p : impl_details.subdiv_permutations) { in ToString()
Dcollective.h80 std::vector<std::vector<int>> subdiv_permutations; member