Home
last modified time | relevance | path

Searched refs:_infer_sharding_spec_from_shards_metadata (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/_shard/sharding_spec/
H A Dtest_sharding_spec.py14 _infer_sharding_spec_from_shards_metadata,
323 spec = _infer_sharding_spec_from_shards_metadata(shards_metadata)
339 spec = _infer_sharding_spec_from_shards_metadata(shards_metadata)
365 spec = _infer_sharding_spec_from_shards_metadata(shards_metadata)
386 spec = _infer_sharding_spec_from_shards_metadata(shards_metadata)
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/
H A D__init__.py4 _infer_sharding_spec_from_shards_metadata,
H A Dapi.py186 def _infer_sharding_spec_from_shards_metadata(shards_metadata): function
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A Dapi.py159 spec = shard_spec._infer_sharding_spec_from_shards_metadata(shards_metadata)
753 spec = shard_spec._infer_sharding_spec_from_shards_metadata(
926 spec = shard_spec._infer_sharding_spec_from_shards_metadata(shards_metadata)