Home
last modified time | relevance | path

Searched refs:impl_details (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcollective.cc71 impl_details.subdiv_offsets.assign( in operator =()
72 other.impl_details.subdiv_offsets.begin(), in operator =()
73 other.impl_details.subdiv_offsets.end()); in operator =()
74 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 =()
79 impl_details.subdiv_source_rank.assign( in operator =()
80 other.impl_details.subdiv_source_rank.begin(), in operator =()
81 other.impl_details.subdiv_source_rank.end()); in operator =()
82 impl_details.dependencies = other.impl_details.dependencies; in operator =()
[all …]
Dcollective.h98 CollImplDetails impl_details; member
/external/tensorflow/tensorflow/core/common_runtime/
Dring_alg.cc118 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
137 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
142 col_params->instance.impl_details.subdiv_offsets.reserve(num_subdivs); in GenerateSubdivsInCollectiveParams()
146 col_params->instance.impl_details.subdiv_offsets.push_back(subdiv_offset); in GenerateSubdivsInCollectiveParams()
152 col_params->instance.impl_details.subdiv_offsets) { in GenerateSubdivsInCollectiveParams()
193 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
200 << &col_params->instance.impl_details.subdiv_permutations; in InitializeCollectiveParams()
201 col_params->instance.impl_details.subdiv_permutations.resize( in InitializeCollectiveParams()
202 col_params->instance.impl_details.subdiv_offsets.size()); in InitializeCollectiveParams()
204 col_params->instance.impl_details.subdiv_offsets.size(), -1); in InitializeCollectiveParams()
[all …]
Dring_gatherer.cc45 DCHECK_EQ(col_params->instance.impl_details.collective_name, "RingGather"); in InitializeCollectiveParams()
49 if (!col_params->instance.impl_details.subdiv_offsets.empty() && in InitializeCollectiveParams()
50 (col_params->instance.impl_details.subdiv_offsets.size() > 1 || in InitializeCollectiveParams()
51 col_params->instance.impl_details.subdiv_offsets[0] != 0)) { in InitializeCollectiveParams()
55 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
56 col_params->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParams()
67 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()
Dhierarchical_tree_broadcaster.cc79 CHECK_EQ(col_params->instance.impl_details.collective_name, in InitializeCollectiveParams()
118 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs); in InitializeCollectiveParams()
120 col_params->instance.impl_details.subdiv_source_rank.reserve(num_subdivs); in InitializeCollectiveParams()
128 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
150 col_params->instance.impl_details.subdiv_source_rank.push_back(source_task); in InitializeCollectiveParams()
161 col_params->instance.impl_details.subdiv_permutations[sdi]; in InitializeCollectiveParams()
175 col_params->instance.impl_details.subdiv_source_rank.push_back( in InitializeCollectiveParams()
180 CHECK_GE(col_params->instance.impl_details.subdiv_source_rank[sri], 0); in InitializeCollectiveParams()
223 const auto& impl = cp.instance.impl_details; in TreeRecvFrom()
243 const auto& impl = cp.instance.impl_details; in TreeSendTo()
[all …]
Dring_reducer_test.cc204 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
206 col_params_.instance.impl_details.collective_name = "RingReduce"; in Init()
208 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
212 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi * in Init()
249 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) % in Init()
251 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back( in Init()
378 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduce()
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()
[all …]
Dhierarchical_tree_broadcaster_test.cc61 cp.instance.impl_details.subdiv_source_rank = {S}; \
62 cp.instance.impl_details.subdiv_permutations.push_back( \
263 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
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()
278 col_params_.instance.impl_details 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()
[all …]
Dring_gatherer_test.cc182 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
184 col_params_.instance.impl_details.collective_name = "RingGather"; in Init()
186 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs); in Init()
190 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi * in Init()
227 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) % 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()
405 for (int r = 0; r < static_cast<int>(col_params_.instance.impl_details in DeviceInstance()
[all …]
Dcollective_util.cc56 col_params.instance.impl_details.subdiv_permutations; in SubdivPermDebugString()
68 for (auto o : col_params.instance.impl_details.subdiv_offsets) in SubdivPermDebugString()
74 for (auto src : col_params.instance.impl_details.subdiv_source_rank) in SubdivPermDebugString()
Dring_reducer.cc49 CHECK_EQ(col_params->instance.impl_details.collective_name, "RingReduce"); in InitializeCollectiveParams()
59 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()
Dcollective_param_resolver_local.cc614 cp->instance.impl_details.collective_name = "HierarchicalTreeBroadcast"; in AssignCollectiveType()
617 cp->instance.impl_details.collective_name = "NcclReduce"; in AssignCollectiveType()
619 cp->instance.impl_details.collective_name = "RingReduce"; in AssignCollectiveType()
622 cp->instance.impl_details.collective_name = "RingGather"; in AssignCollectiveType()
624 cp->instance.impl_details.collective_name = "undef"; in AssignCollectiveType()
627 << cp->instance.impl_details.collective_name; in AssignCollectiveType()
673 cp->instance.impl_details.collective_name, &col_impl); in CompleteInstanceFromInitializedIRec()
688 cp->instance.impl_details.collective_name == "NcclReduce" && in CompleteInstanceFromInitializedIRec()
Dcollective_param_resolver_local_test.cc173 cp->instance.impl_details.subdiv_offsets.push_back(0); in TEST_F()
196 EXPECT_EQ(cps[i].instance.impl_details.subdiv_source_rank.size(), 0); in TEST_F()
216 cp->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParamsForBroadcast()
Dbase_collective_executor.cc300 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
313 for (int32 instance : col_params.instance.impl_details.dependencies) { in CheckDependencies()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc54 col_params_.instance.impl_details.dependencies = dependencies_; in CanProceedWithCompute()
133 &col_params_.instance.impl_details.subdiv_offsets)); in CollectiveReduceOpKernel()
243 col_params_.instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastSendOpKernel()
312 col_params_.instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastRecvOpKernel()
Dcollective_nccl_reducer.cc33 col_params->instance.impl_details.collective_name != "NcclReduce") { in InitializeCollectiveParams()
37 col_params->instance.impl_details.collective_name, in InitializeCollectiveParams()
Dcollective_nccl_reducer_test.cc125 col_params_.instance.impl_details.collective_name = "NcclReduce"; in Init()
209 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduce()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc62 for (int32 offset : instance.impl_details.subdiv_offsets) { in CompleteInstanceCall()
157 cp->instance.impl_details.subdiv_offsets.push_back(offset); in CompleteInstanceAsync()
Dcollective_param_resolver_distributed_test.cc200 cp.instance.impl_details.subdiv_offsets.push_back(0); in DefineCollectiveParams()