Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/_shard/sharding_spec/
H A 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)
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/
H A Dapi.py17 validate_non_overlapping_shards_metadata,
166 validate_non_overlapping_shards_metadata(self.shards)
H A D_internals.py72 def validate_non_overlapping_shards_metadata(shards: List[ShardMetadata]): function
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dutils.py10 validate_non_overlapping_shards_metadata,
255 validate_non_overlapping_shards_metadata(
H A Dapi.py21 validate_non_overlapping_shards_metadata,
174 validate_non_overlapping_shards_metadata(shards_metadata)
1002 validate_non_overlapping_shards_metadata(shards_metadata)