/data/disk/lighthouse/venv/badge-speaker-benchmark-community/lib/python3.11/site-packages/torch/distributed/_shard/

6 directories 7 files 19 KiB total
List Grid
Name
Size Modified
Up
__pycache__/ —
checkpoint/ —
sharded_optim/ —
sharded_tensor/ —
sharding_plan/ —
sharding_spec/ —
__init__.py
87 B
_utils.py
1.0 KiB
api.py
12 KiB
common_op_utils.py
2.1 KiB
metadata.py
2.2 KiB
op_registry_utils.py
1.0 KiB
sharder.py
901 B