Directory listing for /docker/overlay2/3zz8x9b62tflwpsgpg0x56grl/diff/usr/local/lib/python3.11/site-packages/torch/distributed/_spmd/

• __init__.py
• __pycache__/
• api.py
• batch_dim_utils.py
• comm_tensor.py
• config.py
• data_parallel.py
• distribute.py
• experimental_ops.py
• gm_transformation.py
• graph_optimization.py
• graph_utils.py
• iter_graph_module.py
• log_utils.py
• parallel_mode.py
• partial_lower.py