Home
last modified time | relevance | path

Searched +path:pytorch +path:torch +path:distributed +path:tensor (Results 1 – 25 of 51) sorted by relevance

123

/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_utils.py
H A D_shards_wrapper.py
H A Dplacement_types.py
H A D_collective_utils.py
H A D_op_schema.py
H A D_dispatch.py
H A D_sharding_prop.py
H A D__init__.py
H A D_random.py
H A D_api.py
H A D_redistribute.py
H A D_dtensor_spec.py
H A D_tp_conv.py
H A Ddevice_mesh.py
H A DREADME.md
/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_matrix_ops.py
H A D_embedding_ops.py
H A D_random_ops.py
H A Dutils.py
H A D_common_rules.py
H A D__init__.py
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D__init__.py
/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A Dddp.py
/aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/
H A Dtorchrec_sharding_example.py
H A Dconvnext_example.py

123