Home
last modified time | relevance | path

Searched refs:collective_name (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcollective.cc102 strings::StrAppend(&v, "}, collective_name=", impl_details.collective_name, in ToString()
180 const string& collective_name, in Lookup() argument
182 return LookupHelper(collective_name, implementation, false); in Lookup()
187 const string& collective_name, in LookupParamResolverInstance() argument
189 return LookupHelper(collective_name, implementation, true); in LookupParamResolverInstance()
201 Status CollectiveRegistry::Register(const string& collective_name, in Register() argument
205 if (reg_info.name == collective_name) in Register()
207 collective_name); in Register()
209 registry->emplace_back(collective_name, std::move(factory)); in Register()
215 const string& collective_name, in LookupHelper() argument
[all …]
Dcollective.h70 string collective_name; member
409 static Status Lookup(const string& collective_name,
417 const string& collective_name,
431 static Status Register(const string& collective_name, Factory factory);
433 static Status LookupHelper(const string& collective_name,
442 CollectiveRegistration(const string& collective_name, in CollectiveRegistration() argument
444 TF_CHECK_OK(CollectiveRegistry::Register(collective_name, factory)); in CollectiveRegistration()
/external/tensorflow/tensorflow/core/common_runtime/
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()
Dring_alg.cc118 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
137 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
Dring_gatherer.cc45 DCHECK_EQ(col_params->instance.impl_details.collective_name, "RingGather"); in InitializeCollectiveParams()
Dring_reducer.cc49 CHECK_EQ(col_params->instance.impl_details.collective_name, "RingReduce"); in InitializeCollectiveParams()
Dbase_collective_executor.cc300 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
Dring_gatherer_test.cc184 col_params_.instance.impl_details.collective_name = "RingGather"; in Init()
541 cp.instance.impl_details.collective_name = "RingGather"; in SetUpCollectiveParams()
Dring_reducer_test.cc206 col_params_.instance.impl_details.collective_name = "RingReduce"; in Init()
568 cp.instance.impl_details.collective_name = "RingReduce"; in SetUpCollectiveParams()
Dhierarchical_tree_broadcaster.cc79 CHECK_EQ(col_params->instance.impl_details.collective_name, in InitializeCollectiveParams()
Dhierarchical_tree_broadcaster_test.cc540 cp->instance.impl_details.collective_name = "HierarchicalTreeBroadcast"; in PrepColParamsForSubdivPermsTest()
792 cp.instance.impl_details.collective_name = "HierarchicalTreeBroadcast"; in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
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()