Searched defs:new_all_reduce_groups (Results 1 – 2 of 2) sorted by relevance
68 std::vector<std::vector<int64_t>> new_all_reduce_groups = { in TEST_F() local108 std::vector<std::vector<int64_t>> new_all_reduce_groups = { in TEST_F() local153 std::vector<std::vector<int64_t>> new_all_reduce_groups = { in TEST_F() local
56 std::vector<ReplicaGroup> new_all_reduce_groups; member98 std::vector<ReplicaGroup> new_all_reduce_groups(scatter_group_size); in TryDecomposeReplicaGroup() local126 std::vector<ReplicaGroup> new_all_reduce_groups; in TryDecomposeReplicaGroups() local