| 64 | } |
| 65 | |
| 66 | OpSharding Tuple(const ShapeTree<OpSharding>& shardings) { |
| 67 | OpSharding result; |
| 68 | result.set_type(OpSharding::TUPLE); |
| 69 | for (const auto& index_to_sharding : shardings.leaves()) { |
| 70 | *result.add_tuple_shardings() = index_to_sharding.second; |
| 71 | } |
| 72 | return result; |
| 73 | } |
| 74 | |
| 75 | } // namespace sharding_builder |
| 76 | } // namespace xla |