Google Git
Sign in
android / platform / external / pytorch / d1afb7d43d / . / torch / distributed / _spmd
tree: 5bf438bc5d2fcbe7c298aea9eecf9d3ab304abfa [path history] [tgz]
  1. __init__.py
  2. api.py
  3. batch_dim_utils.py
  4. comm_tensor.py
  5. config.py
  6. data_parallel.py
  7. distribute.py
  8. experimental_ops.py
  9. gm_transformation.py
  10. graph_optimization.py
  11. graph_utils.py
  12. iter_graph_module.py
  13. log_utils.py
  14. parallel_mode.py
  15. partial_lower.py