Searched refs:subshardings (Results 1 – 1 of 1) sorted by relevance
1417 std::vector<HloSharding> subshardings = sharding.tuple_elements(); in Preprocess() local1418 for (HloSharding& subsharding : subshardings) { in Preprocess()1423 return HloSharding::Tuple(shape, subshardings); in Preprocess()