file 1 file 2 # same commits # commits file 1 # commits file 2 torchrec/distributed/embedding_sharding.py torchrec/distributed/dist_data.py 9 12 14 torchrec/distributed/tw_sharding.py torchrec/distributed/rw_sharding.py 8 9 9 torchrec/distributed/types.py torchrec/distributed/embedding_sharding.py 8 10 12 torchrec/distributed/embedding_sharding.py torchrec/distributed/cw_sharding.py 7 12 12 torchrec/distributed/embedding_types.py torchrec/distributed/cw_sharding.py 7 9 12 torchrec/distributed/embeddingbag.py torchrec/distributed/dist_data.py 7 9 14 torchrec/distributed/embeddingbag.py torchrec/distributed/embedding_sharding.py 7 9 12 torchrec/distributed/tw_sharding.py torchrec/distributed/cw_sharding.py 7 9 12 torchrec/distributed/tw_sharding.py torchrec/distributed/embedding_sharding.py 7 9 12 torchrec/distributed/tw_sharding.py torchrec/distributed/embeddingbag.py 7 9 9 torchrec/distributed/twrw_sharding.py torchrec/distributed/embedding_sharding.py 7 9 12 torchrec/distributed/twrw_sharding.py torchrec/distributed/rw_sharding.py 7 9 9 torchrec/distributed/types.py torchrec/distributed/cw_sharding.py 7 10 12 torchrec/distributed/types.py torchrec/distributed/embedding_types.py 7 10 9 torchrec/distributed/comm_ops.py torchrec/distributed/dist_data.py 7 7 14 torchrec/distributed/cw_sharding.py torchrec/distributed/embedding_lookup.py 6 12 16 torchrec/distributed/dist_data.py torchrec/distributed/cw_sharding.py 6 14 12 torchrec/distributed/dp_sharding.py torchrec/distributed/cw_sharding.py 6 7 12 torchrec/distributed/embedding_sharding.py torchrec/distributed/dp_sharding.py 6 12 7 torchrec/distributed/embedding_types.py torchrec/distributed/embedding_sharding.py 6 9 12 torchrec/distributed/embeddingbag.py torchrec/distributed/cw_sharding.py 6 9 12 torchrec/distributed/rw_sharding.py torchrec/distributed/embedding_lookup.py 6 9 16 torchrec/distributed/rw_sharding.py torchrec/distributed/cw_sharding.py 6 9 12 torchrec/distributed/rw_sharding.py torchrec/distributed/embedding_sharding.py 6 9 12 torchrec/distributed/rw_sharding.py torchrec/distributed/embeddingbag.py 6 9 9 torchrec/distributed/tw_sharding.py torchrec/distributed/dist_data.py 6 9 14 torchrec/distributed/tw_sharding.py torchrec/distributed/embedding_types.py 6 9 9 torchrec/distributed/twrw_sharding.py torchrec/distributed/embedding_lookup.py 6 9 16 torchrec/distributed/twrw_sharding.py torchrec/distributed/cw_sharding.py 6 9 12 torchrec/distributed/twrw_sharding.py torchrec/distributed/dist_data.py 6 9 14 torchrec/distributed/twrw_sharding.py torchrec/distributed/dp_sharding.py 6 9 7 torchrec/distributed/twrw_sharding.py torchrec/distributed/tw_sharding.py 6 9 9 torchrec/distributed/types.py torchrec/distributed/dist_data.py 6 10 14 torchrec/distributed/types.py torchrec/distributed/embeddingbag.py 6 10 9 torchrec/distributed/types.py torchrec/distributed/rw_sharding.py 6 10 9 torchrec/distributed/types.py torchrec/distributed/tw_sharding.py 6 10 9 torchrec/distributed/types.py torchrec/distributed/twrw_sharding.py 6 10 9 torchrec/distributed/dp_sharding.py torchrec/distributed/dist_data.py 5 7 14 torchrec/distributed/embedding_sharding.py torchrec/distributed/embedding_lookup.py 5 12 16 torchrec/distributed/embedding_types.py torchrec/distributed/embedding_lookup.py 5 9 16 torchrec/distributed/embeddingbag.py torchrec/distributed/embedding_lookup.py 5 9 16 torchrec/distributed/embeddingbag.py torchrec/distributed/embedding_types.py 5 9 9 torchrec/distributed/rw_sharding.py torchrec/distributed/dist_data.py 5 9 14 torchrec/distributed/rw_sharding.py torchrec/distributed/dp_sharding.py 5 9 7 torchrec/distributed/rw_sharding.py torchrec/distributed/embedding_types.py 5 9 9 torchrec/distributed/tw_sharding.py torchrec/distributed/embedding_lookup.py 5 9 16 torchrec/distributed/tw_sharding.py torchrec/distributed/dp_sharding.py 5 9 7 torchrec/distributed/twrw_sharding.py torchrec/distributed/embeddingbag.py 5 9 9 torchrec/distributed/types.py torchrec/distributed/embedding_lookup.py 5 10 16 torchrec/distributed/types.py torchrec/distributed/dp_sharding.py 5 10 7 torchrec/distributed/types.py torchrec/distributed/model_parallel.py 5 10 7 torchrec/distributed/comm_ops.py torchrec/distributed/embedding_sharding.py 5 7 12 torchrec/distributed/comm_ops.py torchrec/distributed/types.py 5 7 10 torchrec/distributed/train_pipeline.py torchrec/distributed/embedding_sharding.py 5 7 12 torchrec/distributed/train_pipeline.py torchrec/distributed/types.py 5 7 10 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/planner/enumerators.py 5 6 5 torchrec/optim/keyed.py torchrec/distributed/embedding_lookup.py 4 8 16 torchrec/distributed/dist_data.py torchrec/distributed/embedding_lookup.py 4 14 16 torchrec/distributed/dp_sharding.py torchrec/distributed/embedding_lookup.py 4 7 16 torchrec/distributed/embedding_types.py torchrec/distributed/dist_data.py 4 9 14 torchrec/distributed/embedding_types.py torchrec/distributed/dp_sharding.py 4 9 7 torchrec/distributed/embeddingbag.py torchrec/distributed/dp_sharding.py 4 9 7 torchrec/distributed/embeddingbag.py torchrec/optim/keyed.py 4 9 8 torchrec/distributed/model_parallel.py torchrec/distributed/embedding_lookup.py 4 7 16 torchrec/distributed/model_parallel.py torchrec/distributed/cw_sharding.py 4 7 12 torchrec/distributed/model_parallel.py torchrec/distributed/dist_data.py 4 7 14 torchrec/distributed/model_parallel.py torchrec/distributed/embedding_sharding.py 4 7 12 torchrec/distributed/model_parallel.py torchrec/distributed/embedding_types.py 4 7 9 torchrec/distributed/model_parallel.py torchrec/distributed/embeddingbag.py 4 7 9 torchrec/distributed/rw_sharding.py torchrec/distributed/model_parallel.py 4 9 7 torchrec/distributed/tw_sharding.py torchrec/distributed/model_parallel.py 4 9 7 torchrec/distributed/twrw_sharding.py torchrec/distributed/embedding_types.py 4 9 9 torchrec/distributed/twrw_sharding.py torchrec/distributed/model_parallel.py 4 9 7 torchrec/distributed/comm_ops.py torchrec/distributed/cw_sharding.py 4 7 12 torchrec/distributed/comm_ops.py torchrec/distributed/twrw_sharding.py 4 7 9 torchrec/distributed/comm_ops.py torchrec/distributed/embeddingbag.py 4 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/cw_sharding.py 4 7 12 torchrec/distributed/train_pipeline.py torchrec/distributed/dist_data.py 4 7 14 torchrec/distributed/train_pipeline.py torchrec/distributed/twrw_sharding.py 4 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/comm_ops.py 4 7 7 torchrec/distributed/planner/stats.py torchrec/distributed/planner/enumerators.py 4 6 5 torchrec/distributed/planner/stats.py torchrec/distributed/planner/shard_estimators.py 4 6 6 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/shard_estimators.py 4 4 6 torchrec/quant/embedding_modules.py torchrec/distributed/embedding_lookup.py 4 8 16 torchrec/quant/embedding_modules.py torchrec/distributed/dist_data.py 4 8 14 torchrec/quant/embedding_modules.py torchrec/distributed/comm_ops.py 4 8 7 torchrec/distributed/planner/types.py torchrec/distributed/planner/shard_estimators.py 4 6 6 torchrec/distributed/planner/types.py torchrec/distributed/planner/planners.py 4 6 9 torchrec/distributed/planner/utils.py torchrec/distributed/planner/constants.py 3 6 3 torchrec/distributed/grouped_position_weighted.py torchrec/distributed/embedding_lookup.py 3 4 16 torchrec/optim/keyed.py torchrec/distributed/grouped_position_weighted.py 3 8 4 torchrec/distributed/cw_sharding.py torchrec/distributed/grouped_position_weighted.py 3 12 4 torchrec/distributed/cw_sharding.py torchrec/optim/keyed.py 3 12 8 torchrec/distributed/dist_data.py torchrec/distributed/grouped_position_weighted.py 3 14 4 torchrec/distributed/dist_data.py torchrec/optim/keyed.py 3 14 8 torchrec/distributed/embedding_sharding.py torchrec/distributed/grouped_position_weighted.py 3 12 4 torchrec/distributed/embedding_sharding.py torchrec/optim/keyed.py 3 12 8 torchrec/distributed/embeddingbag.py torchrec/distributed/grouped_position_weighted.py 3 9 4 torchrec/distributed/model_parallel.py torchrec/distributed/dp_sharding.py 3 7 7 torchrec/distributed/model_parallel.py torchrec/distributed/planner/utils.py 3 7 6 torchrec/distributed/model_parallel.py torchrec/distributed/grouped_position_weighted.py 3 7 4 torchrec/distributed/model_parallel.py torchrec/optim/keyed.py 3 7 8 torchrec/distributed/rw_sharding.py torchrec/distributed/grouped_position_weighted.py 3 9 4 torchrec/distributed/rw_sharding.py torchrec/optim/keyed.py 3 9 8 torchrec/distributed/tw_sharding.py torchrec/distributed/grouped_position_weighted.py 3 9 4 torchrec/distributed/tw_sharding.py torchrec/optim/keyed.py 3 9 8 torchrec/distributed/twrw_sharding.py torchrec/distributed/grouped_position_weighted.py 3 9 4 torchrec/distributed/twrw_sharding.py torchrec/optim/keyed.py 3 9 8 torchrec/distributed/types.py torchrec/distributed/grouped_position_weighted.py 3 10 4 torchrec/distributed/types.py torchrec/optim/keyed.py 3 10 8 torchrec/distributed/comm_ops.py torchrec/distributed/dp_sharding.py 3 7 7 torchrec/distributed/comm_ops.py torchrec/distributed/embedding_lookup.py 3 7 16 torchrec/distributed/comm_ops.py torchrec/distributed/grouped_position_weighted.py 3 7 4 torchrec/distributed/comm_ops.py torchrec/optim/keyed.py 3 7 8 torchrec/distributed/comm_ops.py torchrec/distributed/model_parallel.py 3 7 7 torchrec/distributed/comm_ops.py torchrec/distributed/rw_sharding.py 3 7 9 torchrec/distributed/comm_ops.py torchrec/distributed/tw_sharding.py 3 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/dp_sharding.py 3 7 7 torchrec/distributed/train_pipeline.py torchrec/distributed/embedding_types.py 3 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/embedding_lookup.py 3 7 16 torchrec/distributed/train_pipeline.py torchrec/distributed/grouped_position_weighted.py 3 7 4 torchrec/distributed/train_pipeline.py torchrec/optim/keyed.py 3 7 8 torchrec/distributed/train_pipeline.py torchrec/distributed/embeddingbag.py 3 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/model_parallel.py 3 7 7 torchrec/distributed/train_pipeline.py torchrec/distributed/rw_sharding.py 3 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/tw_sharding.py 3 7 9 torchrec/distributed/utils.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/distributed/utils.py torchrec/distributed/dist_data.py 3 3 14 torchrec/distributed/utils.py torchrec/distributed/dp_sharding.py 3 3 7 torchrec/distributed/utils.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/distributed/utils.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/distributed/utils.py torchrec/distributed/types.py 3 3 10 torchrec/distributed/utils.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/distributed/utils.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/__init__.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/__init__.py torchrec/distributed/embedding_types.py 3 3 9 torchrec/__init__.py torchrec/distributed/types.py 3 3 10 torchrec/__init__.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/datasets/utils.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/datasets/utils.py torchrec/distributed/embedding_types.py 3 3 9 torchrec/datasets/utils.py torchrec/distributed/types.py 3 3 10 torchrec/datasets/utils.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/datasets/utils.py torchrec/__init__.py 3 3 3 torchrec/sparse/jagged_tensor.py torchrec/distributed/embedding_sharding.py 3 5 12 torchrec/sparse/jagged_tensor.py torchrec/distributed/embedding_types.py 3 5 9 torchrec/sparse/jagged_tensor.py torchrec/distributed/types.py 3 5 10 torchrec/sparse/jagged_tensor.py torchrec/distributed/train_pipeline.py 3 5 7 torchrec/sparse/jagged_tensor.py torchrec/__init__.py 3 5 3 torchrec/sparse/jagged_tensor.py torchrec/datasets/utils.py 3 5 3 torchrec/sparse/jagged_tensor.py torchrec/distributed/dist_data.py 3 5 14 torchrec/sparse/jagged_tensor.py torchrec/distributed/comm_ops.py 3 5 7 torchrec/distributed/planner/partitioners.py torchrec/distributed/planner/enumerators.py 3 3 5 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/planner/partitioners.py 3 6 3 torchrec/distributed/planner/stats.py torchrec/distributed/planner/partitioners.py 3 6 3 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/enumerators.py 3 4 5 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/partitioners.py 3 4 3 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/stats.py 3 4 6 torchrec/distributed/planner/planners.py torchrec/distributed/embedding_lookup.py 3 9 16 torchrec/distributed/planner/planners.py torchrec/distributed/model_parallel.py 3 9 7 torchrec/distributed/planner/planners.py torchrec/distributed/planner/shard_estimators.py 3 9 6 torchrec/distributed/planner/planners.py torchrec/distributed/planner/storage_reservations.py 3 9 4 torchrec/quant/embedding_modules.py torchrec/distributed/grouped_position_weighted.py 3 8 4 torchrec/quant/embedding_modules.py torchrec/optim/keyed.py 3 8 8 torchrec/quant/embedding_modules.py torchrec/distributed/cw_sharding.py 3 8 12 torchrec/quant/embedding_modules.py torchrec/distributed/embedding_sharding.py 3 8 12 torchrec/quant/embedding_modules.py torchrec/distributed/embeddingbag.py 3 8 9 torchrec/quant/embedding_modules.py torchrec/distributed/model_parallel.py 3 8 7 torchrec/quant/embedding_modules.py torchrec/distributed/rw_sharding.py 3 8 9 torchrec/quant/embedding_modules.py torchrec/distributed/tw_sharding.py 3 8 9 torchrec/quant/embedding_modules.py torchrec/distributed/twrw_sharding.py 3 8 9 torchrec/quant/embedding_modules.py torchrec/distributed/types.py 3 8 10 torchrec/quant/embedding_modules.py torchrec/distributed/train_pipeline.py 3 8 7 torchrec/quant/embedding_modules.py torchrec/sparse/jagged_tensor.py 3 8 5 torchrec/datasets/random.py torchrec/datasets/criteo.py 3 3 4 torchrec/datasets/scripts/preprocess_criteo.py torchrec/datasets/criteo.py 3 3 4 torchrec/distributed/collective_utils.py torchrec/distributed/embedding_lookup.py 3 3 16 torchrec/distributed/collective_utils.py torchrec/distributed/grouped_position_weighted.py 3 3 4 torchrec/distributed/collective_utils.py torchrec/optim/keyed.py 3 3 8 torchrec/distributed/collective_utils.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/distributed/collective_utils.py torchrec/distributed/dist_data.py 3 3 14 torchrec/distributed/collective_utils.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/distributed/collective_utils.py torchrec/distributed/embeddingbag.py 3 3 9 torchrec/distributed/collective_utils.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/distributed/collective_utils.py torchrec/distributed/rw_sharding.py 3 3 9 torchrec/distributed/collective_utils.py torchrec/distributed/tw_sharding.py 3 3 9 torchrec/distributed/collective_utils.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/distributed/collective_utils.py torchrec/distributed/types.py 3 3 10 torchrec/distributed/collective_utils.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/distributed/collective_utils.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/distributed/collective_utils.py torchrec/quant/embedding_modules.py 3 3 8 torchrec/distributed/comm.py torchrec/distributed/embedding_lookup.py 3 3 16 torchrec/distributed/comm.py torchrec/distributed/grouped_position_weighted.py 3 3 4 torchrec/distributed/comm.py torchrec/optim/keyed.py 3 3 8 torchrec/distributed/comm.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/distributed/comm.py torchrec/distributed/dist_data.py 3 3 14 torchrec/distributed/comm.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/distributed/comm.py torchrec/distributed/embeddingbag.py 3 3 9 torchrec/distributed/comm.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/distributed/comm.py torchrec/distributed/rw_sharding.py 3 3 9 torchrec/distributed/comm.py torchrec/distributed/tw_sharding.py 3 3 9 torchrec/distributed/comm.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/distributed/comm.py torchrec/distributed/types.py 3 3 10 torchrec/distributed/comm.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/distributed/comm.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/distributed/comm.py torchrec/quant/embedding_modules.py 3 3 8 torchrec/distributed/comm.py torchrec/distributed/collective_utils.py 3 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/utils.py 3 3 6 torchrec/distributed/planner/__init__.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/planners.py 3 3 9 torchrec/distributed/planner/types.py torchrec/distributed/planner/utils.py 3 6 6 torchrec/distributed/planner/types.py torchrec/distributed/model_parallel.py 3 6 7 torchrec/distributed/planner/types.py torchrec/distributed/planner/enumerators.py 3 6 5 torchrec/distributed/planner/types.py torchrec/distributed/planner/stats.py 3 6 6 torchrec/distributed/planner/types.py torchrec/distributed/planner/storage_reservations.py 3 6 4 torchrec/distributed/planner/types.py torchrec/distributed/planner/__init__.py 3 6 3 torchrec/models/dlrm.py torchrec/models/deepfm.py 3 4 3 torchrec/modules/activation.py torchrec/models/dlrm.py 3 3 4 torchrec/modules/crossnet.py torchrec/models/dlrm.py 3 6 4 torchrec/modules/crossnet.py torchrec/modules/__init__.py 3 6 3 torchrec/modules/crossnet.py torchrec/modules/activation.py 3 6 3 torchrec/modules/deepfm.py torchrec/models/dlrm.py 3 3 4 torchrec/modules/deepfm.py torchrec/modules/activation.py 3 3 3 torchrec/modules/deepfm.py torchrec/modules/crossnet.py 3 3 6 torchrec/modules/embedding_modules.py torchrec/distributed/embedding_lookup.py 3 4 16 torchrec/modules/embedding_modules.py torchrec/distributed/grouped_position_weighted.py 3 4 4 torchrec/modules/embedding_modules.py torchrec/optim/keyed.py 3 4 8 torchrec/modules/embedding_modules.py torchrec/distributed/cw_sharding.py 3 4 12 torchrec/modules/embedding_modules.py torchrec/distributed/dist_data.py 3 4 14 torchrec/modules/embedding_modules.py torchrec/distributed/embedding_sharding.py 3 4 12 torchrec/modules/embedding_modules.py torchrec/distributed/embeddingbag.py 3 4 9 torchrec/modules/embedding_modules.py torchrec/distributed/model_parallel.py 3 4 7 torchrec/modules/embedding_modules.py torchrec/distributed/rw_sharding.py 3 4 9 torchrec/modules/embedding_modules.py torchrec/distributed/tw_sharding.py 3 4 9 torchrec/modules/embedding_modules.py torchrec/distributed/twrw_sharding.py 3 4 9 torchrec/modules/embedding_modules.py torchrec/distributed/types.py 3 4 10 torchrec/modules/embedding_modules.py torchrec/distributed/comm_ops.py 3 4 7 torchrec/modules/embedding_modules.py torchrec/distributed/train_pipeline.py 3 4 7 torchrec/modules/embedding_modules.py torchrec/quant/embedding_modules.py 3 4 8 torchrec/modules/embedding_modules.py torchrec/distributed/collective_utils.py 3 4 3 torchrec/modules/embedding_modules.py torchrec/distributed/comm.py 3 4 3 torchrec/modules/feature_processor.py torchrec/distributed/embedding_lookup.py 3 3 16 torchrec/modules/feature_processor.py torchrec/distributed/grouped_position_weighted.py 3 3 4 torchrec/modules/feature_processor.py torchrec/optim/keyed.py 3 3 8 torchrec/modules/feature_processor.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/modules/feature_processor.py torchrec/distributed/dist_data.py 3 3 14 torchrec/modules/feature_processor.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/modules/feature_processor.py torchrec/distributed/embeddingbag.py 3 3 9 torchrec/modules/feature_processor.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/modules/feature_processor.py torchrec/distributed/rw_sharding.py 3 3 9 torchrec/modules/feature_processor.py torchrec/distributed/tw_sharding.py 3 3 9 torchrec/modules/feature_processor.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/modules/feature_processor.py torchrec/distributed/types.py 3 3 10 torchrec/modules/feature_processor.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/modules/feature_processor.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/modules/feature_processor.py torchrec/quant/embedding_modules.py 3 3 8 torchrec/modules/feature_processor.py torchrec/distributed/collective_utils.py 3 3 3 torchrec/modules/feature_processor.py torchrec/distributed/comm.py 3 3 3 torchrec/modules/feature_processor.py torchrec/modules/embedding_modules.py 3 3 4 torchrec/modules/score_learning.py torchrec/distributed/embedding_lookup.py 3 3 16 torchrec/modules/score_learning.py torchrec/distributed/grouped_position_weighted.py 3 3 4 torchrec/modules/score_learning.py torchrec/optim/keyed.py 3 3 8 torchrec/modules/score_learning.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/modules/score_learning.py torchrec/distributed/dist_data.py 3 3 14 torchrec/modules/score_learning.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/modules/score_learning.py torchrec/distributed/embeddingbag.py 3 3 9 torchrec/modules/score_learning.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/modules/score_learning.py torchrec/distributed/rw_sharding.py 3 3 9 torchrec/modules/score_learning.py torchrec/distributed/tw_sharding.py 3 3 9 torchrec/modules/score_learning.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/modules/score_learning.py torchrec/distributed/types.py 3 3 10 torchrec/modules/score_learning.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/modules/score_learning.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/modules/score_learning.py torchrec/quant/embedding_modules.py 3 3 8 torchrec/modules/score_learning.py torchrec/distributed/collective_utils.py 3 3 3 torchrec/modules/score_learning.py torchrec/distributed/comm.py 3 3 3 torchrec/modules/score_learning.py torchrec/modules/embedding_modules.py 3 3 4 torchrec/modules/score_learning.py torchrec/modules/feature_processor.py 3 3 3 torchrec/modules/utils.py torchrec/distributed/embedding_lookup.py 3 3 16 torchrec/modules/utils.py torchrec/distributed/grouped_position_weighted.py 3 3 4 torchrec/modules/utils.py torchrec/optim/keyed.py 3 3 8 torchrec/modules/utils.py torchrec/distributed/cw_sharding.py 3 3 12 torchrec/modules/utils.py torchrec/distributed/dist_data.py 3 3 14 torchrec/modules/utils.py torchrec/distributed/embedding_sharding.py 3 3 12 torchrec/modules/utils.py torchrec/distributed/embeddingbag.py 3 3 9 torchrec/modules/utils.py torchrec/distributed/model_parallel.py 3 3 7 torchrec/modules/utils.py torchrec/distributed/rw_sharding.py 3 3 9 torchrec/modules/utils.py torchrec/distributed/tw_sharding.py 3 3 9 torchrec/modules/utils.py torchrec/distributed/twrw_sharding.py 3 3 9 torchrec/modules/utils.py torchrec/distributed/types.py 3 3 10 torchrec/modules/utils.py torchrec/distributed/comm_ops.py 3 3 7 torchrec/modules/utils.py torchrec/distributed/train_pipeline.py 3 3 7 torchrec/modules/utils.py torchrec/quant/embedding_modules.py 3 3 8 torchrec/modules/utils.py torchrec/distributed/collective_utils.py 3 3 3 torchrec/modules/utils.py torchrec/distributed/comm.py 3 3 3 torchrec/modules/utils.py torchrec/modules/embedding_modules.py 3 3 4 torchrec/modules/utils.py torchrec/modules/feature_processor.py 3 3 3 torchrec/modules/utils.py torchrec/modules/score_learning.py 3 3 3 torchrec/optim/__init__.py torchrec/models/dlrm.py 3 3 4 torchrec/optim/__init__.py torchrec/modules/activation.py 3 3 3 torchrec/optim/__init__.py torchrec/modules/crossnet.py 3 3 6 torchrec/optim/__init__.py torchrec/modules/deepfm.py 3 3 3 torchrec/distributed/planner/utils.py torchrec/distributed/embedding_lookup.py 2 6 16 torchrec/distributed/grouped_position_weighted.py torchrec/distributed/planner/utils.py 2 4 6 torchrec/optim/keyed.py torchrec/distributed/planner/utils.py 2 8 6 torchrec/distributed/cw_sharding.py torchrec/distributed/planner/utils.py 2 12 6 torchrec/distributed/dist_data.py torchrec/distributed/planner/utils.py 2 14 6 torchrec/distributed/dp_sharding.py torchrec/distributed/planner/utils.py 2 7 6 torchrec/distributed/dp_sharding.py torchrec/distributed/grouped_position_weighted.py 2 7 4 torchrec/distributed/dp_sharding.py torchrec/optim/keyed.py 2 7 8 torchrec/distributed/embedding_sharding.py torchrec/distributed/planner/utils.py 2 12 6 torchrec/distributed/embedding_types.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/embedding_types.py torchrec/distributed/grouped_position_weighted.py 2 9 4 torchrec/distributed/embedding_types.py torchrec/optim/keyed.py 2 9 8 torchrec/distributed/embeddingbag.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/model_parallel.py torchrec/distributed/planner/constants.py 2 7 3 torchrec/distributed/rw_sharding.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/tw_sharding.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/twrw_sharding.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/types.py torchrec/distributed/planner/utils.py 2 10 6 torchrec/distributed/comm_ops.py torchrec/distributed/planner/utils.py 2 7 6 torchrec/distributed/comm_ops.py torchrec/distributed/embedding_types.py 2 7 9 torchrec/distributed/train_pipeline.py torchrec/distributed/planner/utils.py 2 7 6 torchrec/distributed/utils.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/distributed/utils.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/distributed/utils.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/distributed/utils.py torchrec/optim/keyed.py 2 3 8 torchrec/distributed/utils.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/distributed/utils.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/distributed/utils.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/distributed/utils.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/distributed/utils.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/__init__.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/__init__.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/__init__.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/__init__.py torchrec/optim/keyed.py 2 3 8 torchrec/__init__.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/__init__.py torchrec/distributed/dist_data.py 2 3 14 torchrec/__init__.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/__init__.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/__init__.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/__init__.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/__init__.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/__init__.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/__init__.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/__init__.py torchrec/distributed/utils.py 2 3 3 torchrec/datasets/utils.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/datasets/utils.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/datasets/utils.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/datasets/utils.py torchrec/optim/keyed.py 2 3 8 torchrec/datasets/utils.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/datasets/utils.py torchrec/distributed/dist_data.py 2 3 14 torchrec/datasets/utils.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/datasets/utils.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/datasets/utils.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/datasets/utils.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/datasets/utils.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/datasets/utils.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/datasets/utils.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/datasets/utils.py torchrec/distributed/utils.py 2 3 3 torchrec/sparse/jagged_tensor.py torchrec/distributed/embedding_lookup.py 2 5 16 torchrec/sparse/jagged_tensor.py torchrec/distributed/planner/utils.py 2 5 6 torchrec/sparse/jagged_tensor.py torchrec/distributed/grouped_position_weighted.py 2 5 4 torchrec/sparse/jagged_tensor.py torchrec/optim/keyed.py 2 5 8 torchrec/sparse/jagged_tensor.py torchrec/distributed/cw_sharding.py 2 5 12 torchrec/sparse/jagged_tensor.py torchrec/distributed/dp_sharding.py 2 5 7 torchrec/sparse/jagged_tensor.py torchrec/distributed/embeddingbag.py 2 5 9 torchrec/sparse/jagged_tensor.py torchrec/distributed/model_parallel.py 2 5 7 torchrec/sparse/jagged_tensor.py torchrec/distributed/rw_sharding.py 2 5 9 torchrec/sparse/jagged_tensor.py torchrec/distributed/tw_sharding.py 2 5 9 torchrec/sparse/jagged_tensor.py torchrec/distributed/twrw_sharding.py 2 5 9 torchrec/sparse/jagged_tensor.py torchrec/distributed/utils.py 2 5 3 torchrec/distributed/planner/enumerators.py torchrec/distributed/cw_sharding.py 2 5 12 torchrec/distributed/planner/enumerators.py torchrec/distributed/embedding_types.py 2 5 9 torchrec/distributed/planner/enumerators.py torchrec/distributed/embeddingbag.py 2 5 9 torchrec/distributed/planner/enumerators.py torchrec/distributed/rw_sharding.py 2 5 9 torchrec/distributed/planner/enumerators.py torchrec/distributed/tw_sharding.py 2 5 9 torchrec/distributed/planner/enumerators.py torchrec/distributed/types.py 2 5 10 torchrec/distributed/planner/enumerators.py torchrec/distributed/planner/constants.py 2 5 3 torchrec/distributed/planner/enumerators.py torchrec/distributed/planner/utils.py 2 5 6 torchrec/distributed/planner/enumerators.py torchrec/distributed/model_parallel.py 2 5 7 torchrec/distributed/planner/partitioners.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/distributed/planner/partitioners.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/distributed/planner/partitioners.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/distributed/planner/partitioners.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/distributed/planner/partitioners.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/distributed/planner/partitioners.py torchrec/distributed/types.py 2 3 10 torchrec/distributed/planner/partitioners.py torchrec/distributed/planner/constants.py 2 3 3 torchrec/distributed/planner/partitioners.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/distributed/planner/partitioners.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/cw_sharding.py 2 6 12 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/embedding_types.py 2 6 9 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/embeddingbag.py 2 6 9 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/rw_sharding.py 2 6 9 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/tw_sharding.py 2 6 9 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/types.py 2 6 10 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/planner/constants.py 2 6 3 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/planner/utils.py 2 6 6 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/model_parallel.py 2 6 7 torchrec/distributed/planner/stats.py torchrec/distributed/cw_sharding.py 2 6 12 torchrec/distributed/planner/stats.py torchrec/distributed/embedding_types.py 2 6 9 torchrec/distributed/planner/stats.py torchrec/distributed/embeddingbag.py 2 6 9 torchrec/distributed/planner/stats.py torchrec/distributed/rw_sharding.py 2 6 9 torchrec/distributed/planner/stats.py torchrec/distributed/tw_sharding.py 2 6 9 torchrec/distributed/planner/stats.py torchrec/distributed/types.py 2 6 10 torchrec/distributed/planner/stats.py torchrec/distributed/planner/constants.py 2 6 3 torchrec/distributed/planner/stats.py torchrec/distributed/planner/utils.py 2 6 6 torchrec/distributed/planner/stats.py torchrec/distributed/model_parallel.py 2 6 7 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/cw_sharding.py 2 4 12 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/embedding_types.py 2 4 9 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/embeddingbag.py 2 4 9 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/rw_sharding.py 2 4 9 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/tw_sharding.py 2 4 9 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/types.py 2 4 10 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/constants.py 2 4 3 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/planner/utils.py 2 4 6 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/model_parallel.py 2 4 7 torchrec/distributed/planner/planners.py torchrec/distributed/planner/constants.py 2 9 3 torchrec/distributed/planner/planners.py torchrec/distributed/planner/utils.py 2 9 6 torchrec/distributed/planner/planners.py torchrec/distributed/grouped_position_weighted.py 2 9 4 torchrec/distributed/planner/planners.py torchrec/optim/keyed.py 2 9 8 torchrec/distributed/planner/planners.py torchrec/distributed/cw_sharding.py 2 9 12 torchrec/distributed/planner/planners.py torchrec/distributed/dist_data.py 2 9 14 torchrec/distributed/planner/planners.py torchrec/distributed/embedding_sharding.py 2 9 12 torchrec/distributed/planner/planners.py torchrec/distributed/embeddingbag.py 2 9 9 torchrec/distributed/planner/planners.py torchrec/distributed/rw_sharding.py 2 9 9 torchrec/distributed/planner/planners.py torchrec/distributed/tw_sharding.py 2 9 9 torchrec/distributed/planner/planners.py torchrec/distributed/twrw_sharding.py 2 9 9 torchrec/distributed/planner/planners.py torchrec/distributed/types.py 2 9 10 torchrec/distributed/planner/planners.py torchrec/distributed/comm_ops.py 2 9 7 torchrec/distributed/planner/planners.py torchrec/distributed/train_pipeline.py 2 9 7 torchrec/distributed/planner/planners.py torchrec/distributed/planner/enumerators.py 2 9 5 torchrec/distributed/planner/planners.py torchrec/distributed/planner/partitioners.py 2 9 3 torchrec/distributed/planner/planners.py torchrec/distributed/planner/stats.py 2 9 6 torchrec/quant/embedding_modules.py torchrec/distributed/planner/utils.py 2 8 6 torchrec/quant/embedding_modules.py torchrec/distributed/dp_sharding.py 2 8 7 torchrec/quant/embedding_modules.py torchrec/distributed/embedding_types.py 2 8 9 torchrec/quant/embedding_modules.py torchrec/distributed/utils.py 2 8 3 torchrec/quant/embedding_modules.py torchrec/__init__.py 2 8 3 torchrec/quant/embedding_modules.py torchrec/datasets/utils.py 2 8 3 torchrec/quant/embedding_modules.py torchrec/distributed/planner/planners.py 2 8 9 setup.py torchrec/distributed/embedding_lookup.py 2 4 16 setup.py torchrec/distributed/planner/utils.py 2 4 6 setup.py torchrec/distributed/grouped_position_weighted.py 2 4 4 setup.py torchrec/optim/keyed.py 2 4 8 setup.py torchrec/distributed/cw_sharding.py 2 4 12 setup.py torchrec/distributed/dist_data.py 2 4 14 setup.py torchrec/distributed/dp_sharding.py 2 4 7 setup.py torchrec/distributed/embedding_sharding.py 2 4 12 setup.py torchrec/distributed/embedding_types.py 2 4 9 setup.py torchrec/distributed/embeddingbag.py 2 4 9 setup.py torchrec/distributed/model_parallel.py 2 4 7 setup.py torchrec/distributed/rw_sharding.py 2 4 9 setup.py torchrec/distributed/tw_sharding.py 2 4 9 setup.py torchrec/distributed/twrw_sharding.py 2 4 9 setup.py torchrec/distributed/types.py 2 4 10 setup.py torchrec/distributed/comm_ops.py 2 4 7 setup.py torchrec/distributed/train_pipeline.py 2 4 7 setup.py torchrec/distributed/utils.py 2 4 3 setup.py torchrec/__init__.py 2 4 3 setup.py torchrec/datasets/utils.py 2 4 3 setup.py torchrec/sparse/jagged_tensor.py 2 4 5 setup.py torchrec/quant/embedding_modules.py 2 4 8 torchrec/datasets/__init__.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/datasets/__init__.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/datasets/__init__.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/datasets/__init__.py torchrec/optim/keyed.py 2 2 8 torchrec/datasets/__init__.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/datasets/__init__.py torchrec/distributed/dist_data.py 2 2 14 torchrec/datasets/__init__.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/datasets/__init__.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/datasets/__init__.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/datasets/__init__.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/datasets/__init__.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/datasets/__init__.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/datasets/__init__.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/datasets/__init__.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/datasets/__init__.py torchrec/distributed/types.py 2 2 10 torchrec/datasets/__init__.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/datasets/__init__.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/datasets/__init__.py torchrec/distributed/utils.py 2 2 3 torchrec/datasets/__init__.py torchrec/__init__.py 2 2 3 torchrec/datasets/__init__.py torchrec/datasets/utils.py 2 2 3 torchrec/datasets/__init__.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/datasets/__init__.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/datasets/__init__.py setup.py 2 2 4 torchrec/datasets/criteo.py torchrec/distributed/embedding_lookup.py 2 4 16 torchrec/datasets/criteo.py torchrec/distributed/planner/utils.py 2 4 6 torchrec/datasets/criteo.py torchrec/distributed/grouped_position_weighted.py 2 4 4 torchrec/datasets/criteo.py torchrec/optim/keyed.py 2 4 8 torchrec/datasets/criteo.py torchrec/distributed/cw_sharding.py 2 4 12 torchrec/datasets/criteo.py torchrec/distributed/dist_data.py 2 4 14 torchrec/datasets/criteo.py torchrec/distributed/dp_sharding.py 2 4 7 torchrec/datasets/criteo.py torchrec/distributed/embedding_sharding.py 2 4 12 torchrec/datasets/criteo.py torchrec/distributed/embedding_types.py 2 4 9 torchrec/datasets/criteo.py torchrec/distributed/embeddingbag.py 2 4 9 torchrec/datasets/criteo.py torchrec/distributed/model_parallel.py 2 4 7 torchrec/datasets/criteo.py torchrec/distributed/rw_sharding.py 2 4 9 torchrec/datasets/criteo.py torchrec/distributed/tw_sharding.py 2 4 9 torchrec/datasets/criteo.py torchrec/distributed/twrw_sharding.py 2 4 9 torchrec/datasets/criteo.py torchrec/distributed/types.py 2 4 10 torchrec/datasets/criteo.py torchrec/distributed/comm_ops.py 2 4 7 torchrec/datasets/criteo.py torchrec/distributed/train_pipeline.py 2 4 7 torchrec/datasets/criteo.py torchrec/distributed/utils.py 2 4 3 torchrec/datasets/criteo.py torchrec/__init__.py 2 4 3 torchrec/datasets/criteo.py torchrec/datasets/utils.py 2 4 3 torchrec/datasets/criteo.py torchrec/sparse/jagged_tensor.py 2 4 5 torchrec/datasets/criteo.py torchrec/quant/embedding_modules.py 2 4 8 torchrec/datasets/criteo.py setup.py 2 4 4 torchrec/datasets/criteo.py torchrec/datasets/__init__.py 2 4 2 torchrec/datasets/movielens.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/datasets/movielens.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/datasets/movielens.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/datasets/movielens.py torchrec/optim/keyed.py 2 2 8 torchrec/datasets/movielens.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/datasets/movielens.py torchrec/distributed/dist_data.py 2 2 14 torchrec/datasets/movielens.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/datasets/movielens.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/datasets/movielens.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/datasets/movielens.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/datasets/movielens.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/datasets/movielens.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/datasets/movielens.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/datasets/movielens.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/datasets/movielens.py torchrec/distributed/types.py 2 2 10 torchrec/datasets/movielens.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/datasets/movielens.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/datasets/movielens.py torchrec/distributed/utils.py 2 2 3 torchrec/datasets/movielens.py torchrec/__init__.py 2 2 3 torchrec/datasets/movielens.py torchrec/datasets/utils.py 2 2 3 torchrec/datasets/movielens.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/datasets/movielens.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/datasets/movielens.py setup.py 2 2 4 torchrec/datasets/movielens.py torchrec/datasets/__init__.py 2 2 2 torchrec/datasets/movielens.py torchrec/datasets/criteo.py 2 2 4 torchrec/datasets/random.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/datasets/random.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/datasets/random.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/datasets/random.py torchrec/optim/keyed.py 2 3 8 torchrec/datasets/random.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/datasets/random.py torchrec/distributed/dist_data.py 2 3 14 torchrec/datasets/random.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/datasets/random.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/datasets/random.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/datasets/random.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/datasets/random.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/datasets/random.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/datasets/random.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/datasets/random.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/datasets/random.py torchrec/distributed/types.py 2 3 10 torchrec/datasets/random.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/datasets/random.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/datasets/random.py torchrec/distributed/utils.py 2 3 3 torchrec/datasets/random.py torchrec/__init__.py 2 3 3 torchrec/datasets/random.py torchrec/datasets/utils.py 2 3 3 torchrec/datasets/random.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/datasets/random.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/datasets/random.py setup.py 2 3 4 torchrec/datasets/random.py torchrec/datasets/__init__.py 2 3 2 torchrec/datasets/random.py torchrec/datasets/movielens.py 2 3 2 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/datasets/scripts/preprocess_criteo.py torchrec/optim/keyed.py 2 3 8 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/dist_data.py 2 3 14 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/types.py 2 3 10 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/utils.py 2 3 3 torchrec/datasets/scripts/preprocess_criteo.py torchrec/__init__.py 2 3 3 torchrec/datasets/scripts/preprocess_criteo.py torchrec/datasets/utils.py 2 3 3 torchrec/datasets/scripts/preprocess_criteo.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/datasets/scripts/preprocess_criteo.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/datasets/scripts/preprocess_criteo.py setup.py 2 3 4 torchrec/datasets/scripts/preprocess_criteo.py torchrec/datasets/__init__.py 2 3 2 torchrec/datasets/scripts/preprocess_criteo.py torchrec/datasets/movielens.py 2 3 2 torchrec/datasets/scripts/preprocess_criteo.py torchrec/datasets/random.py 2 3 3 torchrec/distributed/__init__.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/distributed/__init__.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/distributed/__init__.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/distributed/__init__.py torchrec/optim/keyed.py 2 2 8 torchrec/distributed/__init__.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/distributed/__init__.py torchrec/distributed/dist_data.py 2 2 14 torchrec/distributed/__init__.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/distributed/__init__.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/distributed/__init__.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/distributed/__init__.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/distributed/__init__.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/distributed/__init__.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/distributed/__init__.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/distributed/__init__.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/distributed/__init__.py torchrec/distributed/types.py 2 2 10 torchrec/distributed/__init__.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/distributed/__init__.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/distributed/__init__.py torchrec/distributed/utils.py 2 2 3 torchrec/distributed/__init__.py torchrec/__init__.py 2 2 3 torchrec/distributed/__init__.py torchrec/datasets/utils.py 2 2 3 torchrec/distributed/__init__.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/distributed/__init__.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/distributed/__init__.py setup.py 2 2 4 torchrec/distributed/__init__.py torchrec/datasets/__init__.py 2 2 2 torchrec/distributed/__init__.py torchrec/datasets/criteo.py 2 2 4 torchrec/distributed/__init__.py torchrec/datasets/movielens.py 2 2 2 torchrec/distributed/__init__.py torchrec/datasets/random.py 2 2 3 torchrec/distributed/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/distributed/collective_utils.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/distributed/collective_utils.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/distributed/collective_utils.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/distributed/collective_utils.py torchrec/distributed/utils.py 2 3 3 torchrec/distributed/collective_utils.py torchrec/__init__.py 2 3 3 torchrec/distributed/collective_utils.py torchrec/datasets/utils.py 2 3 3 torchrec/distributed/collective_utils.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/distributed/collective_utils.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/distributed/collective_utils.py setup.py 2 3 4 torchrec/distributed/collective_utils.py torchrec/datasets/__init__.py 2 3 2 torchrec/distributed/collective_utils.py torchrec/datasets/criteo.py 2 3 4 torchrec/distributed/collective_utils.py torchrec/datasets/movielens.py 2 3 2 torchrec/distributed/collective_utils.py torchrec/datasets/random.py 2 3 3 torchrec/distributed/collective_utils.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/distributed/collective_utils.py torchrec/distributed/__init__.py 2 3 2 torchrec/distributed/comm.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/distributed/comm.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/distributed/comm.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/distributed/comm.py torchrec/distributed/utils.py 2 3 3 torchrec/distributed/comm.py torchrec/__init__.py 2 3 3 torchrec/distributed/comm.py torchrec/datasets/utils.py 2 3 3 torchrec/distributed/comm.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/distributed/comm.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/distributed/comm.py setup.py 2 3 4 torchrec/distributed/comm.py torchrec/datasets/__init__.py 2 3 2 torchrec/distributed/comm.py torchrec/datasets/criteo.py 2 3 4 torchrec/distributed/comm.py torchrec/datasets/movielens.py 2 3 2 torchrec/distributed/comm.py torchrec/datasets/random.py 2 3 3 torchrec/distributed/comm.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/distributed/comm.py torchrec/distributed/__init__.py 2 3 2 torchrec/distributed/embedding.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/distributed/embedding.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/distributed/embedding.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/distributed/embedding.py torchrec/optim/keyed.py 2 2 8 torchrec/distributed/embedding.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/distributed/embedding.py torchrec/distributed/dist_data.py 2 2 14 torchrec/distributed/embedding.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/distributed/embedding.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/distributed/embedding.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/distributed/embedding.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/distributed/embedding.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/distributed/embedding.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/distributed/embedding.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/distributed/embedding.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/distributed/embedding.py torchrec/distributed/types.py 2 2 10 torchrec/distributed/embedding.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/distributed/embedding.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/distributed/embedding.py torchrec/distributed/utils.py 2 2 3 torchrec/distributed/embedding.py torchrec/__init__.py 2 2 3 torchrec/distributed/embedding.py torchrec/datasets/utils.py 2 2 3 torchrec/distributed/embedding.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/distributed/embedding.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/distributed/embedding.py setup.py 2 2 4 torchrec/distributed/embedding.py torchrec/datasets/__init__.py 2 2 2 torchrec/distributed/embedding.py torchrec/datasets/criteo.py 2 2 4 torchrec/distributed/embedding.py torchrec/datasets/movielens.py 2 2 2 torchrec/distributed/embedding.py torchrec/datasets/random.py 2 2 3 torchrec/distributed/embedding.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/distributed/embedding.py torchrec/distributed/__init__.py 2 2 2 torchrec/distributed/embedding.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/distributed/embedding.py torchrec/distributed/comm.py 2 2 3 torchrec/distributed/planner/__init__.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/constants.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/distributed/planner/__init__.py torchrec/optim/keyed.py 2 3 8 torchrec/distributed/planner/__init__.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/distributed/planner/__init__.py torchrec/distributed/dist_data.py 2 3 14 torchrec/distributed/planner/__init__.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/distributed/planner/__init__.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/distributed/planner/__init__.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/distributed/types.py 2 3 10 torchrec/distributed/planner/__init__.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/distributed/planner/__init__.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/distributed/planner/__init__.py torchrec/distributed/utils.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/__init__.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/datasets/utils.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/enumerators.py 2 3 5 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/partitioners.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/shard_estimators.py 2 3 6 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/stats.py 2 3 6 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/storage_reservations.py 2 3 4 torchrec/distributed/planner/__init__.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/distributed/planner/__init__.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/distributed/planner/__init__.py setup.py 2 3 4 torchrec/distributed/planner/__init__.py torchrec/datasets/__init__.py 2 3 2 torchrec/distributed/planner/__init__.py torchrec/datasets/criteo.py 2 3 4 torchrec/distributed/planner/__init__.py torchrec/datasets/movielens.py 2 3 2 torchrec/distributed/planner/__init__.py torchrec/datasets/random.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/__init__.py 2 3 2 torchrec/distributed/planner/__init__.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/comm.py 2 3 3 torchrec/distributed/planner/__init__.py torchrec/distributed/embedding.py 2 3 2 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/constants.py 2 2 3 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/distributed/planner/perf_models.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/enumerators.py 2 2 5 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/partitioners.py 2 2 3 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/shard_estimators.py 2 2 6 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/stats.py 2 2 6 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/storage_reservations.py 2 2 4 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/planners.py 2 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/constants.py 2 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/distributed/planner/proposers.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/enumerators.py 2 3 5 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/partitioners.py 2 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/shard_estimators.py 2 3 6 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/stats.py 2 3 6 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/storage_reservations.py 2 3 4 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/planner/perf_models.py 2 3 2 torchrec/distributed/planner/types.py torchrec/distributed/embedding_lookup.py 2 6 16 torchrec/distributed/planner/types.py torchrec/distributed/planner/constants.py 2 6 3 torchrec/distributed/planner/types.py torchrec/distributed/grouped_position_weighted.py 2 6 4 torchrec/distributed/planner/types.py torchrec/optim/keyed.py 2 6 8 torchrec/distributed/planner/types.py torchrec/distributed/cw_sharding.py 2 6 12 torchrec/distributed/planner/types.py torchrec/distributed/dist_data.py 2 6 14 torchrec/distributed/planner/types.py torchrec/distributed/dp_sharding.py 2 6 7 torchrec/distributed/planner/types.py torchrec/distributed/embedding_sharding.py 2 6 12 torchrec/distributed/planner/types.py torchrec/distributed/embedding_types.py 2 6 9 torchrec/distributed/planner/types.py torchrec/distributed/embeddingbag.py 2 6 9 torchrec/distributed/planner/types.py torchrec/distributed/rw_sharding.py 2 6 9 torchrec/distributed/planner/types.py torchrec/distributed/tw_sharding.py 2 6 9 torchrec/distributed/planner/types.py torchrec/distributed/twrw_sharding.py 2 6 9 torchrec/distributed/planner/types.py torchrec/distributed/types.py 2 6 10 torchrec/distributed/planner/types.py torchrec/distributed/comm_ops.py 2 6 7 torchrec/distributed/planner/types.py torchrec/distributed/train_pipeline.py 2 6 7 torchrec/distributed/planner/types.py torchrec/distributed/utils.py 2 6 3 torchrec/distributed/planner/types.py torchrec/__init__.py 2 6 3 torchrec/distributed/planner/types.py torchrec/datasets/utils.py 2 6 3 torchrec/distributed/planner/types.py torchrec/sparse/jagged_tensor.py 2 6 5 torchrec/distributed/planner/types.py torchrec/distributed/planner/partitioners.py 2 6 3 torchrec/distributed/planner/types.py torchrec/quant/embedding_modules.py 2 6 8 torchrec/distributed/planner/types.py setup.py 2 6 4 torchrec/distributed/planner/types.py torchrec/datasets/__init__.py 2 6 2 torchrec/distributed/planner/types.py torchrec/datasets/criteo.py 2 6 4 torchrec/distributed/planner/types.py torchrec/datasets/movielens.py 2 6 2 torchrec/distributed/planner/types.py torchrec/datasets/random.py 2 6 3 torchrec/distributed/planner/types.py torchrec/datasets/scripts/preprocess_criteo.py 2 6 3 torchrec/distributed/planner/types.py torchrec/distributed/__init__.py 2 6 2 torchrec/distributed/planner/types.py torchrec/distributed/collective_utils.py 2 6 3 torchrec/distributed/planner/types.py torchrec/distributed/comm.py 2 6 3 torchrec/distributed/planner/types.py torchrec/distributed/embedding.py 2 6 2 torchrec/distributed/planner/types.py torchrec/distributed/planner/perf_models.py 2 6 2 torchrec/distributed/planner/types.py torchrec/distributed/planner/proposers.py 2 6 3 torchrec/fx/__init__.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/fx/__init__.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/fx/__init__.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/fx/__init__.py torchrec/optim/keyed.py 2 2 8 torchrec/fx/__init__.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/fx/__init__.py torchrec/distributed/dist_data.py 2 2 14 torchrec/fx/__init__.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/fx/__init__.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/fx/__init__.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/fx/__init__.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/fx/__init__.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/fx/__init__.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/fx/__init__.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/fx/__init__.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/fx/__init__.py torchrec/distributed/types.py 2 2 10 torchrec/fx/__init__.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/fx/__init__.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/fx/__init__.py torchrec/distributed/utils.py 2 2 3 torchrec/fx/__init__.py torchrec/__init__.py 2 2 3 torchrec/fx/__init__.py torchrec/datasets/utils.py 2 2 3 torchrec/fx/__init__.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/fx/__init__.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/fx/__init__.py setup.py 2 2 4 torchrec/fx/__init__.py torchrec/datasets/__init__.py 2 2 2 torchrec/fx/__init__.py torchrec/datasets/criteo.py 2 2 4 torchrec/fx/__init__.py torchrec/datasets/movielens.py 2 2 2 torchrec/fx/__init__.py torchrec/datasets/random.py 2 2 3 torchrec/fx/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/fx/__init__.py torchrec/distributed/__init__.py 2 2 2 torchrec/fx/__init__.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/fx/__init__.py torchrec/distributed/comm.py 2 2 3 torchrec/fx/__init__.py torchrec/distributed/embedding.py 2 2 2 torchrec/fx/__init__.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/fx/__init__.py torchrec/distributed/planner/types.py 2 2 6 torchrec/fx/tracer.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/fx/tracer.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/fx/tracer.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/fx/tracer.py torchrec/optim/keyed.py 2 3 8 torchrec/fx/tracer.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/fx/tracer.py torchrec/distributed/dist_data.py 2 3 14 torchrec/fx/tracer.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/fx/tracer.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/fx/tracer.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/fx/tracer.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/fx/tracer.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/fx/tracer.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/fx/tracer.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/fx/tracer.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/fx/tracer.py torchrec/distributed/types.py 2 3 10 torchrec/fx/tracer.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/fx/tracer.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/fx/tracer.py torchrec/distributed/utils.py 2 3 3 torchrec/fx/tracer.py torchrec/__init__.py 2 3 3 torchrec/fx/tracer.py torchrec/datasets/utils.py 2 3 3 torchrec/fx/tracer.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/fx/tracer.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/fx/tracer.py setup.py 2 3 4 torchrec/fx/tracer.py torchrec/datasets/__init__.py 2 3 2 torchrec/fx/tracer.py torchrec/datasets/criteo.py 2 3 4 torchrec/fx/tracer.py torchrec/datasets/movielens.py 2 3 2 torchrec/fx/tracer.py torchrec/datasets/random.py 2 3 3 torchrec/fx/tracer.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/fx/tracer.py torchrec/distributed/__init__.py 2 3 2 torchrec/fx/tracer.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/fx/tracer.py torchrec/distributed/comm.py 2 3 3 torchrec/fx/tracer.py torchrec/distributed/embedding.py 2 3 2 torchrec/fx/tracer.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/fx/tracer.py torchrec/distributed/planner/types.py 2 3 6 torchrec/fx/tracer.py torchrec/fx/__init__.py 2 3 2 torchrec/linter/module_linter.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/linter/module_linter.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/linter/module_linter.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/linter/module_linter.py torchrec/optim/keyed.py 2 2 8 torchrec/linter/module_linter.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/linter/module_linter.py torchrec/distributed/dist_data.py 2 2 14 torchrec/linter/module_linter.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/linter/module_linter.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/linter/module_linter.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/linter/module_linter.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/linter/module_linter.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/linter/module_linter.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/linter/module_linter.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/linter/module_linter.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/linter/module_linter.py torchrec/distributed/types.py 2 2 10 torchrec/linter/module_linter.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/linter/module_linter.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/linter/module_linter.py torchrec/distributed/utils.py 2 2 3 torchrec/linter/module_linter.py torchrec/__init__.py 2 2 3 torchrec/linter/module_linter.py torchrec/datasets/utils.py 2 2 3 torchrec/linter/module_linter.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/linter/module_linter.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/linter/module_linter.py setup.py 2 2 4 torchrec/linter/module_linter.py torchrec/datasets/__init__.py 2 2 2 torchrec/linter/module_linter.py torchrec/datasets/criteo.py 2 2 4 torchrec/linter/module_linter.py torchrec/datasets/movielens.py 2 2 2 torchrec/linter/module_linter.py torchrec/datasets/random.py 2 2 3 torchrec/linter/module_linter.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/linter/module_linter.py torchrec/distributed/__init__.py 2 2 2 torchrec/linter/module_linter.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/linter/module_linter.py torchrec/distributed/comm.py 2 2 3 torchrec/linter/module_linter.py torchrec/distributed/embedding.py 2 2 2 torchrec/linter/module_linter.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/linter/module_linter.py torchrec/distributed/planner/types.py 2 2 6 torchrec/linter/module_linter.py torchrec/fx/__init__.py 2 2 2 torchrec/linter/module_linter.py torchrec/fx/tracer.py 2 2 3 torchrec/models/deepfm.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/models/deepfm.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/models/deepfm.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/models/deepfm.py torchrec/optim/keyed.py 2 3 8 torchrec/models/deepfm.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/models/deepfm.py torchrec/distributed/dist_data.py 2 3 14 torchrec/models/deepfm.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/models/deepfm.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/models/deepfm.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/models/deepfm.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/models/deepfm.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/models/deepfm.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/models/deepfm.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/models/deepfm.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/models/deepfm.py torchrec/distributed/types.py 2 3 10 torchrec/models/deepfm.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/models/deepfm.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/models/deepfm.py torchrec/distributed/utils.py 2 3 3 torchrec/models/deepfm.py torchrec/__init__.py 2 3 3 torchrec/models/deepfm.py torchrec/datasets/utils.py 2 3 3 torchrec/models/deepfm.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/models/deepfm.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/models/deepfm.py setup.py 2 3 4 torchrec/models/deepfm.py torchrec/datasets/__init__.py 2 3 2 torchrec/models/deepfm.py torchrec/datasets/criteo.py 2 3 4 torchrec/models/deepfm.py torchrec/datasets/movielens.py 2 3 2 torchrec/models/deepfm.py torchrec/datasets/random.py 2 3 3 torchrec/models/deepfm.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/models/deepfm.py torchrec/distributed/__init__.py 2 3 2 torchrec/models/deepfm.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/models/deepfm.py torchrec/distributed/comm.py 2 3 3 torchrec/models/deepfm.py torchrec/distributed/embedding.py 2 3 2 torchrec/models/deepfm.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/models/deepfm.py torchrec/distributed/planner/types.py 2 3 6 torchrec/models/deepfm.py torchrec/fx/__init__.py 2 3 2 torchrec/models/deepfm.py torchrec/fx/tracer.py 2 3 3 torchrec/models/deepfm.py torchrec/linter/module_linter.py 2 3 2 torchrec/models/dlrm.py torchrec/distributed/embedding_lookup.py 2 4 16 torchrec/models/dlrm.py torchrec/distributed/planner/utils.py 2 4 6 torchrec/models/dlrm.py torchrec/distributed/grouped_position_weighted.py 2 4 4 torchrec/models/dlrm.py torchrec/optim/keyed.py 2 4 8 torchrec/models/dlrm.py torchrec/distributed/cw_sharding.py 2 4 12 torchrec/models/dlrm.py torchrec/distributed/dist_data.py 2 4 14 torchrec/models/dlrm.py torchrec/distributed/dp_sharding.py 2 4 7 torchrec/models/dlrm.py torchrec/distributed/embedding_sharding.py 2 4 12 torchrec/models/dlrm.py torchrec/distributed/embedding_types.py 2 4 9 torchrec/models/dlrm.py torchrec/distributed/embeddingbag.py 2 4 9 torchrec/models/dlrm.py torchrec/distributed/model_parallel.py 2 4 7 torchrec/models/dlrm.py torchrec/distributed/rw_sharding.py 2 4 9 torchrec/models/dlrm.py torchrec/distributed/tw_sharding.py 2 4 9 torchrec/models/dlrm.py torchrec/distributed/twrw_sharding.py 2 4 9 torchrec/models/dlrm.py torchrec/distributed/types.py 2 4 10 torchrec/models/dlrm.py torchrec/distributed/comm_ops.py 2 4 7 torchrec/models/dlrm.py torchrec/distributed/train_pipeline.py 2 4 7 torchrec/models/dlrm.py torchrec/distributed/utils.py 2 4 3 torchrec/models/dlrm.py torchrec/__init__.py 2 4 3 torchrec/models/dlrm.py torchrec/datasets/utils.py 2 4 3 torchrec/models/dlrm.py torchrec/sparse/jagged_tensor.py 2 4 5 torchrec/models/dlrm.py torchrec/quant/embedding_modules.py 2 4 8 torchrec/models/dlrm.py setup.py 2 4 4 torchrec/models/dlrm.py torchrec/datasets/__init__.py 2 4 2 torchrec/models/dlrm.py torchrec/datasets/criteo.py 2 4 4 torchrec/models/dlrm.py torchrec/datasets/movielens.py 2 4 2 torchrec/models/dlrm.py torchrec/datasets/random.py 2 4 3 torchrec/models/dlrm.py torchrec/datasets/scripts/preprocess_criteo.py 2 4 3 torchrec/models/dlrm.py torchrec/distributed/__init__.py 2 4 2 torchrec/models/dlrm.py torchrec/distributed/collective_utils.py 2 4 3 torchrec/models/dlrm.py torchrec/distributed/comm.py 2 4 3 torchrec/models/dlrm.py torchrec/distributed/embedding.py 2 4 2 torchrec/models/dlrm.py torchrec/distributed/planner/__init__.py 2 4 3 torchrec/models/dlrm.py torchrec/distributed/planner/types.py 2 4 6 torchrec/models/dlrm.py torchrec/fx/__init__.py 2 4 2 torchrec/models/dlrm.py torchrec/fx/tracer.py 2 4 3 torchrec/models/dlrm.py torchrec/linter/module_linter.py 2 4 2 torchrec/models/dlrm.py torchrec/models/__init__.py 2 4 2 torchrec/modules/__init__.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/modules/__init__.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/__init__.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/modules/__init__.py torchrec/optim/keyed.py 2 3 8 torchrec/modules/__init__.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/modules/__init__.py torchrec/distributed/dist_data.py 2 3 14 torchrec/modules/__init__.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/__init__.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/modules/__init__.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/__init__.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/modules/__init__.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/modules/__init__.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/modules/__init__.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/modules/__init__.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/modules/__init__.py torchrec/distributed/types.py 2 3 10 torchrec/modules/__init__.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/modules/__init__.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/modules/__init__.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/__init__.py torchrec/__init__.py 2 3 3 torchrec/modules/__init__.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/__init__.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/__init__.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/modules/__init__.py setup.py 2 3 4 torchrec/modules/__init__.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/__init__.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/__init__.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/__init__.py torchrec/datasets/random.py 2 3 3 torchrec/modules/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/__init__.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/__init__.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/modules/__init__.py torchrec/distributed/comm.py 2 3 3 torchrec/modules/__init__.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/__init__.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/__init__.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/__init__.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/__init__.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/__init__.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/__init__.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/__init__.py torchrec/models/dlrm.py 2 3 4 torchrec/modules/activation.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/modules/activation.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/activation.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/modules/activation.py torchrec/optim/keyed.py 2 3 8 torchrec/modules/activation.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/modules/activation.py torchrec/distributed/dist_data.py 2 3 14 torchrec/modules/activation.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/activation.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/modules/activation.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/activation.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/modules/activation.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/modules/activation.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/modules/activation.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/modules/activation.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/modules/activation.py torchrec/distributed/types.py 2 3 10 torchrec/modules/activation.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/modules/activation.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/modules/activation.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/activation.py torchrec/__init__.py 2 3 3 torchrec/modules/activation.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/activation.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/activation.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/modules/activation.py setup.py 2 3 4 torchrec/modules/activation.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/activation.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/activation.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/activation.py torchrec/datasets/random.py 2 3 3 torchrec/modules/activation.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/activation.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/activation.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/modules/activation.py torchrec/distributed/comm.py 2 3 3 torchrec/modules/activation.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/activation.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/activation.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/activation.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/activation.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/activation.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/activation.py torchrec/models/__init__.py 2 3 2 torchrec/modules/activation.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/activation.py torchrec/modules/__init__.py 2 3 3 torchrec/modules/crossnet.py torchrec/distributed/embedding_lookup.py 2 6 16 torchrec/modules/crossnet.py torchrec/distributed/planner/utils.py 2 6 6 torchrec/modules/crossnet.py torchrec/distributed/grouped_position_weighted.py 2 6 4 torchrec/modules/crossnet.py torchrec/optim/keyed.py 2 6 8 torchrec/modules/crossnet.py torchrec/distributed/cw_sharding.py 2 6 12 torchrec/modules/crossnet.py torchrec/distributed/dist_data.py 2 6 14 torchrec/modules/crossnet.py torchrec/distributed/dp_sharding.py 2 6 7 torchrec/modules/crossnet.py torchrec/distributed/embedding_sharding.py 2 6 12 torchrec/modules/crossnet.py torchrec/distributed/embedding_types.py 2 6 9 torchrec/modules/crossnet.py torchrec/distributed/embeddingbag.py 2 6 9 torchrec/modules/crossnet.py torchrec/distributed/model_parallel.py 2 6 7 torchrec/modules/crossnet.py torchrec/distributed/rw_sharding.py 2 6 9 torchrec/modules/crossnet.py torchrec/distributed/tw_sharding.py 2 6 9 torchrec/modules/crossnet.py torchrec/distributed/twrw_sharding.py 2 6 9 torchrec/modules/crossnet.py torchrec/distributed/types.py 2 6 10 torchrec/modules/crossnet.py torchrec/distributed/comm_ops.py 2 6 7 torchrec/modules/crossnet.py torchrec/distributed/train_pipeline.py 2 6 7 torchrec/modules/crossnet.py torchrec/distributed/utils.py 2 6 3 torchrec/modules/crossnet.py torchrec/__init__.py 2 6 3 torchrec/modules/crossnet.py torchrec/datasets/utils.py 2 6 3 torchrec/modules/crossnet.py torchrec/sparse/jagged_tensor.py 2 6 5 torchrec/modules/crossnet.py torchrec/quant/embedding_modules.py 2 6 8 torchrec/modules/crossnet.py setup.py 2 6 4 torchrec/modules/crossnet.py torchrec/datasets/__init__.py 2 6 2 torchrec/modules/crossnet.py torchrec/datasets/criteo.py 2 6 4 torchrec/modules/crossnet.py torchrec/datasets/movielens.py 2 6 2 torchrec/modules/crossnet.py torchrec/datasets/random.py 2 6 3 torchrec/modules/crossnet.py torchrec/datasets/scripts/preprocess_criteo.py 2 6 3 torchrec/modules/crossnet.py torchrec/distributed/__init__.py 2 6 2 torchrec/modules/crossnet.py torchrec/distributed/collective_utils.py 2 6 3 torchrec/modules/crossnet.py torchrec/distributed/comm.py 2 6 3 torchrec/modules/crossnet.py torchrec/distributed/embedding.py 2 6 2 torchrec/modules/crossnet.py torchrec/distributed/planner/__init__.py 2 6 3 torchrec/modules/crossnet.py torchrec/distributed/planner/types.py 2 6 6 torchrec/modules/crossnet.py torchrec/fx/__init__.py 2 6 2 torchrec/modules/crossnet.py torchrec/fx/tracer.py 2 6 3 torchrec/modules/crossnet.py torchrec/linter/module_linter.py 2 6 2 torchrec/modules/crossnet.py torchrec/models/__init__.py 2 6 2 torchrec/modules/crossnet.py torchrec/models/deepfm.py 2 6 3 torchrec/modules/deepfm.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/modules/deepfm.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/deepfm.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/modules/deepfm.py torchrec/optim/keyed.py 2 3 8 torchrec/modules/deepfm.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/modules/deepfm.py torchrec/distributed/dist_data.py 2 3 14 torchrec/modules/deepfm.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/deepfm.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/modules/deepfm.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/deepfm.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/modules/deepfm.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/modules/deepfm.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/modules/deepfm.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/modules/deepfm.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/modules/deepfm.py torchrec/distributed/types.py 2 3 10 torchrec/modules/deepfm.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/modules/deepfm.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/modules/deepfm.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/deepfm.py torchrec/__init__.py 2 3 3 torchrec/modules/deepfm.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/deepfm.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/deepfm.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/modules/deepfm.py setup.py 2 3 4 torchrec/modules/deepfm.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/deepfm.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/deepfm.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/deepfm.py torchrec/datasets/random.py 2 3 3 torchrec/modules/deepfm.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/deepfm.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/deepfm.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/modules/deepfm.py torchrec/distributed/comm.py 2 3 3 torchrec/modules/deepfm.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/deepfm.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/deepfm.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/deepfm.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/deepfm.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/deepfm.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/deepfm.py torchrec/models/__init__.py 2 3 2 torchrec/modules/deepfm.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/deepfm.py torchrec/modules/__init__.py 2 3 3 torchrec/modules/embedding_configs.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/modules/embedding_configs.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/modules/embedding_configs.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/modules/embedding_configs.py torchrec/optim/keyed.py 2 2 8 torchrec/modules/embedding_configs.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/modules/embedding_configs.py torchrec/distributed/dist_data.py 2 2 14 torchrec/modules/embedding_configs.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/modules/embedding_configs.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/modules/embedding_configs.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/modules/embedding_configs.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/types.py 2 2 10 torchrec/modules/embedding_configs.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/modules/embedding_configs.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/modules/embedding_configs.py torchrec/distributed/utils.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/__init__.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/datasets/utils.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/modules/embedding_configs.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/modules/embedding_configs.py setup.py 2 2 4 torchrec/modules/embedding_configs.py torchrec/datasets/__init__.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/datasets/criteo.py 2 2 4 torchrec/modules/embedding_configs.py torchrec/datasets/movielens.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/datasets/random.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/__init__.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/comm.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/embedding.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/planner/types.py 2 2 6 torchrec/modules/embedding_configs.py torchrec/fx/__init__.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/fx/tracer.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/linter/module_linter.py 2 2 2 torchrec/modules/embedding_configs.py torchrec/models/deepfm.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/models/dlrm.py 2 2 4 torchrec/modules/embedding_configs.py torchrec/modules/__init__.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/modules/activation.py 2 2 3 torchrec/modules/embedding_configs.py torchrec/modules/crossnet.py 2 2 6 torchrec/modules/embedding_configs.py torchrec/modules/deepfm.py 2 2 3 torchrec/modules/embedding_modules.py torchrec/distributed/planner/utils.py 2 4 6 torchrec/modules/embedding_modules.py torchrec/distributed/dp_sharding.py 2 4 7 torchrec/modules/embedding_modules.py torchrec/distributed/embedding_types.py 2 4 9 torchrec/modules/embedding_modules.py torchrec/distributed/utils.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/__init__.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/datasets/utils.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/sparse/jagged_tensor.py 2 4 5 torchrec/modules/embedding_modules.py torchrec/distributed/planner/planners.py 2 4 9 torchrec/modules/embedding_modules.py setup.py 2 4 4 torchrec/modules/embedding_modules.py torchrec/datasets/__init__.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/datasets/criteo.py 2 4 4 torchrec/modules/embedding_modules.py torchrec/datasets/movielens.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/datasets/random.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/datasets/scripts/preprocess_criteo.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/distributed/__init__.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/distributed/embedding.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/distributed/planner/__init__.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/distributed/planner/types.py 2 4 6 torchrec/modules/embedding_modules.py torchrec/fx/__init__.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/fx/tracer.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/linter/module_linter.py 2 4 2 torchrec/modules/embedding_modules.py torchrec/models/deepfm.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/models/dlrm.py 2 4 4 torchrec/modules/embedding_modules.py torchrec/modules/__init__.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/modules/activation.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/modules/crossnet.py 2 4 6 torchrec/modules/embedding_modules.py torchrec/modules/deepfm.py 2 4 3 torchrec/modules/embedding_modules.py torchrec/modules/embedding_configs.py 2 4 2 torchrec/modules/feature_processor.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/feature_processor.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/feature_processor.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/feature_processor.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/feature_processor.py torchrec/__init__.py 2 3 3 torchrec/modules/feature_processor.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/feature_processor.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/feature_processor.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/modules/feature_processor.py setup.py 2 3 4 torchrec/modules/feature_processor.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/feature_processor.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/feature_processor.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/feature_processor.py torchrec/datasets/random.py 2 3 3 torchrec/modules/feature_processor.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/feature_processor.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/feature_processor.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/feature_processor.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/feature_processor.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/feature_processor.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/feature_processor.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/feature_processor.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/feature_processor.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/feature_processor.py torchrec/models/dlrm.py 2 3 4 torchrec/modules/feature_processor.py torchrec/modules/__init__.py 2 3 3 torchrec/modules/feature_processor.py torchrec/modules/activation.py 2 3 3 torchrec/modules/feature_processor.py torchrec/modules/crossnet.py 2 3 6 torchrec/modules/feature_processor.py torchrec/modules/deepfm.py 2 3 3 torchrec/modules/feature_processor.py torchrec/modules/embedding_configs.py 2 3 2 torchrec/modules/lazy_extension.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/modules/lazy_extension.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/modules/lazy_extension.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/modules/lazy_extension.py torchrec/optim/keyed.py 2 2 8 torchrec/modules/lazy_extension.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/modules/lazy_extension.py torchrec/distributed/dist_data.py 2 2 14 torchrec/modules/lazy_extension.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/modules/lazy_extension.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/modules/lazy_extension.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/modules/lazy_extension.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/types.py 2 2 10 torchrec/modules/lazy_extension.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/modules/lazy_extension.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/modules/lazy_extension.py torchrec/distributed/utils.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/__init__.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/datasets/utils.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/modules/lazy_extension.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/modules/lazy_extension.py setup.py 2 2 4 torchrec/modules/lazy_extension.py torchrec/datasets/__init__.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/datasets/criteo.py 2 2 4 torchrec/modules/lazy_extension.py torchrec/datasets/movielens.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/datasets/random.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/__init__.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/comm.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/embedding.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/planner/types.py 2 2 6 torchrec/modules/lazy_extension.py torchrec/fx/__init__.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/fx/tracer.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/linter/module_linter.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/models/deepfm.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/models/dlrm.py 2 2 4 torchrec/modules/lazy_extension.py torchrec/modules/__init__.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/modules/activation.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/modules/crossnet.py 2 2 6 torchrec/modules/lazy_extension.py torchrec/modules/deepfm.py 2 2 3 torchrec/modules/lazy_extension.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/modules/lazy_extension.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/modules/lazy_extension.py torchrec/modules/feature_processor.py 2 2 3 torchrec/modules/mlp.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/modules/mlp.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/modules/mlp.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/modules/mlp.py torchrec/optim/keyed.py 2 2 8 torchrec/modules/mlp.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/modules/mlp.py torchrec/distributed/dist_data.py 2 2 14 torchrec/modules/mlp.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/modules/mlp.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/modules/mlp.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/modules/mlp.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/modules/mlp.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/modules/mlp.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/modules/mlp.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/modules/mlp.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/modules/mlp.py torchrec/distributed/types.py 2 2 10 torchrec/modules/mlp.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/modules/mlp.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/modules/mlp.py torchrec/distributed/utils.py 2 2 3 torchrec/modules/mlp.py torchrec/__init__.py 2 2 3 torchrec/modules/mlp.py torchrec/datasets/utils.py 2 2 3 torchrec/modules/mlp.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/modules/mlp.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/modules/mlp.py setup.py 2 2 4 torchrec/modules/mlp.py torchrec/datasets/__init__.py 2 2 2 torchrec/modules/mlp.py torchrec/datasets/criteo.py 2 2 4 torchrec/modules/mlp.py torchrec/datasets/movielens.py 2 2 2 torchrec/modules/mlp.py torchrec/datasets/random.py 2 2 3 torchrec/modules/mlp.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/modules/mlp.py torchrec/distributed/__init__.py 2 2 2 torchrec/modules/mlp.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/modules/mlp.py torchrec/distributed/comm.py 2 2 3 torchrec/modules/mlp.py torchrec/distributed/embedding.py 2 2 2 torchrec/modules/mlp.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/modules/mlp.py torchrec/distributed/planner/types.py 2 2 6 torchrec/modules/mlp.py torchrec/fx/__init__.py 2 2 2 torchrec/modules/mlp.py torchrec/fx/tracer.py 2 2 3 torchrec/modules/mlp.py torchrec/linter/module_linter.py 2 2 2 torchrec/modules/mlp.py torchrec/models/deepfm.py 2 2 3 torchrec/modules/mlp.py torchrec/models/dlrm.py 2 2 4 torchrec/modules/mlp.py torchrec/modules/__init__.py 2 2 3 torchrec/modules/mlp.py torchrec/modules/activation.py 2 2 3 torchrec/modules/mlp.py torchrec/modules/crossnet.py 2 2 6 torchrec/modules/mlp.py torchrec/modules/deepfm.py 2 2 3 torchrec/modules/mlp.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/modules/mlp.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/modules/mlp.py torchrec/modules/feature_processor.py 2 2 3 torchrec/modules/mlp.py torchrec/modules/lazy_extension.py 2 2 2 torchrec/modules/score_learning.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/score_learning.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/score_learning.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/score_learning.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/score_learning.py torchrec/__init__.py 2 3 3 torchrec/modules/score_learning.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/score_learning.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/score_learning.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/modules/score_learning.py setup.py 2 3 4 torchrec/modules/score_learning.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/score_learning.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/score_learning.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/score_learning.py torchrec/datasets/random.py 2 3 3 torchrec/modules/score_learning.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/score_learning.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/score_learning.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/score_learning.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/score_learning.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/score_learning.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/score_learning.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/score_learning.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/score_learning.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/score_learning.py torchrec/models/dlrm.py 2 3 4 torchrec/modules/score_learning.py torchrec/modules/__init__.py 2 3 3 torchrec/modules/score_learning.py torchrec/modules/activation.py 2 3 3 torchrec/modules/score_learning.py torchrec/modules/crossnet.py 2 3 6 torchrec/modules/score_learning.py torchrec/modules/deepfm.py 2 3 3 torchrec/modules/score_learning.py torchrec/modules/embedding_configs.py 2 3 2 torchrec/modules/score_learning.py torchrec/modules/lazy_extension.py 2 3 2 torchrec/modules/score_learning.py torchrec/modules/mlp.py 2 3 2 torchrec/modules/utils.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/modules/utils.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/modules/utils.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/modules/utils.py torchrec/distributed/utils.py 2 3 3 torchrec/modules/utils.py torchrec/__init__.py 2 3 3 torchrec/modules/utils.py torchrec/datasets/utils.py 2 3 3 torchrec/modules/utils.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/modules/utils.py torchrec/distributed/planner/planners.py 2 3 9 torchrec/modules/utils.py setup.py 2 3 4 torchrec/modules/utils.py torchrec/datasets/__init__.py 2 3 2 torchrec/modules/utils.py torchrec/datasets/criteo.py 2 3 4 torchrec/modules/utils.py torchrec/datasets/movielens.py 2 3 2 torchrec/modules/utils.py torchrec/datasets/random.py 2 3 3 torchrec/modules/utils.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/modules/utils.py torchrec/distributed/__init__.py 2 3 2 torchrec/modules/utils.py torchrec/distributed/embedding.py 2 3 2 torchrec/modules/utils.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/modules/utils.py torchrec/distributed/planner/types.py 2 3 6 torchrec/modules/utils.py torchrec/fx/__init__.py 2 3 2 torchrec/modules/utils.py torchrec/fx/tracer.py 2 3 3 torchrec/modules/utils.py torchrec/linter/module_linter.py 2 3 2 torchrec/modules/utils.py torchrec/models/deepfm.py 2 3 3 torchrec/modules/utils.py torchrec/models/dlrm.py 2 3 4 torchrec/modules/utils.py torchrec/modules/__init__.py 2 3 3 torchrec/modules/utils.py torchrec/modules/activation.py 2 3 3 torchrec/modules/utils.py torchrec/modules/crossnet.py 2 3 6 torchrec/modules/utils.py torchrec/modules/deepfm.py 2 3 3 torchrec/modules/utils.py torchrec/modules/embedding_configs.py 2 3 2 torchrec/modules/utils.py torchrec/modules/lazy_extension.py 2 3 2 torchrec/modules/utils.py torchrec/modules/mlp.py 2 3 2 torchrec/optim/__init__.py torchrec/distributed/embedding_lookup.py 2 3 16 torchrec/optim/__init__.py torchrec/distributed/planner/utils.py 2 3 6 torchrec/optim/__init__.py torchrec/distributed/grouped_position_weighted.py 2 3 4 torchrec/optim/__init__.py torchrec/optim/keyed.py 2 3 8 torchrec/optim/__init__.py torchrec/distributed/cw_sharding.py 2 3 12 torchrec/optim/__init__.py torchrec/distributed/dist_data.py 2 3 14 torchrec/optim/__init__.py torchrec/distributed/dp_sharding.py 2 3 7 torchrec/optim/__init__.py torchrec/distributed/embedding_sharding.py 2 3 12 torchrec/optim/__init__.py torchrec/distributed/embedding_types.py 2 3 9 torchrec/optim/__init__.py torchrec/distributed/embeddingbag.py 2 3 9 torchrec/optim/__init__.py torchrec/distributed/model_parallel.py 2 3 7 torchrec/optim/__init__.py torchrec/distributed/rw_sharding.py 2 3 9 torchrec/optim/__init__.py torchrec/distributed/tw_sharding.py 2 3 9 torchrec/optim/__init__.py torchrec/distributed/twrw_sharding.py 2 3 9 torchrec/optim/__init__.py torchrec/distributed/types.py 2 3 10 torchrec/optim/__init__.py torchrec/distributed/comm_ops.py 2 3 7 torchrec/optim/__init__.py torchrec/distributed/train_pipeline.py 2 3 7 torchrec/optim/__init__.py torchrec/distributed/utils.py 2 3 3 torchrec/optim/__init__.py torchrec/__init__.py 2 3 3 torchrec/optim/__init__.py torchrec/datasets/utils.py 2 3 3 torchrec/optim/__init__.py torchrec/sparse/jagged_tensor.py 2 3 5 torchrec/optim/__init__.py torchrec/quant/embedding_modules.py 2 3 8 torchrec/optim/__init__.py setup.py 2 3 4 torchrec/optim/__init__.py torchrec/datasets/__init__.py 2 3 2 torchrec/optim/__init__.py torchrec/datasets/criteo.py 2 3 4 torchrec/optim/__init__.py torchrec/datasets/movielens.py 2 3 2 torchrec/optim/__init__.py torchrec/datasets/random.py 2 3 3 torchrec/optim/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 3 3 torchrec/optim/__init__.py torchrec/distributed/__init__.py 2 3 2 torchrec/optim/__init__.py torchrec/distributed/collective_utils.py 2 3 3 torchrec/optim/__init__.py torchrec/distributed/comm.py 2 3 3 torchrec/optim/__init__.py torchrec/distributed/embedding.py 2 3 2 torchrec/optim/__init__.py torchrec/distributed/planner/__init__.py 2 3 3 torchrec/optim/__init__.py torchrec/distributed/planner/types.py 2 3 6 torchrec/optim/__init__.py torchrec/fx/__init__.py 2 3 2 torchrec/optim/__init__.py torchrec/fx/tracer.py 2 3 3 torchrec/optim/__init__.py torchrec/linter/module_linter.py 2 3 2 torchrec/optim/__init__.py torchrec/models/__init__.py 2 3 2 torchrec/optim/__init__.py torchrec/models/deepfm.py 2 3 3 torchrec/optim/__init__.py torchrec/modules/__init__.py 2 3 3 torchrec/optim/__init__.py torchrec/modules/embedding_configs.py 2 3 2 torchrec/optim/__init__.py torchrec/modules/embedding_modules.py 2 3 4 torchrec/optim/__init__.py torchrec/modules/feature_processor.py 2 3 3 torchrec/optim/__init__.py torchrec/modules/lazy_extension.py 2 3 2 torchrec/optim/__init__.py torchrec/modules/mlp.py 2 3 2 torchrec/optim/__init__.py torchrec/modules/score_learning.py 2 3 3 torchrec/optim/__init__.py torchrec/modules/utils.py 2 3 3 torchrec/optim/clipping.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/optim/clipping.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/optim/clipping.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/optim/clipping.py torchrec/optim/keyed.py 2 2 8 torchrec/optim/clipping.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/optim/clipping.py torchrec/distributed/dist_data.py 2 2 14 torchrec/optim/clipping.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/optim/clipping.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/optim/clipping.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/optim/clipping.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/optim/clipping.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/optim/clipping.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/optim/clipping.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/optim/clipping.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/optim/clipping.py torchrec/distributed/types.py 2 2 10 torchrec/optim/clipping.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/optim/clipping.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/optim/clipping.py torchrec/distributed/utils.py 2 2 3 torchrec/optim/clipping.py torchrec/__init__.py 2 2 3 torchrec/optim/clipping.py torchrec/datasets/utils.py 2 2 3 torchrec/optim/clipping.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/optim/clipping.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/optim/clipping.py setup.py 2 2 4 torchrec/optim/clipping.py torchrec/datasets/__init__.py 2 2 2 torchrec/optim/clipping.py torchrec/datasets/criteo.py 2 2 4 torchrec/optim/clipping.py torchrec/datasets/movielens.py 2 2 2 torchrec/optim/clipping.py torchrec/datasets/random.py 2 2 3 torchrec/optim/clipping.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/optim/clipping.py torchrec/distributed/__init__.py 2 2 2 torchrec/optim/clipping.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/optim/clipping.py torchrec/distributed/comm.py 2 2 3 torchrec/optim/clipping.py torchrec/distributed/embedding.py 2 2 2 torchrec/optim/clipping.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/optim/clipping.py torchrec/distributed/planner/types.py 2 2 6 torchrec/optim/clipping.py torchrec/fx/__init__.py 2 2 2 torchrec/optim/clipping.py torchrec/fx/tracer.py 2 2 3 torchrec/optim/clipping.py torchrec/linter/module_linter.py 2 2 2 torchrec/optim/clipping.py torchrec/models/deepfm.py 2 2 3 torchrec/optim/clipping.py torchrec/models/dlrm.py 2 2 4 torchrec/optim/clipping.py torchrec/modules/__init__.py 2 2 3 torchrec/optim/clipping.py torchrec/modules/activation.py 2 2 3 torchrec/optim/clipping.py torchrec/modules/crossnet.py 2 2 6 torchrec/optim/clipping.py torchrec/modules/deepfm.py 2 2 3 torchrec/optim/clipping.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/optim/clipping.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/optim/clipping.py torchrec/modules/feature_processor.py 2 2 3 torchrec/optim/clipping.py torchrec/modules/lazy_extension.py 2 2 2 torchrec/optim/clipping.py torchrec/modules/mlp.py 2 2 2 torchrec/optim/clipping.py torchrec/modules/score_learning.py 2 2 3 torchrec/optim/clipping.py torchrec/modules/utils.py 2 2 3 torchrec/optim/clipping.py torchrec/optim/__init__.py 2 2 3 torchrec/optim/fused.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/optim/fused.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/optim/fused.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/optim/fused.py torchrec/optim/keyed.py 2 2 8 torchrec/optim/fused.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/optim/fused.py torchrec/distributed/dist_data.py 2 2 14 torchrec/optim/fused.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/optim/fused.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/optim/fused.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/optim/fused.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/optim/fused.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/optim/fused.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/optim/fused.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/optim/fused.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/optim/fused.py torchrec/distributed/types.py 2 2 10 torchrec/optim/fused.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/optim/fused.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/optim/fused.py torchrec/distributed/utils.py 2 2 3 torchrec/optim/fused.py torchrec/__init__.py 2 2 3 torchrec/optim/fused.py torchrec/datasets/utils.py 2 2 3 torchrec/optim/fused.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/optim/fused.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/optim/fused.py setup.py 2 2 4 torchrec/optim/fused.py torchrec/datasets/__init__.py 2 2 2 torchrec/optim/fused.py torchrec/datasets/criteo.py 2 2 4 torchrec/optim/fused.py torchrec/datasets/movielens.py 2 2 2 torchrec/optim/fused.py torchrec/datasets/random.py 2 2 3 torchrec/optim/fused.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/optim/fused.py torchrec/distributed/__init__.py 2 2 2 torchrec/optim/fused.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/optim/fused.py torchrec/distributed/comm.py 2 2 3 torchrec/optim/fused.py torchrec/distributed/embedding.py 2 2 2 torchrec/optim/fused.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/optim/fused.py torchrec/distributed/planner/types.py 2 2 6 torchrec/optim/fused.py torchrec/fx/__init__.py 2 2 2 torchrec/optim/fused.py torchrec/fx/tracer.py 2 2 3 torchrec/optim/fused.py torchrec/linter/module_linter.py 2 2 2 torchrec/optim/fused.py torchrec/models/deepfm.py 2 2 3 torchrec/optim/fused.py torchrec/models/dlrm.py 2 2 4 torchrec/optim/fused.py torchrec/modules/__init__.py 2 2 3 torchrec/optim/fused.py torchrec/modules/activation.py 2 2 3 torchrec/optim/fused.py torchrec/modules/crossnet.py 2 2 6 torchrec/optim/fused.py torchrec/modules/deepfm.py 2 2 3 torchrec/optim/fused.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/optim/fused.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/optim/fused.py torchrec/modules/feature_processor.py 2 2 3 torchrec/optim/fused.py torchrec/modules/lazy_extension.py 2 2 2 torchrec/optim/fused.py torchrec/modules/mlp.py 2 2 2 torchrec/optim/fused.py torchrec/modules/score_learning.py 2 2 3 torchrec/optim/fused.py torchrec/modules/utils.py 2 2 3 torchrec/optim/fused.py torchrec/optim/__init__.py 2 2 3 torchrec/optim/fused.py torchrec/optim/clipping.py 2 2 2 torchrec/optim/warmup.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/optim/warmup.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/optim/warmup.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/optim/warmup.py torchrec/optim/keyed.py 2 2 8 torchrec/optim/warmup.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/optim/warmup.py torchrec/distributed/dist_data.py 2 2 14 torchrec/optim/warmup.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/optim/warmup.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/optim/warmup.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/optim/warmup.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/optim/warmup.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/optim/warmup.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/optim/warmup.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/optim/warmup.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/optim/warmup.py torchrec/distributed/types.py 2 2 10 torchrec/optim/warmup.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/optim/warmup.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/optim/warmup.py torchrec/distributed/utils.py 2 2 3 torchrec/optim/warmup.py torchrec/__init__.py 2 2 3 torchrec/optim/warmup.py torchrec/datasets/utils.py 2 2 3 torchrec/optim/warmup.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/optim/warmup.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/optim/warmup.py setup.py 2 2 4 torchrec/optim/warmup.py torchrec/datasets/__init__.py 2 2 2 torchrec/optim/warmup.py torchrec/datasets/criteo.py 2 2 4 torchrec/optim/warmup.py torchrec/datasets/movielens.py 2 2 2 torchrec/optim/warmup.py torchrec/datasets/random.py 2 2 3 torchrec/optim/warmup.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/optim/warmup.py torchrec/distributed/__init__.py 2 2 2 torchrec/optim/warmup.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/optim/warmup.py torchrec/distributed/comm.py 2 2 3 torchrec/optim/warmup.py torchrec/distributed/embedding.py 2 2 2 torchrec/optim/warmup.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/optim/warmup.py torchrec/distributed/planner/types.py 2 2 6 torchrec/optim/warmup.py torchrec/fx/__init__.py 2 2 2 torchrec/optim/warmup.py torchrec/fx/tracer.py 2 2 3 torchrec/optim/warmup.py torchrec/linter/module_linter.py 2 2 2 torchrec/optim/warmup.py torchrec/models/deepfm.py 2 2 3 torchrec/optim/warmup.py torchrec/models/dlrm.py 2 2 4 torchrec/optim/warmup.py torchrec/modules/__init__.py 2 2 3 torchrec/optim/warmup.py torchrec/modules/activation.py 2 2 3 torchrec/optim/warmup.py torchrec/modules/crossnet.py 2 2 6 torchrec/optim/warmup.py torchrec/modules/deepfm.py 2 2 3 torchrec/optim/warmup.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/optim/warmup.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/optim/warmup.py torchrec/modules/feature_processor.py 2 2 3 torchrec/optim/warmup.py torchrec/modules/lazy_extension.py 2 2 2 torchrec/optim/warmup.py torchrec/modules/mlp.py 2 2 2 torchrec/optim/warmup.py torchrec/modules/score_learning.py 2 2 3 torchrec/optim/warmup.py torchrec/modules/utils.py 2 2 3 torchrec/optim/warmup.py torchrec/optim/__init__.py 2 2 3 torchrec/optim/warmup.py torchrec/optim/clipping.py 2 2 2 torchrec/optim/warmup.py torchrec/optim/fused.py 2 2 2 torchrec/quant/__init__.py torchrec/distributed/embedding_lookup.py 2 2 16 torchrec/quant/__init__.py torchrec/distributed/planner/utils.py 2 2 6 torchrec/quant/__init__.py torchrec/distributed/grouped_position_weighted.py 2 2 4 torchrec/quant/__init__.py torchrec/optim/keyed.py 2 2 8 torchrec/quant/__init__.py torchrec/distributed/cw_sharding.py 2 2 12 torchrec/quant/__init__.py torchrec/distributed/dist_data.py 2 2 14 torchrec/quant/__init__.py torchrec/distributed/dp_sharding.py 2 2 7 torchrec/quant/__init__.py torchrec/distributed/embedding_sharding.py 2 2 12 torchrec/quant/__init__.py torchrec/distributed/embedding_types.py 2 2 9 torchrec/quant/__init__.py torchrec/distributed/embeddingbag.py 2 2 9 torchrec/quant/__init__.py torchrec/distributed/model_parallel.py 2 2 7 torchrec/quant/__init__.py torchrec/distributed/rw_sharding.py 2 2 9 torchrec/quant/__init__.py torchrec/distributed/tw_sharding.py 2 2 9 torchrec/quant/__init__.py torchrec/distributed/twrw_sharding.py 2 2 9 torchrec/quant/__init__.py torchrec/distributed/types.py 2 2 10 torchrec/quant/__init__.py torchrec/distributed/comm_ops.py 2 2 7 torchrec/quant/__init__.py torchrec/distributed/train_pipeline.py 2 2 7 torchrec/quant/__init__.py torchrec/distributed/utils.py 2 2 3 torchrec/quant/__init__.py torchrec/__init__.py 2 2 3 torchrec/quant/__init__.py torchrec/datasets/utils.py 2 2 3 torchrec/quant/__init__.py torchrec/sparse/jagged_tensor.py 2 2 5 torchrec/quant/__init__.py torchrec/quant/embedding_modules.py 2 2 8 torchrec/quant/__init__.py setup.py 2 2 4 torchrec/quant/__init__.py torchrec/datasets/__init__.py 2 2 2 torchrec/quant/__init__.py torchrec/datasets/criteo.py 2 2 4 torchrec/quant/__init__.py torchrec/datasets/movielens.py 2 2 2 torchrec/quant/__init__.py torchrec/datasets/random.py 2 2 3 torchrec/quant/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 2 2 3 torchrec/quant/__init__.py torchrec/distributed/__init__.py 2 2 2 torchrec/quant/__init__.py torchrec/distributed/collective_utils.py 2 2 3 torchrec/quant/__init__.py torchrec/distributed/comm.py 2 2 3 torchrec/quant/__init__.py torchrec/distributed/embedding.py 2 2 2 torchrec/quant/__init__.py torchrec/distributed/planner/__init__.py 2 2 3 torchrec/quant/__init__.py torchrec/distributed/planner/types.py 2 2 6 torchrec/quant/__init__.py torchrec/fx/__init__.py 2 2 2 torchrec/quant/__init__.py torchrec/fx/tracer.py 2 2 3 torchrec/quant/__init__.py torchrec/linter/module_linter.py 2 2 2 torchrec/quant/__init__.py torchrec/models/deepfm.py 2 2 3 torchrec/quant/__init__.py torchrec/models/dlrm.py 2 2 4 torchrec/quant/__init__.py torchrec/modules/__init__.py 2 2 3 torchrec/quant/__init__.py torchrec/modules/activation.py 2 2 3 torchrec/quant/__init__.py torchrec/modules/crossnet.py 2 2 6 torchrec/quant/__init__.py torchrec/modules/deepfm.py 2 2 3 torchrec/quant/__init__.py torchrec/modules/embedding_configs.py 2 2 2 torchrec/quant/__init__.py torchrec/modules/embedding_modules.py 2 2 4 torchrec/quant/__init__.py torchrec/modules/feature_processor.py 2 2 3 torchrec/quant/__init__.py torchrec/modules/lazy_extension.py 2 2 2 torchrec/quant/__init__.py torchrec/modules/mlp.py 2 2 2 torchrec/quant/__init__.py torchrec/modules/score_learning.py 2 2 3 torchrec/quant/__init__.py torchrec/modules/utils.py 2 2 3 torchrec/quant/__init__.py torchrec/optim/__init__.py 2 2 3 torchrec/quant/__init__.py torchrec/optim/clipping.py 2 2 2 torchrec/quant/__init__.py torchrec/optim/fused.py 2 2 2 torchrec/quant/__init__.py torchrec/optim/warmup.py 2 2 2 torchrec/sparse/__init__.py torchrec/models/__init__.py 2 2 2 torchrec/sparse/__init__.py torchrec/models/dlrm.py 2 2 4 torchrec/sparse/__init__.py torchrec/modules/activation.py 2 2 3 torchrec/sparse/__init__.py torchrec/modules/crossnet.py 2 2 6 torchrec/sparse/__init__.py torchrec/modules/deepfm.py 2 2 3 torchrec/sparse/__init__.py torchrec/optim/__init__.py 2 2 3 torchrec/distributed/embedding_kernel.py torchrec/distributed/batched_embedding_kernel.py 1 1 1 torchrec/distributed/embedding_lookup.py torchrec/distributed/batched_embedding_kernel.py 1 16 1 torchrec/distributed/embedding_lookup.py torchrec/distributed/embedding_kernel.py 1 16 1 torchrec/distributed/quant_embedding_kernel.py torchrec/distributed/batched_embedding_kernel.py 1 1 1 torchrec/distributed/quant_embedding_kernel.py torchrec/distributed/embedding_kernel.py 1 1 1 torchrec/distributed/quant_embedding_kernel.py torchrec/distributed/embedding_lookup.py 1 1 16 torchrec/datasets/experimental/criteo_dataframes.py torchrec/datasets/experimental/__init__.py 1 1 1 torchrec/distributed/planner/constants.py torchrec/distributed/embedding_lookup.py 1 3 16 torchrec/distributed/grouped_position_weighted.py torchrec/distributed/planner/constants.py 1 4 3 torchrec/optim/keyed.py torchrec/distributed/planner/constants.py 1 8 3 torchrec/distributed/cw_sharding.py torchrec/distributed/planner/constants.py 1 12 3 torchrec/distributed/dist_data.py torchrec/distributed/planner/constants.py 1 14 3 torchrec/distributed/dp_sharding.py torchrec/distributed/planner/constants.py 1 7 3 torchrec/distributed/embedding_sharding.py torchrec/distributed/planner/constants.py 1 12 3 torchrec/distributed/embedding_types.py torchrec/distributed/planner/constants.py 1 9 3 torchrec/distributed/embeddingbag.py torchrec/distributed/planner/constants.py 1 9 3 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/embedding_lookup.py 1 1 16 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/cw_sharding.py 1 1 12 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/dist_data.py 1 1 14 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/dp_sharding.py 1 1 7 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/embedding_sharding.py 1 1 12 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/embedding_types.py 1 1 9 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/embeddingbag.py 1 1 9 torchrec/distributed/quant_embeddingbag.py torchrec/distributed/model_parallel.py 1 1 7 torchrec/distributed/rw_sharding.py torchrec/distributed/quant_embeddingbag.py 1 9 1 torchrec/distributed/rw_sharding.py torchrec/distributed/planner/constants.py 1 9 3 torchrec/distributed/tw_sharding.py torchrec/distributed/quant_embeddingbag.py 1 9 1 torchrec/distributed/tw_sharding.py torchrec/distributed/planner/constants.py 1 9 3 torchrec/distributed/twrw_sharding.py torchrec/distributed/quant_embeddingbag.py 1 9 1 torchrec/distributed/twrw_sharding.py torchrec/distributed/planner/constants.py 1 9 3 torchrec/distributed/types.py torchrec/distributed/quant_embeddingbag.py 1 10 1 torchrec/distributed/types.py torchrec/distributed/planner/constants.py 1 10 3 torchrec/distributed/comm_ops.py torchrec/distributed/planner/constants.py 1 7 3 torchrec/distributed/train_pipeline.py torchrec/distributed/planner/constants.py 1 7 3 torchrec/distributed/utils.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/__init__.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/datasets/utils.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/sparse/jagged_tensor.py torchrec/distributed/planner/constants.py 1 5 3 torchrec/streamable.py torchrec/distributed/embedding_sharding.py 1 1 12 torchrec/streamable.py torchrec/distributed/embedding_types.py 1 1 9 torchrec/streamable.py torchrec/distributed/types.py 1 1 10 torchrec/streamable.py torchrec/distributed/train_pipeline.py 1 1 7 torchrec/streamable.py torchrec/__init__.py 1 1 3 torchrec/streamable.py torchrec/datasets/utils.py 1 1 3 torchrec/streamable.py torchrec/sparse/jagged_tensor.py 1 1 5 torchrec/distributed/planner/enumerators.py torchrec/distributed/embedding_lookup.py 1 5 16 torchrec/distributed/planner/enumerators.py torchrec/distributed/grouped_position_weighted.py 1 5 4 torchrec/distributed/planner/enumerators.py torchrec/optim/keyed.py 1 5 8 torchrec/distributed/planner/enumerators.py torchrec/distributed/dist_data.py 1 5 14 torchrec/distributed/planner/enumerators.py torchrec/distributed/dp_sharding.py 1 5 7 torchrec/distributed/planner/enumerators.py torchrec/distributed/embedding_sharding.py 1 5 12 torchrec/distributed/planner/enumerators.py torchrec/distributed/twrw_sharding.py 1 5 9 torchrec/distributed/planner/enumerators.py torchrec/distributed/comm_ops.py 1 5 7 torchrec/distributed/planner/enumerators.py torchrec/distributed/train_pipeline.py 1 5 7 torchrec/distributed/planner/enumerators.py torchrec/distributed/utils.py 1 5 3 torchrec/distributed/planner/enumerators.py torchrec/__init__.py 1 5 3 torchrec/distributed/planner/enumerators.py torchrec/datasets/utils.py 1 5 3 torchrec/distributed/planner/enumerators.py torchrec/sparse/jagged_tensor.py 1 5 5 torchrec/distributed/planner/partitioners.py torchrec/distributed/embedding_lookup.py 1 3 16 torchrec/distributed/planner/partitioners.py torchrec/distributed/grouped_position_weighted.py 1 3 4 torchrec/distributed/planner/partitioners.py torchrec/optim/keyed.py 1 3 8 torchrec/distributed/planner/partitioners.py torchrec/distributed/dist_data.py 1 3 14 torchrec/distributed/planner/partitioners.py torchrec/distributed/dp_sharding.py 1 3 7 torchrec/distributed/planner/partitioners.py torchrec/distributed/embedding_sharding.py 1 3 12 torchrec/distributed/planner/partitioners.py torchrec/distributed/twrw_sharding.py 1 3 9 torchrec/distributed/planner/partitioners.py torchrec/distributed/comm_ops.py 1 3 7 torchrec/distributed/planner/partitioners.py torchrec/distributed/train_pipeline.py 1 3 7 torchrec/distributed/planner/partitioners.py torchrec/distributed/utils.py 1 3 3 torchrec/distributed/planner/partitioners.py torchrec/__init__.py 1 3 3 torchrec/distributed/planner/partitioners.py torchrec/datasets/utils.py 1 3 3 torchrec/distributed/planner/partitioners.py torchrec/sparse/jagged_tensor.py 1 3 5 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/embedding_lookup.py 1 6 16 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/grouped_position_weighted.py 1 6 4 torchrec/distributed/planner/shard_estimators.py torchrec/optim/keyed.py 1 6 8 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/dist_data.py 1 6 14 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/dp_sharding.py 1 6 7 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/embedding_sharding.py 1 6 12 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/twrw_sharding.py 1 6 9 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/comm_ops.py 1 6 7 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/train_pipeline.py 1 6 7 torchrec/distributed/planner/shard_estimators.py torchrec/distributed/utils.py 1 6 3 torchrec/distributed/planner/shard_estimators.py torchrec/__init__.py 1 6 3 torchrec/distributed/planner/shard_estimators.py torchrec/datasets/utils.py 1 6 3 torchrec/distributed/planner/shard_estimators.py torchrec/sparse/jagged_tensor.py 1 6 5 torchrec/distributed/planner/stats.py torchrec/distributed/embedding_lookup.py 1 6 16 torchrec/distributed/planner/stats.py torchrec/distributed/grouped_position_weighted.py 1 6 4 torchrec/distributed/planner/stats.py torchrec/optim/keyed.py 1 6 8 torchrec/distributed/planner/stats.py torchrec/distributed/dist_data.py 1 6 14 torchrec/distributed/planner/stats.py torchrec/distributed/dp_sharding.py 1 6 7 torchrec/distributed/planner/stats.py torchrec/distributed/embedding_sharding.py 1 6 12 torchrec/distributed/planner/stats.py torchrec/distributed/twrw_sharding.py 1 6 9 torchrec/distributed/planner/stats.py torchrec/distributed/comm_ops.py 1 6 7 torchrec/distributed/planner/stats.py torchrec/distributed/train_pipeline.py 1 6 7 torchrec/distributed/planner/stats.py torchrec/distributed/utils.py 1 6 3 torchrec/distributed/planner/stats.py torchrec/__init__.py 1 6 3 torchrec/distributed/planner/stats.py torchrec/datasets/utils.py 1 6 3 torchrec/distributed/planner/stats.py torchrec/sparse/jagged_tensor.py 1 6 5 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/embedding_lookup.py 1 4 16 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/grouped_position_weighted.py 1 4 4 torchrec/distributed/planner/storage_reservations.py torchrec/optim/keyed.py 1 4 8 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/dist_data.py 1 4 14 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/dp_sharding.py 1 4 7 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/embedding_sharding.py 1 4 12 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/twrw_sharding.py 1 4 9 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/comm_ops.py 1 4 7 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/train_pipeline.py 1 4 7 torchrec/distributed/planner/storage_reservations.py torchrec/distributed/utils.py 1 4 3 torchrec/distributed/planner/storage_reservations.py torchrec/__init__.py 1 4 3 torchrec/distributed/planner/storage_reservations.py torchrec/datasets/utils.py 1 4 3 torchrec/distributed/planner/storage_reservations.py torchrec/sparse/jagged_tensor.py 1 4 5 torchrec/distributed/twcw_sharding.py torchrec/distributed/cw_sharding.py 1 1 12 torchrec/distributed/twcw_sharding.py torchrec/distributed/embedding_types.py 1 1 9 torchrec/distributed/twcw_sharding.py torchrec/distributed/embeddingbag.py 1 1 9 torchrec/distributed/twcw_sharding.py torchrec/distributed/rw_sharding.py 1 1 9 torchrec/distributed/twcw_sharding.py torchrec/distributed/tw_sharding.py 1 1 9 torchrec/distributed/twcw_sharding.py torchrec/distributed/types.py 1 1 10 torchrec/distributed/twcw_sharding.py torchrec/distributed/planner/enumerators.py 1 1 5 torchrec/distributed/twcw_sharding.py torchrec/distributed/planner/partitioners.py 1 1 3 torchrec/distributed/twcw_sharding.py torchrec/distributed/planner/shard_estimators.py 1 1 6 torchrec/distributed/twcw_sharding.py torchrec/distributed/planner/stats.py 1 1 6 torchrec/distributed/twcw_sharding.py torchrec/distributed/planner/storage_reservations.py 1 1 4 torchrec/distributed/planner/planners.py torchrec/distributed/dp_sharding.py 1 9 7 torchrec/distributed/planner/planners.py torchrec/distributed/embedding_types.py 1 9 9 torchrec/distributed/planner/planners.py torchrec/distributed/utils.py 1 9 3 torchrec/distributed/planner/planners.py torchrec/__init__.py 1 9 3 torchrec/distributed/planner/planners.py torchrec/datasets/utils.py 1 9 3 torchrec/distributed/planner/planners.py torchrec/sparse/jagged_tensor.py 1 9 5 torchrec/sparse/jagged_tensor_ops.h torchrec/sparse/jagged_tensor_ops.cpp 1 1 1 torchrec/quant/embedding_modules.py torchrec/distributed/planner/constants.py 1 8 3 torchrec/quant/embedding_modules.py torchrec/distributed/planner/enumerators.py 1 8 5 torchrec/quant/embedding_modules.py torchrec/distributed/planner/partitioners.py 1 8 3 torchrec/quant/embedding_modules.py torchrec/distributed/planner/shard_estimators.py 1 8 6 torchrec/quant/embedding_modules.py torchrec/distributed/planner/stats.py 1 8 6 torchrec/quant/embedding_modules.py torchrec/distributed/planner/storage_reservations.py 1 8 4 setup.py torchrec/distributed/planner/constants.py 1 4 3 setup.py torchrec/distributed/planner/enumerators.py 1 4 5 setup.py torchrec/distributed/planner/partitioners.py 1 4 3 setup.py torchrec/distributed/planner/shard_estimators.py 1 4 6 setup.py torchrec/distributed/planner/stats.py 1 4 6 setup.py torchrec/distributed/planner/storage_reservations.py 1 4 4 setup.py torchrec/distributed/planner/planners.py 1 4 9 torchrec/datasets/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/datasets/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/datasets/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/datasets/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/datasets/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/datasets/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/datasets/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/datasets/criteo.py torchrec/distributed/planner/constants.py 1 4 3 torchrec/datasets/criteo.py torchrec/distributed/planner/enumerators.py 1 4 5 torchrec/datasets/criteo.py torchrec/distributed/planner/partitioners.py 1 4 3 torchrec/datasets/criteo.py torchrec/distributed/planner/shard_estimators.py 1 4 6 torchrec/datasets/criteo.py torchrec/distributed/planner/stats.py 1 4 6 torchrec/datasets/criteo.py torchrec/distributed/planner/storage_reservations.py 1 4 4 torchrec/datasets/criteo.py torchrec/distributed/planner/planners.py 1 4 9 torchrec/datasets/movielens.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/datasets/movielens.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/datasets/movielens.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/datasets/movielens.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/datasets/movielens.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/datasets/movielens.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/datasets/movielens.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/datasets/random.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/datasets/random.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/datasets/random.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/datasets/random.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/datasets/random.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/datasets/random.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/datasets/random.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/datasets/scripts/preprocess_criteo.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/distributed/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/distributed/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/distributed/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/distributed/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/distributed/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/distributed/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/distributed/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/distributed/collective_utils.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/distributed/collective_utils.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/distributed/collective_utils.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/distributed/collective_utils.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/distributed/collective_utils.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/distributed/collective_utils.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/distributed/comm.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/distributed/comm.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/distributed/comm.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/distributed/comm.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/distributed/comm.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/distributed/comm.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/distributed/embedding.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/distributed/embedding.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/distributed/embedding.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/distributed/embedding.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/distributed/embedding.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/distributed/embedding.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/distributed/embedding.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/embedding_lookup.py 1 2 16 torchrec/distributed/planner/perf_models.py torchrec/distributed/grouped_position_weighted.py 1 2 4 torchrec/distributed/planner/perf_models.py torchrec/optim/keyed.py 1 2 8 torchrec/distributed/planner/perf_models.py torchrec/distributed/cw_sharding.py 1 2 12 torchrec/distributed/planner/perf_models.py torchrec/distributed/dist_data.py 1 2 14 torchrec/distributed/planner/perf_models.py torchrec/distributed/dp_sharding.py 1 2 7 torchrec/distributed/planner/perf_models.py torchrec/distributed/embedding_sharding.py 1 2 12 torchrec/distributed/planner/perf_models.py torchrec/distributed/embedding_types.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/embeddingbag.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/rw_sharding.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/tw_sharding.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/twrw_sharding.py 1 2 9 torchrec/distributed/planner/perf_models.py torchrec/distributed/types.py 1 2 10 torchrec/distributed/planner/perf_models.py torchrec/distributed/comm_ops.py 1 2 7 torchrec/distributed/planner/perf_models.py torchrec/distributed/train_pipeline.py 1 2 7 torchrec/distributed/planner/perf_models.py torchrec/distributed/utils.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/__init__.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/datasets/utils.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/sparse/jagged_tensor.py 1 2 5 torchrec/distributed/planner/perf_models.py torchrec/quant/embedding_modules.py 1 2 8 torchrec/distributed/planner/perf_models.py setup.py 1 2 4 torchrec/distributed/planner/perf_models.py torchrec/datasets/__init__.py 1 2 2 torchrec/distributed/planner/perf_models.py torchrec/datasets/criteo.py 1 2 4 torchrec/distributed/planner/perf_models.py torchrec/datasets/movielens.py 1 2 2 torchrec/distributed/planner/perf_models.py torchrec/datasets/random.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/datasets/scripts/preprocess_criteo.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/distributed/__init__.py 1 2 2 torchrec/distributed/planner/perf_models.py torchrec/distributed/collective_utils.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/distributed/comm.py 1 2 3 torchrec/distributed/planner/perf_models.py torchrec/distributed/embedding.py 1 2 2 torchrec/distributed/planner/proposers.py torchrec/distributed/embedding_lookup.py 1 3 16 torchrec/distributed/planner/proposers.py torchrec/distributed/grouped_position_weighted.py 1 3 4 torchrec/distributed/planner/proposers.py torchrec/optim/keyed.py 1 3 8 torchrec/distributed/planner/proposers.py torchrec/distributed/cw_sharding.py 1 3 12 torchrec/distributed/planner/proposers.py torchrec/distributed/dist_data.py 1 3 14 torchrec/distributed/planner/proposers.py torchrec/distributed/dp_sharding.py 1 3 7 torchrec/distributed/planner/proposers.py torchrec/distributed/embedding_sharding.py 1 3 12 torchrec/distributed/planner/proposers.py torchrec/distributed/embedding_types.py 1 3 9 torchrec/distributed/planner/proposers.py torchrec/distributed/embeddingbag.py 1 3 9 torchrec/distributed/planner/proposers.py torchrec/distributed/rw_sharding.py 1 3 9 torchrec/distributed/planner/proposers.py torchrec/distributed/tw_sharding.py 1 3 9 torchrec/distributed/planner/proposers.py torchrec/distributed/twrw_sharding.py 1 3 9 torchrec/distributed/planner/proposers.py torchrec/distributed/types.py 1 3 10 torchrec/distributed/planner/proposers.py torchrec/distributed/comm_ops.py 1 3 7 torchrec/distributed/planner/proposers.py torchrec/distributed/train_pipeline.py 1 3 7 torchrec/distributed/planner/proposers.py torchrec/distributed/utils.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/__init__.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/datasets/utils.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/sparse/jagged_tensor.py 1 3 5 torchrec/distributed/planner/proposers.py torchrec/quant/embedding_modules.py 1 3 8 torchrec/distributed/planner/proposers.py setup.py 1 3 4 torchrec/distributed/planner/proposers.py torchrec/datasets/__init__.py 1 3 2 torchrec/distributed/planner/proposers.py torchrec/datasets/criteo.py 1 3 4 torchrec/distributed/planner/proposers.py torchrec/datasets/movielens.py 1 3 2 torchrec/distributed/planner/proposers.py torchrec/datasets/random.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/datasets/scripts/preprocess_criteo.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/__init__.py 1 3 2 torchrec/distributed/planner/proposers.py torchrec/distributed/collective_utils.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/comm.py 1 3 3 torchrec/distributed/planner/proposers.py torchrec/distributed/embedding.py 1 3 2 torchrec/fx/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/fx/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/fx/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/fx/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/fx/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/fx/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/fx/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/fx/__init__.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/fx/__init__.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/fx/tracer.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/fx/tracer.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/fx/tracer.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/fx/tracer.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/fx/tracer.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/fx/tracer.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/fx/tracer.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/fx/tracer.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/fx/tracer.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/linter/module_linter.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/linter/module_linter.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/linter/module_linter.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/linter/module_linter.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/linter/module_linter.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/linter/module_linter.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/linter/module_linter.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/linter/module_linter.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/linter/module_linter.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/embedding_lookup.py 1 2 16 torchrec/models/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/planner/utils.py 1 2 6 torchrec/models/__init__.py torchrec/distributed/grouped_position_weighted.py 1 2 4 torchrec/models/__init__.py torchrec/optim/keyed.py 1 2 8 torchrec/models/__init__.py torchrec/distributed/cw_sharding.py 1 2 12 torchrec/models/__init__.py torchrec/distributed/dist_data.py 1 2 14 torchrec/models/__init__.py torchrec/distributed/dp_sharding.py 1 2 7 torchrec/models/__init__.py torchrec/distributed/embedding_sharding.py 1 2 12 torchrec/models/__init__.py torchrec/distributed/embedding_types.py 1 2 9 torchrec/models/__init__.py torchrec/distributed/embeddingbag.py 1 2 9 torchrec/models/__init__.py torchrec/distributed/model_parallel.py 1 2 7 torchrec/models/__init__.py torchrec/distributed/rw_sharding.py 1 2 9 torchrec/models/__init__.py torchrec/distributed/tw_sharding.py 1 2 9 torchrec/models/__init__.py torchrec/distributed/twrw_sharding.py 1 2 9 torchrec/models/__init__.py torchrec/distributed/types.py 1 2 10 torchrec/models/__init__.py torchrec/distributed/comm_ops.py 1 2 7 torchrec/models/__init__.py torchrec/distributed/train_pipeline.py 1 2 7 torchrec/models/__init__.py torchrec/distributed/utils.py 1 2 3 torchrec/models/__init__.py torchrec/__init__.py 1 2 3 torchrec/models/__init__.py torchrec/datasets/utils.py 1 2 3 torchrec/models/__init__.py torchrec/sparse/jagged_tensor.py 1 2 5 torchrec/models/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/models/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/models/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/models/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/models/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/models/__init__.py torchrec/quant/embedding_modules.py 1 2 8 torchrec/models/__init__.py setup.py 1 2 4 torchrec/models/__init__.py torchrec/datasets/__init__.py 1 2 2 torchrec/models/__init__.py torchrec/datasets/criteo.py 1 2 4 torchrec/models/__init__.py torchrec/datasets/movielens.py 1 2 2 torchrec/models/__init__.py torchrec/datasets/random.py 1 2 3 torchrec/models/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/__init__.py 1 2 2 torchrec/models/__init__.py torchrec/distributed/collective_utils.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/comm.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/embedding.py 1 2 2 torchrec/models/__init__.py torchrec/distributed/planner/__init__.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/models/__init__.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/models/__init__.py torchrec/distributed/planner/types.py 1 2 6 torchrec/models/__init__.py torchrec/fx/__init__.py 1 2 2 torchrec/models/__init__.py torchrec/fx/tracer.py 1 2 3 torchrec/models/__init__.py torchrec/linter/module_linter.py 1 2 2 torchrec/models/deepfm.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/models/deepfm.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/models/deepfm.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/models/deepfm.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/models/deepfm.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/models/deepfm.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/models/deepfm.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/models/deepfm.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/models/deepfm.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/models/deepfm.py torchrec/models/__init__.py 1 3 2 torchrec/models/dlrm.py torchrec/distributed/planner/constants.py 1 4 3 torchrec/models/dlrm.py torchrec/distributed/planner/enumerators.py 1 4 5 torchrec/models/dlrm.py torchrec/distributed/planner/partitioners.py 1 4 3 torchrec/models/dlrm.py torchrec/distributed/planner/shard_estimators.py 1 4 6 torchrec/models/dlrm.py torchrec/distributed/planner/stats.py 1 4 6 torchrec/models/dlrm.py torchrec/distributed/planner/storage_reservations.py 1 4 4 torchrec/models/dlrm.py torchrec/distributed/planner/planners.py 1 4 9 torchrec/models/dlrm.py torchrec/distributed/planner/perf_models.py 1 4 2 torchrec/models/dlrm.py torchrec/distributed/planner/proposers.py 1 4 3 torchrec/modules/__init__.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/__init__.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/__init__.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/__init__.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/__init__.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/__init__.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/__init__.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/modules/__init__.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/__init__.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/__init__.py torchrec/models/__init__.py 1 3 2 torchrec/modules/activation.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/activation.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/activation.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/activation.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/activation.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/activation.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/activation.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/modules/activation.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/activation.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/crossnet.py torchrec/distributed/planner/constants.py 1 6 3 torchrec/modules/crossnet.py torchrec/distributed/planner/enumerators.py 1 6 5 torchrec/modules/crossnet.py torchrec/distributed/planner/partitioners.py 1 6 3 torchrec/modules/crossnet.py torchrec/distributed/planner/shard_estimators.py 1 6 6 torchrec/modules/crossnet.py torchrec/distributed/planner/stats.py 1 6 6 torchrec/modules/crossnet.py torchrec/distributed/planner/storage_reservations.py 1 6 4 torchrec/modules/crossnet.py torchrec/distributed/planner/planners.py 1 6 9 torchrec/modules/crossnet.py torchrec/distributed/planner/perf_models.py 1 6 2 torchrec/modules/crossnet.py torchrec/distributed/planner/proposers.py 1 6 3 torchrec/modules/deepfm.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/deepfm.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/deepfm.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/deepfm.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/deepfm.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/deepfm.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/deepfm.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/modules/deepfm.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/deepfm.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/embedding_configs.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/modules/embedding_configs.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/modules/embedding_configs.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/modules/embedding_configs.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/modules/embedding_configs.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/modules/embedding_configs.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/modules/embedding_configs.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/modules/embedding_configs.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/modules/embedding_configs.py torchrec/models/__init__.py 1 2 2 torchrec/modules/embedding_modules.py torchrec/distributed/planner/constants.py 1 4 3 torchrec/modules/embedding_modules.py torchrec/distributed/planner/enumerators.py 1 4 5 torchrec/modules/embedding_modules.py torchrec/distributed/planner/partitioners.py 1 4 3 torchrec/modules/embedding_modules.py torchrec/distributed/planner/shard_estimators.py 1 4 6 torchrec/modules/embedding_modules.py torchrec/distributed/planner/stats.py 1 4 6 torchrec/modules/embedding_modules.py torchrec/distributed/planner/storage_reservations.py 1 4 4 torchrec/modules/embedding_modules.py torchrec/distributed/planner/perf_models.py 1 4 2 torchrec/modules/embedding_modules.py torchrec/distributed/planner/proposers.py 1 4 3 torchrec/modules/embedding_modules.py torchrec/models/__init__.py 1 4 2 torchrec/modules/feature_processor.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/feature_processor.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/feature_processor.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/feature_processor.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/feature_processor.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/feature_processor.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/feature_processor.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/feature_processor.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/feature_processor.py torchrec/models/__init__.py 1 3 2 torchrec/modules/lazy_extension.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/modules/lazy_extension.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/modules/lazy_extension.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/modules/lazy_extension.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/modules/lazy_extension.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/modules/lazy_extension.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/modules/lazy_extension.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/modules/lazy_extension.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/modules/lazy_extension.py torchrec/models/__init__.py 1 2 2 torchrec/modules/mlp.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/modules/mlp.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/modules/mlp.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/modules/mlp.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/modules/mlp.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/modules/mlp.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/modules/mlp.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/modules/mlp.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/modules/mlp.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/modules/mlp.py torchrec/models/__init__.py 1 2 2 torchrec/modules/score_learning.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/score_learning.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/score_learning.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/score_learning.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/score_learning.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/score_learning.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/score_learning.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/score_learning.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/score_learning.py torchrec/models/__init__.py 1 3 2 torchrec/modules/utils.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/modules/utils.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/modules/utils.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/modules/utils.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/modules/utils.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/modules/utils.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/modules/utils.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/modules/utils.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/modules/utils.py torchrec/models/__init__.py 1 3 2 torchrec/optim/__init__.py torchrec/distributed/planner/constants.py 1 3 3 torchrec/optim/__init__.py torchrec/distributed/planner/enumerators.py 1 3 5 torchrec/optim/__init__.py torchrec/distributed/planner/partitioners.py 1 3 3 torchrec/optim/__init__.py torchrec/distributed/planner/shard_estimators.py 1 3 6 torchrec/optim/__init__.py torchrec/distributed/planner/stats.py 1 3 6 torchrec/optim/__init__.py torchrec/distributed/planner/storage_reservations.py 1 3 4 torchrec/optim/__init__.py torchrec/distributed/planner/planners.py 1 3 9 torchrec/optim/__init__.py torchrec/distributed/planner/perf_models.py 1 3 2 torchrec/optim/__init__.py torchrec/distributed/planner/proposers.py 1 3 3 torchrec/optim/clipping.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/optim/clipping.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/optim/clipping.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/optim/clipping.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/optim/clipping.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/optim/clipping.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/optim/clipping.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/optim/clipping.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/optim/clipping.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/optim/clipping.py torchrec/models/__init__.py 1 2 2 torchrec/optim/fused.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/optim/fused.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/optim/fused.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/optim/fused.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/optim/fused.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/optim/fused.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/optim/fused.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/optim/fused.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/optim/fused.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/optim/fused.py torchrec/models/__init__.py 1 2 2 torchrec/optim/warmup.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/optim/warmup.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/optim/warmup.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/optim/warmup.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/optim/warmup.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/optim/warmup.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/optim/warmup.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/optim/warmup.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/optim/warmup.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/optim/warmup.py torchrec/models/__init__.py 1 2 2 torchrec/quant/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/quant/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/quant/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/quant/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/quant/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/quant/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/quant/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/quant/__init__.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/quant/__init__.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/quant/__init__.py torchrec/models/__init__.py 1 2 2 torchrec/sparse/__init__.py torchrec/distributed/embedding_lookup.py 1 2 16 torchrec/sparse/__init__.py torchrec/distributed/planner/constants.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/planner/utils.py 1 2 6 torchrec/sparse/__init__.py torchrec/distributed/grouped_position_weighted.py 1 2 4 torchrec/sparse/__init__.py torchrec/optim/keyed.py 1 2 8 torchrec/sparse/__init__.py torchrec/distributed/cw_sharding.py 1 2 12 torchrec/sparse/__init__.py torchrec/distributed/dist_data.py 1 2 14 torchrec/sparse/__init__.py torchrec/distributed/dp_sharding.py 1 2 7 torchrec/sparse/__init__.py torchrec/distributed/embedding_sharding.py 1 2 12 torchrec/sparse/__init__.py torchrec/distributed/embedding_types.py 1 2 9 torchrec/sparse/__init__.py torchrec/distributed/embeddingbag.py 1 2 9 torchrec/sparse/__init__.py torchrec/distributed/model_parallel.py 1 2 7 torchrec/sparse/__init__.py torchrec/distributed/rw_sharding.py 1 2 9 torchrec/sparse/__init__.py torchrec/distributed/tw_sharding.py 1 2 9 torchrec/sparse/__init__.py torchrec/distributed/twrw_sharding.py 1 2 9 torchrec/sparse/__init__.py torchrec/distributed/types.py 1 2 10 torchrec/sparse/__init__.py torchrec/distributed/comm_ops.py 1 2 7 torchrec/sparse/__init__.py torchrec/distributed/train_pipeline.py 1 2 7 torchrec/sparse/__init__.py torchrec/distributed/utils.py 1 2 3 torchrec/sparse/__init__.py torchrec/__init__.py 1 2 3 torchrec/sparse/__init__.py torchrec/datasets/utils.py 1 2 3 torchrec/sparse/__init__.py torchrec/sparse/jagged_tensor.py 1 2 5 torchrec/sparse/__init__.py torchrec/distributed/planner/enumerators.py 1 2 5 torchrec/sparse/__init__.py torchrec/distributed/planner/partitioners.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/planner/shard_estimators.py 1 2 6 torchrec/sparse/__init__.py torchrec/distributed/planner/stats.py 1 2 6 torchrec/sparse/__init__.py torchrec/distributed/planner/storage_reservations.py 1 2 4 torchrec/sparse/__init__.py torchrec/distributed/planner/planners.py 1 2 9 torchrec/sparse/__init__.py torchrec/quant/embedding_modules.py 1 2 8 torchrec/sparse/__init__.py setup.py 1 2 4 torchrec/sparse/__init__.py torchrec/datasets/__init__.py 1 2 2 torchrec/sparse/__init__.py torchrec/datasets/criteo.py 1 2 4 torchrec/sparse/__init__.py torchrec/datasets/movielens.py 1 2 2 torchrec/sparse/__init__.py torchrec/datasets/random.py 1 2 3 torchrec/sparse/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/__init__.py 1 2 2 torchrec/sparse/__init__.py torchrec/distributed/collective_utils.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/comm.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/embedding.py 1 2 2 torchrec/sparse/__init__.py torchrec/distributed/planner/__init__.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/planner/perf_models.py 1 2 2 torchrec/sparse/__init__.py torchrec/distributed/planner/proposers.py 1 2 3 torchrec/sparse/__init__.py torchrec/distributed/planner/types.py 1 2 6 torchrec/sparse/__init__.py torchrec/fx/__init__.py 1 2 2 torchrec/sparse/__init__.py torchrec/fx/tracer.py 1 2 3 torchrec/sparse/__init__.py torchrec/linter/module_linter.py 1 2 2 torchrec/sparse/__init__.py torchrec/models/deepfm.py 1 2 3 torchrec/sparse/__init__.py torchrec/modules/__init__.py 1 2 3 torchrec/sparse/__init__.py torchrec/modules/embedding_configs.py 1 2 2 torchrec/sparse/__init__.py torchrec/modules/embedding_modules.py 1 2 4 torchrec/sparse/__init__.py torchrec/modules/feature_processor.py 1 2 3 torchrec/sparse/__init__.py torchrec/modules/lazy_extension.py 1 2 2 torchrec/sparse/__init__.py torchrec/modules/mlp.py 1 2 2 torchrec/sparse/__init__.py torchrec/modules/score_learning.py 1 2 3 torchrec/sparse/__init__.py torchrec/modules/utils.py 1 2 3 torchrec/sparse/__init__.py torchrec/optim/clipping.py 1 2 2 torchrec/sparse/__init__.py torchrec/optim/fused.py 1 2 2 torchrec/sparse/__init__.py torchrec/optim/warmup.py 1 2 2 torchrec/sparse/__init__.py torchrec/quant/__init__.py 1 2 2 torchrec/datasets/scripts/__init__.py torchrec/datasets/criteo.py 1 1 4 torchrec/datasets/scripts/__init__.py torchrec/datasets/scripts/preprocess_criteo.py 1 1 3