Searched refs:num_shards_map (Results 1 – 2 of 2) sorted by relevance
279 split_factor = self._tp_spec.num_shards_map[0]305 num_shards = self._sharding_spec.num_shards_map[0]
173 def num_shards_map(self) -> List[int]: member in DTensorSpec