Path Lines of Code torchrec/distributed/__init__.py 20 torchrec/distributed/batched_embedding_kernel.py 613 torchrec/distributed/collective_utils.py 30 torchrec/distributed/comm.py 90 torchrec/distributed/comm_ops.py 616 torchrec/distributed/cw_sharding.py 139 torchrec/distributed/dist_data.py 415 torchrec/distributed/dp_sharding.py 192 torchrec/distributed/embedding.py 75 torchrec/distributed/embedding_kernel.py 275 torchrec/distributed/embedding_lookup.py 415 torchrec/distributed/embedding_sharding.py 376 torchrec/distributed/embedding_types.py 217 torchrec/distributed/embeddingbag.py 676 torchrec/distributed/grouped_position_weighted.py 84 torchrec/distributed/model_parallel.py 319 torchrec/distributed/planner/__init__.py 9 torchrec/distributed/planner/constants.py 39 torchrec/distributed/planner/enumerators.py 213 torchrec/distributed/planner/partitioners.py 265 torchrec/distributed/planner/perf_models.py 11 torchrec/distributed/planner/planners.py 260 torchrec/distributed/planner/proposers.py 111 torchrec/distributed/planner/shard_estimators.py 644 torchrec/distributed/planner/stats.py 209 torchrec/distributed/planner/storage_reservations.py 114 torchrec/distributed/planner/types.py 251 torchrec/distributed/planner/utils.py 11 torchrec/distributed/quant_embedding_kernel.py 159 torchrec/distributed/quant_embeddingbag.py 155 torchrec/distributed/rw_sharding.py 325 torchrec/distributed/train_pipeline.py 359 torchrec/distributed/tw_sharding.py 351 torchrec/distributed/twcw_sharding.py 21 torchrec/distributed/twrw_sharding.py 422 torchrec/distributed/types.py 265 torchrec/distributed/utils.py 77