/mnt/ServicesNVME/odysseus/data/local/lib/python3.14/site-packages/torch/distributed/tensor/

5 directories 16 files
Name Size Modified
Go up — —
__pycache__/ —
_ops/ —
debug/ —
experimental/ —
parallel/ —
__init__.py 2.4 KiB
_api.py 65 KiB
_collective_utils.py 21 KiB
_decompositions.py 13 KiB
_dispatch.py 34 KiB
_dtensor_spec.py 33 KiB
_nonlinear_redux.py 10 KiB
_op_schema.py 27 KiB
_random.py 20 KiB
_redistribute.py 82 KiB
_sharding_prop.py 47 KiB
_shards_wrapper.py 13 KiB
_tp_conv.py 11 KiB
_utils.py 20 KiB
device_mesh.py 190 B
placement_types.py 60 KiB