Searched refs:all_reduce_group_map (Results 1 – 1 of 1) sorted by relevance
225 all_reduce_group_map; in CreateComputationGroups() local248 all_reduce_group_map[group_ids].push_back(std::move(instructions)); in CreateComputationGroups()254 for (const auto& all_reduce_group : all_reduce_group_map) { in CreateComputationGroups()