Searched refs:flattened_list (Results 1 – 1 of 1) sorted by relevance
43 std::vector<HloSharding> flattened_list; in Tuple() local44 flattened_list.reserve(sub_shardings.leaf_count()); in Tuple()46 flattened_list.push_back(index_to_sharding.second); in Tuple()48 if (flattened_list.empty()) { in Tuple()54 flattened_list.push_back(sub_shardings.element(ShapeIndex({}))); in Tuple()56 return HloSharding(flattened_list); in Tuple()65 std::vector<HloSharding> flattened_list(shardings.begin(), shardings.end()); in Tuple() local66 CHECK_EQ(flattened_list.size(), RequiredLeaves(tuple_shape)) in Tuple()67 << "Flat list has " << flattened_list.size() << ", required " in Tuple()69 return HloSharding(flattened_list); in Tuple()[all …]