Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc95 void operator()(std::string* out, CollGroupMember m) const { in operator ()()
100 Status CheckUserSpecifiedRanks(const std::vector<CollGroupMember> members) { in CheckUserSpecifiedRanks()
224 CollGroupMember member; in CompleteGroupLocal()
452 for (const CollGroupMember& member : gp.members) { in EstablishGlobalRank()
471 for (const CollGroupMember& member : gp->members) { in SetDevPerTask()
490 for (CollGroupMember& member : gr->group.members) { in FinishGroup()
555 [](const CollGroupMember& lhs, const CollGroupMember& rhs) { in CompleteDefaultRanking()
578 std::vector<CollGroupMember> new_members(gp->group_size); in CompleteDefaultRanking()
Dcollective_param_resolver_local_test.cc74 for (const CollGroupMember& member : group.members) { in RunCompleteDefaultRanking()
100 CollGroupMember member; in TEST_F()
Dpermuter_test.cc150 for (const CollGroupMember& member : col_params_->group.members) { in DeviceInstance()
Dhierarchical_tree_broadcaster_test.cc362 CollGroupMember member; in TEST_F()
Dcollective_test_util.cc222 CollGroupMember member; in CreateCollectiveParams()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h62 struct CollGroupMember { struct
81 std::vector<CollGroupMember> members; argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc119 for (const CollGroupMember& m : cp->group.members) { in CompleteParamsAsync()
236 CollGroupMember member; in UpdateGroupCache()
Dcollective_param_resolver_distributed_test.cc275 for (const CollGroupMember& member : cp.group.members) { in ValidateDeviceResolver()
Dworker.cc422 for (const CollGroupMember& member : group_params->members) { in CompleteGroupAsync()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc145 CollGroupMember member; in Init()