| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 102 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 …]
|
| D | collective.h | 101 string collective_name; member
466 static Status Lookup(const string& collective_name,
474 const string& collective_name,
488 static Status Register(const string& collective_name, Factory factory);
490 static Status LookupHelper(const string& collective_name,
499 CollectiveRegistration(const string& collective_name, in CollectiveRegistration() argument
501 TF_CHECK_OK(CollectiveRegistry::Register(collective_name, factory)); in CollectiveRegistration()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.cc | 51 if (expected_name != col_params->instance.impl_details.collective_name) { in InitializeCollectiveParams()
54 col_params->instance.impl_details.collective_name, in InitializeCollectiveParams()
|
| D | collective_nccl_test.cc | 86 NcclTestBase(CollectiveType collective_type, const string& collective_name) in NcclTestBase() argument
88 collective_name_(collective_name), in NcclTestBase()
141 col_params_->instance.impl_details.collective_name = collective_name_; in Init()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 410 << col_params.instance.impl_details.collective_name; in CreateCollective()
416 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
430 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
440 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
447 col_params.instance.impl_details.collective_name, col_impl); in CreateCollective()
|
| D | collective_test_util.cc | 185 const CollectiveTestEnv& test_env, int rank, const string& collective_name, in CreateCollectiveParams() argument
243 col_params->instance.impl_details.collective_name = collective_name; in CreateCollectiveParams()
347 col_params->instance.impl_details.collective_name, &collective_impl)); in RunCollective()
|
| D | collective_test_util.h | 93 const CollectiveTestEnv& test_env, int rank, const string& collective_name,
|
| D | ring_alg.cc | 134 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
153 col_params->instance.impl_details.collective_name); in GenerateSubdivsInCollectiveParams()
|
| D | ring_gatherer.cc | 47 DCHECK_EQ(col_params->instance.impl_details.collective_name, "RingGather"); in InitializeCollectiveParams()
|
| D | collective_param_resolver_local.cc | 710 cp->instance.impl_details.collective_name = GetCollectiveName(cp, use_nccl); in AssignCollectiveType()
712 << cp->instance.impl_details.collective_name; in AssignCollectiveType()
773 cp->instance.impl_details.collective_name, &col_impl); in CompleteInstanceFromInitializedIRec()
|
| D | ring_reducer.cc | 51 CHECK_EQ(col_params->instance.impl_details.collective_name, "RingReduce"); in InitializeCollectiveParams()
|
| D | hierarchical_tree_broadcaster_test.cc | 357 cp->instance.impl_details.collective_name = "HierarchicalTreeBroadcast"; in TEST_F()
|
| D | hierarchical_tree_broadcaster.cc | 81 CHECK_EQ(col_params->instance.impl_details.collective_name, in InitializeCollectiveParams()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 168 params->instance.impl_details.collective_name = "NcclReduce"; in ResolveDeviceAssignment()
|