Home
last modified time | relevance | path

Searched refs:validate_non_overlapping_shards_metadata (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/test/distributed/_shard/sharding_spec/
Dtest_sharding_spec.py26 validate_non_overlapping_shards_metadata,
417 validate_non_overlapping_shards_metadata(shards)
432 validate_non_overlapping_shards_metadata(shards)
447 validate_non_overlapping_shards_metadata(shards)
461 validate_non_overlapping_shards_metadata(shards)
476 validate_non_overlapping_shards_metadata(shards)
491 validate_non_overlapping_shards_metadata(shards)
/external/pytorch/torch/distributed/_shard/sharding_spec/
Dapi.py17 validate_non_overlapping_shards_metadata,
166 validate_non_overlapping_shards_metadata(self.shards)
D_internals.py72 def validate_non_overlapping_shards_metadata(shards: List[ShardMetadata]): function
/external/pytorch/torch/distributed/_shard/sharded_tensor/
Dutils.py10 validate_non_overlapping_shards_metadata,
255 validate_non_overlapping_shards_metadata(
Dapi.py21 validate_non_overlapping_shards_metadata,
174 validate_non_overlapping_shards_metadata(shards_metadata)
1002 validate_non_overlapping_shards_metadata(shards_metadata)