索引 A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | Z A activation (torchrec.modules.embedding_configs.QuantConfig 属性) ADAGRAD (torchrec.distributed.embedding_types.OptimType 属性) ADAM (torchrec.distributed.embedding_types.OptimType 属性) ADAMW (torchrec.distributed.embedding_types.OptimType 属性) add_param_group() (torchrec.optim.keyed.KeyedOptimizer 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) add_params_from_parameter_sharding() (在模块 torchrec.distributed.utils 中) add_prefix_to_state_dict() (在模块 torchrec.distributed.utils 中) algorithm (torchrec.distributed.types.CacheParams 属性), [1] All2All_Pooled_Req (torchrec.distributed.comm_ops 中的类) all2all_pooled_sync() (在模块 torchrec.distributed.comm_ops 中) All2All_Pooled_Wait (torchrec.distributed.comm_ops 中的类) All2All_Seq_Req (torchrec.distributed.comm_ops 中的类) All2All_Seq_Req_Wait (torchrec.distributed.comm_ops 中的类) all2all_sequence_sync() (在模块 torchrec.distributed.comm_ops 中) All2AllDenseInfo (torchrec.distributed.comm_ops 中的类) All2AllPooledInfo (torchrec.distributed.comm_ops 中的类) All2AllSequenceInfo (torchrec.distributed.comm_ops 中的类) All2Allv_Req (torchrec.distributed.comm_ops 中的类) all2allv_sync() (在模块 torchrec.distributed.comm_ops 中) All2Allv_Wait (torchrec.distributed.comm_ops 中的类) All2AllVInfo (torchrec.distributed.comm_ops 中的类) all_gather_base_pooled() (在模块 torchrec.distributed.comm_ops 中) all_gather_base_sync() (在模块 torchrec.distributed.comm_ops 中) all_gather_into_tensor_backward() (在模块 torchrec.distributed.comm_ops 中) all_gather_into_tensor_fake() (在模块 torchrec.distributed.comm_ops 中) all_gather_into_tensor_setup_context() (在模块 torchrec.distributed.comm_ops 中) all_to_all_single_backward() (在模块 torchrec.distributed.comm_ops 中) all_to_all_single_fake() (在模块 torchrec.distributed.comm_ops 中) all_to_all_single_setup_context() (在模块 torchrec.distributed.comm_ops 中) AllGatherBase_Req (torchrec.distributed.comm_ops 中的类) AllGatherBase_Wait (torchrec.distributed.comm_ops 中的类) AllGatherBaseInfo (torchrec.distributed.comm_ops 中的类) allocate_budget() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) alltoall_pooled() (在模块 torchrec.distributed.comm_ops 中) alltoall_sequence() (在模块 torchrec.distributed.comm_ops 中) alltoallv() (在模块 torchrec.distributed.comm_ops 中) append_prefix() (在模块 torchrec.distributed.utils 中) apply() (torchrec.modules.lazy_extension.LazyModuleExtensionMixin 方法) apply_feature_processor() (torchrec.distributed.quant_embeddingbag.ShardedQuantFeatureProcessedEmbeddingBagCollection 方法) apply_mc_method_to_jt_dict() (在模块 torchrec.modules.mc_modules 中) average_threshold_filter() (在模块 torchrec.modules.mc_modules 中) Awaitable (torchrec.distributed.types 中的类) B B_global (torchrec.distributed.comm_ops.All2AllVInfo 属性), [1] B_local (torchrec.distributed.comm_ops.All2AllVInfo 属性), [1] B_local_list (torchrec.distributed.comm_ops.All2AllVInfo 属性), [1] backward (torchrec.distributed.types.QuantizedCommCodecs 属性) backward() (torchrec.distributed.comm_ops.All2All_Pooled_Req 静态方法) (torchrec.distributed.comm_ops.All2All_Pooled_Wait 静态方法) (torchrec.distributed.comm_ops.All2All_Seq_Req 静态方法) (torchrec.distributed.comm_ops.All2All_Seq_Req_Wait 静态方法) (torchrec.distributed.comm_ops.All2Allv_Req 静态方法) (torchrec.distributed.comm_ops.All2Allv_Wait 静态方法) (torchrec.distributed.comm_ops.AllGatherBase_Req 静态方法) (torchrec.distributed.comm_ops.AllGatherBase_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatter_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatter_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatterBase_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatterBase_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatterV_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatterV_Wait 静态方法) (torchrec.distributed.comm_ops.Variable_Batch_All2All_Pooled_Req 静态方法) (torchrec.distributed.comm_ops.Variable_Batch_All2All_Pooled_Wait 静态方法) (torchrec.distributed.embedding_lookup.CommOpGradientScaling 静态方法) backward_recat_tensor (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] bare_named_parameters() (torchrec.distributed.model_parallel.DistributedModelParallel 方法) BaseCwEmbeddingSharding (torchrec.distributed.sharding.cw_sharding 中的类) BaseDpEmbeddingSharding (torchrec.distributed.sharding.dp_sharding 中的类) BaseEmbeddingConfig (torchrec.modules.embedding_configs 中的类) BaseEmbeddingDist (torchrec.distributed.embedding_sharding 中的类) BaseEmbeddingLookup (torchrec.distributed.embedding_types 中的类) BaseEmbeddingSharder (torchrec.distributed.embedding_types 中的类) BaseFeatureProcessor (torchrec.modules.feature_processor 中的类) BaseGroupedFeatureProcessor (torchrec.distributed.embedding_types 中的类) (torchrec.modules.feature_processor 中的类) BaseManagedCollisionEmbeddingCollection (torchrec.modules.mc_embedding_modules 中的类) BaseQuantEmbeddingSharder (torchrec.distributed.embedding_types 中的类) BaseRwEmbeddingSharding (torchrec.distributed.sharding.rw_sharding 中的类) BaseSparseFeaturesDist (torchrec.distributed.embedding_sharding 中的类) BaseTwEmbeddingSharding (torchrec.distributed.sharding.tw_sharding 中的类) BaseTwRwEmbeddingSharding (torchrec.distributed.sharding.twrw_sharding 中的类) batch_size (torchrec.distributed.comm_ops.All2AllDenseInfo 属性) (torchrec.distributed.planner.types.ShardingOption 属性) batch_size_per_feature_pre_a2a (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性), [1] batch_size_per_rank (torchrec.distributed.comm_ops.All2AllPooledInfo 属性), [1] batch_size_per_rank_per_feature (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性), [1] batch_sizes (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] batching_metadata() (torchrec.inference.modules.PredictFactory 方法) batching_metadata_json() (torchrec.inference.modules.PredictFactory 方法) BatchingMetadata (torchrec.inference.modules 中的类) best() (torchrec.distributed.planner.utils.LuusJaakolaSearch 方法) BinarySearchPredicate (torchrec.distributed.planner.utils 中的类) bounds_check_mode (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) bucket_mapping_tensor (torchrec.distributed.embedding_types.InputDistOutputs 属性) bucketize_kjt_before_all2all() (在模块 torchrec.distributed.embedding_sharding 中) bucketize_kjt_inference() (在模块 torchrec.distributed.embedding_sharding 中) bucketized_length (torchrec.distributed.embedding_types.InputDistOutputs 属性) build_affine_storage_model() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) bwd_comms (torchrec.distributed.planner.types.Perf 属性) bwd_compute (torchrec.distributed.planner.types.Perf 属性) bwd_compute_multiplier (torchrec.distributed.planner.types.Topology 属性) bytes_to_gb() (在模块 torchrec.distributed.planner.utils 中) bytes_to_mb() (在模块 torchrec.distributed.planner.utils 中) C cache_load_factor (torchrec.distributed.planner.types.ShardingOption 属性) cache_params (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) cacheability (torchrec.distributed.planner.shard_estimators.EmbeddingOffloadStats 属性) (torchrec.distributed.types.CacheStatistics 属性) CacheParams (torchrec.distributed.types 中的类) CacheStatistics (torchrec.distributed.types 中的类) calc_quantized_size() (torchrec.distributed.types.NoOpQuantizedCommCodec 方法) (torchrec.distributed.types.QuantizedCommCodec 方法) calculate_pipeline_io_cost() (在模块 torchrec.distributed.planner.shard_estimators 中) calculate_shard_storages() (在模块 torchrec.distributed.planner.shard_estimators 中) callbacks (torchrec.distributed.dist_data.PooledEmbeddingsAllToAll 属性), [1] (torchrec.distributed.dist_data.PooledEmbeddingsAwaitable 属性), [1] (torchrec.distributed.dist_data.VariableBatchPooledEmbeddingsAllToAll 属性), [1] (torchrec.distributed.types.Awaitable 属性) check_module_output_dimension() (在模块 torchrec.modules.utils 中) clamp() (torchrec.distributed.planner.utils.LuusJaakolaSearch 方法) clf_to_bytes() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) coalesce_history_metadata() (torchrec.modules.mc_modules.DistanceLFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LRU_EvictionPolicy 方法) (torchrec.modules.mc_modules.MCHEvictionPolicy 方法) codecs (torchrec.distributed.comm_ops.All2AllPooledInfo 属性), [1] (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] (torchrec.distributed.comm_ops.All2AllVInfo 属性) (torchrec.distributed.comm_ops.AllGatherBaseInfo 属性) (torchrec.distributed.comm_ops.ReduceScatterBaseInfo 属性) (torchrec.distributed.comm_ops.ReduceScatterInfo 属性) (torchrec.distributed.comm_ops.ReduceScatterVInfo 属性), [1] (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性), [1] collective_plan() (torchrec.distributed.planner.planners.EmbeddingShardingPlanner 方法) (torchrec.distributed.planner.planners.HeteroEmbeddingShardingPlanner 方法) (torchrec.distributed.types.ShardingPlanner 方法) COLUMN_WISE (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) CombinedOptimizer (torchrec.optim.keyed 中的类) CommOp (torchrec.distributed.types 中的类) CommOpGradientScaling (torchrec.distributed.embedding_lookup 中的类) compute() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantFeatureProcessedEmbeddingBagCollection 方法) (torchrec.distributed.types.ShardedModule 方法) compute_and_output_dist() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.types.ShardedModule 方法) compute_device (torchrec.distributed.planner.types.Topology 属性) compute_kernel (torchrec.distributed.embedding_types.EmbeddingAttributes 属性) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) compute_kernel_to_embedding_location() (在模块 torchrec.distributed.embedding_types 中) compute_kernels (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] compute_kernels() (torchrec.distributed.embedding_types.BaseEmbeddingSharder 方法) (torchrec.distributed.embedding_types.BaseQuantEmbeddingSharder 方法) (torchrec.distributed.quant_embeddingbag.QuantFeatureProcessedEmbeddingBagCollectionSharder 方法) (torchrec.distributed.types.ModuleSharder 方法) ComputeJTDictToKJT (torchrec.sparse.jagged_tensor 中的类) ComputeKernel (torchrec.distributed.types 中的类) ComputeKJTToJTDict (torchrec.sparse.jagged_tensor 中的类) concat() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) CONSTANT (torchrec.optim.warmup.WarmupPolicy 属性) construct_jagged_tensors() (在模块 torchrec.modules.utils 中) construct_jagged_tensors_inference() (在模块 torchrec.modules.utils 中) construct_modulelist_from_single_module() (在模块 torchrec.modules.utils 中) construct_output_kt() (在模块 torchrec.distributed.embeddingbag 中) convert_list_of_modules_to_modulelist() (在模块 torchrec.modules.utils 中) convert_to_fbgemm_types() (在模块 torchrec.distributed.utils 中) copy() (torchrec.distributed.model_parallel.DistributedModelParallel 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.utils.CopyableMixin 方法) copy_to_device() (在模块 torchrec.distributed.utils 中) CopyableMixin (torchrec.distributed.utils 中的类) COSINE_ANNEALING_WARM_RESTARTS (torchrec.optim.warmup.WarmupPolicy 属性) create_arg() (torchrec.fx.tracer.Tracer 方法) create_context() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.distributed.mc_embedding.ShardedManagedCollisionEmbeddingCollection 方法) (torchrec.distributed.mc_embeddingbag.ShardedManagedCollisionEmbeddingBagCollection 方法) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.types.NoOpQuantizedCommCodec 方法) (torchrec.distributed.types.QuantizedCommCodec 方法) (torchrec.distributed.types.ShardedModule 方法) create_embedding_bag_sharding() (在模块 torchrec.distributed.embeddingbag 中) create_embedding_sharding() (在模块 torchrec.distributed.embedding 中) create_infer_embedding_bag_sharding() (在模块 torchrec.distributed.quant_embeddingbag 中) create_input_dist() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.CwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.DpPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.InferRwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.RwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.InferTwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.TwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.TwRwPooledEmbeddingSharding 方法) create_lookup() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.CwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.DpPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.InferRwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.RwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.InferTwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.TwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.TwRwPooledEmbeddingSharding 方法) create_mc_sharding() (在模块 torchrec.distributed.mc_modules 中) create_output_dist() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.CwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.DpPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.InferRwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.RwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.InferTwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.TwPooledEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.TwRwPooledEmbeddingSharding 方法) create_predict_module() (torchrec.inference.modules.PredictFactory 方法) create_sharding_infos_by_sharding() (在模块 torchrec.distributed.embedding 中) (在模块 torchrec.distributed.embeddingbag 中) create_sharding_infos_by_sharding_device_group() (在模块 torchrec.distributed.embedding 中) (在模块 torchrec.distributed.embeddingbag 中) CrossNet (torchrec.modules.crossnet 中的类) cumsum_dim_sum_per_rank_tensor (torchrec.distributed.comm_ops.All2AllPooledInfo 属性), [1] CustomTopologyData (torchrec.distributed.planner.types 中的类) CwPooledEmbeddingSharding (torchrec.distributed.sharding.cw_sharding 中的类) D D_local_list (torchrec.distributed.comm_ops.All2AllVInfo 属性), [1] DATA_PARALLEL (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) data_type (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) data_type_to_dtype() (在模块 torchrec.modules.embedding_configs 中) data_type_to_sparse_type() (在模块 torchrec.modules.embedding_configs 中) DataParallelWrapper (torchrec.distributed.model_parallel 中的类) ddr (torchrec.distributed.planner.types.Storage 属性) DDR (torchrec.distributed.types.ParameterStorage 属性) ddr_mem_bw (torchrec.distributed.planner.types.Topology 属性) decay_iters (torchrec.optim.warmup.WarmupStage 属性) decode() (torchrec.distributed.types.NoOpQuantizedCommCodec 方法) (torchrec.distributed.types.QuantizedCommCodec 方法) DeepFM (torchrec.modules.deepfm 中的类) DEFAULT (torchrec.distributed.types.ComputeKernel 属性) DefaultDataParallelWrapper (torchrec.distributed.model_parallel 中的类) DENSE (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) DenseArch (torchrec.models.deepfm 中的类) dependency (torchrec.distributed.planner.types.ShardingOption 属性) deterministic_dedup() (在模块 torchrec.modules.utils 中) device (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) (torchrec.distributed.planner.partitioners.OrderedDeviceHardware 属性) DEVICE (torchrec.distributed.planner.types.PartitionByType 属性) device (torchrec.inference.modules.BatchingMetadata 属性) (torchrec.modules.embedding_modules.EmbeddingBagCollection 属性) (torchrec.modules.embedding_modules.EmbeddingCollection 属性) (torchrec.modules.mc_modules.ManagedCollisionModule 属性) (torchrec.quant.embedding_modules.EmbeddingBagCollection 属性) (torchrec.quant.embedding_modules.EmbeddingCollection 属性) device() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) device_group (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] DeviceHardware (torchrec.distributed.planner.types 中的类) devices (torchrec.distributed.planner.types.Topology 属性) dim_sum() (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) dim_sum_per_rank (torchrec.distributed.comm_ops.All2AllPooledInfo 属性), [1] (torchrec.distributed.comm_ops.All2AllVInfo 属性) dim_sum_per_rank_tensor (torchrec.distributed.comm_ops.All2AllPooledInfo 属性), [1] dims_sum_per_rank (torchrec.distributed.comm_ops.All2AllVInfo 属性) dist_init() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) dist_labels() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) dist_splits() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) dist_tensors() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) DistanceLFU_EvictionPolicy (torchrec.modules.mc_modules 中的类) DistributedModelParallel (torchrec.distributed.model_parallel 中的类) divisor (torchrec.distributed.embeddingbag.EmbeddingBagCollectionContext 属性) DpPooledEmbeddingDist (torchrec.distributed.sharding.dp_sharding 中的类) DpPooledEmbeddingSharding (torchrec.distributed.sharding.dp_sharding 中的类) DpSparseFeaturesDist (torchrec.distributed.sharding.dp_sharding 中的类) dtensor_metadata (torchrec.distributed.embedding_types.ShardedMetaConfig 属性) DTensorMetadata (torchrec.distributed.embedding_types 中的类) dtype_to_data_type() (在模块 torchrec.modules.embedding_configs 中) dummy_tensor() (在模块 torchrec.distributed.embedding_lookup 中) dynamic_threshold_filter() (在模块 torchrec.modules.mc_modules 中) E emb_dim_per_rank_per_feature (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性), [1] embedding_bag_configs() (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollectionInterface 方法) (torchrec.quant.embedding_modules.EmbeddingBagCollection 方法) embedding_bags (torchrec.distributed.quant_embeddingbag.ShardedQuantFeatureProcessedEmbeddingBagCollection 属性) (torchrec.quant.embedding_modules.FeatureProcessedEmbeddingBagCollection 属性) embedding_config (torchrec.distributed.embedding_sharding.EmbeddingShardingInfo 属性) embedding_configs() (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 方法) (torchrec.modules.mc_modules.ManagedCollisionCollection 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) embedding_dim (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) (torchrec.modules.embedding_configs.EmbeddingConfig 属性) embedding_dim() (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) embedding_dims() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) (torchrec.distributed.sharding.cw_sharding.BaseCwEmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.BaseTwRwEmbeddingSharding 方法) embedding_names (torchrec.modules.embedding_configs.EmbeddingTableConfig 属性) embedding_names() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) (torchrec.distributed.sharding.cw_sharding.BaseCwEmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.BaseTwRwEmbeddingSharding 方法) embedding_names_by_table() (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) embedding_names_per_rank() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.BaseTwRwEmbeddingSharding 方法) embedding_shard_metadata() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.BaseTwRwEmbeddingSharding 方法) embedding_tables (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) embedding_tables() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) EmbeddingAttributes (torchrec.distributed.embedding_types 中的类) EmbeddingAwaitable (torchrec.distributed.embeddingbag 中的类) EmbeddingBagCollection (torchrec.modules.embedding_modules 中的类) (torchrec.quant.embedding_modules 中的类) EmbeddingBagCollectionAwaitable (torchrec.distributed.embeddingbag 中的类) EmbeddingBagCollectionContext (torchrec.distributed.embeddingbag 中的类) EmbeddingBagCollectionInterface (torchrec.modules.embedding_modules 中的类) EmbeddingBagCollectionSharder (torchrec.distributed.embeddingbag 中的类) EmbeddingBagConfig (torchrec.modules.embedding_configs 中的类) EmbeddingBagSharder (torchrec.distributed.embeddingbag 中的类) EmbeddingCollection (torchrec.modules.embedding_modules 中的类) (torchrec.quant.embedding_modules 中的类) EmbeddingCollectionAwaitable (torchrec.distributed.embedding 中的类) EmbeddingCollectionContext (torchrec.distributed.embedding 中的类) EmbeddingCollectionInterface (torchrec.modules.embedding_modules 中的类) EmbeddingCollectionSharder (torchrec.distributed.embedding 中的类) EmbeddingComputeKernel (torchrec.distributed.embedding_types 中的类) EmbeddingConfig (torchrec.modules.embedding_configs 中的类) EmbeddingEnumerator (torchrec.distributed.planner.enumerators 中的类) EmbeddingModuleShardingPlan (torchrec.distributed.types 中的类) EmbeddingOffloadScaleupProposer (torchrec.distributed.planner.proposers 中的类) EmbeddingOffloadStats (torchrec.distributed.planner.shard_estimators 中的类) EmbeddingPerfEstimator (torchrec.distributed.planner.shard_estimators 中的类) embeddings_cat_empty_rank_handle() (在模块 torchrec.distributed.embedding_lookup 中) embeddings_cat_empty_rank_handle_inference() (在模块 torchrec.distributed.embedding_lookup 中) EmbeddingsAllToOne (torchrec.distributed.dist_data 类)，[1] EmbeddingsAllToOneReduce (torchrec.distributed.dist_data 类)，[1] EmbeddingSharding (torchrec.distributed.embedding_sharding 类) EmbeddingShardingContext (torchrec.distributed.embedding_sharding 类) EmbeddingShardingInfo (torchrec.distributed.embedding_sharding 类) EmbeddingShardingPlanner (torchrec.distributed.planner.planners 类) EmbeddingStats (torchrec.distributed.planner.stats 类) EmbeddingStorageEstimator (torchrec.distributed.planner.shard_estimators 类) EmbeddingTableConfig (torchrec.modules.embedding_configs 类) empty() (torchrec.sparse.jagged_tensor.JaggedTensor 静态方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) empty_like() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) EmptyFusedOptimizer (torchrec.optim.fused 类) encode() (torchrec.distributed.types.NoOpQuantizedCommCodec 方法) (torchrec.distributed.types.QuantizedCommCodec 方法) enforce_hbm (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) enumerate() (torchrec.distributed.planner.enumerators.EmbeddingEnumerator 方法) (torchrec.distributed.planner.types.Enumerator 方法) Enumerator (torchrec.distributed.planner.types 类) equal_splits (torchrec.distributed.comm_ops.ReduceScatterVInfo 属性)，[1] estimate() (torchrec.distributed.planner.shard_estimators.EmbeddingPerfEstimator 方法) (torchrec.distributed.planner.shard_estimators.EmbeddingStorageEstimator 方法) (torchrec.distributed.planner.types.ShardEstimator 方法) estimate_cache_miss_rate() (torchrec.distributed.planner.shard_estimators.EmbeddingOffloadStats 静态方法) evict() (在 torchrec.modules.mc_embedding_modules 模块中) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.modules.mc_modules.ManagedCollisionCollection 方法) (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) evictions_per_table (torchrec.distributed.mc_embeddingbag.ManagedCollisionEmbeddingBagCollectionContext 属性) expected_lookups (torchrec.distributed.planner.shard_estimators.EmbeddingOffloadStats 属性) (torchrec.distributed.types.CacheStatistics 属性) expected_miss_rate() (torchrec.distributed.planner.shard_estimators.EmbeddingOffloadStats 方法) (torchrec.distributed.types.CacheStatistics 方法) extra_repr() (torchrec.distributed.embedding_types.ShardedEmbeddingModule 方法) extract_module_or_tensor_callable() (在 torchrec.modules.utils 模块中) F FactorizationMachine (torchrec.modules.deepfm 类) feature_hash_sizes() (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) feature_names (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) (torchrec.modules.embedding_configs.EmbeddingConfig 属性) feature_names() (torchrec.distributed.embedding_types.FeatureShardingMixIn 方法) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) (torchrec.distributed.sharding.dp_sharding.BaseDpEmbeddingSharding 方法) (torchrec.distributed.sharding.rw_sharding.BaseRwEmbeddingSharding 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) (torchrec.distributed.sharding.twrw_sharding.BaseTwRwEmbeddingSharding 方法) feature_names_per_rank() (torchrec.distributed.embedding_types.FeatureShardingMixIn 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) FeatureProcessedEmbeddingBagCollection (torchrec.quant.embedding_modules 类) features (torchrec.distributed.embedding_types.InputDistOutputs 属性) features_per_rank() (torchrec.distributed.embedding_types.FeatureShardingMixIn 方法) (torchrec.distributed.sharding.tw_sharding.BaseTwEmbeddingSharding 方法) FeatureShardingMixIn (torchrec.distributed.embedding_types 类) feedback() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 方法) (torchrec.distributed.planner.proposers.GreedyProposer 方法) (torchrec.distributed.planner.proposers.GridSearchProposer 方法) (torchrec.distributed.planner.proposers.UniformProposer 方法) (torchrec.distributed.planner.types.Proposer 方法) filter_state_dict() (在 torchrec.distributed.utils 模块中) fits_in() (torchrec.distributed.planner.types.Storage 方法) FixedPercentageStorageReservation (torchrec.distributed.planner.storage_reservations 类) flatten_feature_lengths() (在 torchrec.distributed.quant_embeddingbag 模块中) flatten_kjt_list() (在 torchrec.sparse.jagged_tensor 模块中) flatten_lengths() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) flush() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) FMInteractionArch (torchrec.models.deepfm 类) for_each_module_of_type_do() (在 torchrec.quant.embedding_modules 模块中) ForkedPdb (torchrec.distributed.utils 类) forward (torchrec.distributed.types.QuantizedCommCodecs 属性) forward() (torchrec.distributed.comm_ops.All2All_Pooled_Req 静态方法) (torchrec.distributed.comm_ops.All2All_Pooled_Wait 静态方法) (torchrec.distributed.comm_ops.All2All_Seq_Req 静态方法) (torchrec.distributed.comm_ops.All2All_Seq_Req_Wait 静态方法) (torchrec.distributed.comm_ops.All2Allv_Req 静态方法) (torchrec.distributed.comm_ops.All2Allv_Wait 静态方法) (torchrec.distributed.comm_ops.AllGatherBase_Req 静态方法) (torchrec.distributed.comm_ops.AllGatherBase_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatter_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatter_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatterBase_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatterBase_Wait 静态方法) (torchrec.distributed.comm_ops.ReduceScatterV_Req 静态方法) (torchrec.distributed.comm_ops.ReduceScatterV_Wait 静态方法) (torchrec.distributed.comm_ops.Variable_Batch_All2All_Pooled_Req 静态方法) (torchrec.distributed.comm_ops.Variable_Batch_All2All_Pooled_Wait 静态方法) (torchrec.distributed.dist_data.EmbeddingsAllToOne 方法)，[1] (torchrec.distributed.dist_data.EmbeddingsAllToOneReduce 方法)，[1] (torchrec.distributed.dist_data.KJTAllToAll 方法)，[1] (torchrec.distributed.dist_data.KJTOneToAll 方法)，[1] (torchrec.distributed.dist_data.MergePooledEmbeddingsModule 方法)，[1] (torchrec.distributed.dist_data.PooledEmbeddingsAllGather 方法)，[1] (torchrec.distributed.dist_data.PooledEmbeddingsAllToAll 方法)，[1] (torchrec.distributed.dist_data.PooledEmbeddingsReduceScatter 方法)，[1] (torchrec.distributed.dist_data.SeqEmbeddingsAllToOne 方法)，[1] (torchrec.distributed.dist_data.SequenceEmbeddingsAllToAll 方法)，[1] (torchrec.distributed.dist_data.TensorAllToAll 方法)，[1] (torchrec.distributed.dist_data.TensorValuesAllToAll 方法)，[1] (torchrec.distributed.dist_data.VariableBatchPooledEmbeddingsAllToAll 方法)，[1] (torchrec.distributed.dist_data.VariableBatchPooledEmbeddingsReduceScatter 方法)，[1] (torchrec.distributed.embedding_lookup.CommOpGradientScaling 静态方法) (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedLookupMixin 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_sharding.BaseEmbeddingDist 方法) (torchrec.distributed.embedding_sharding.BaseSparseFeaturesDist 方法) (torchrec.distributed.embedding_types.BaseEmbeddingLookup 方法) (torchrec.distributed.embedding_types.BaseGroupedFeatureProcessor 方法) (torchrec.distributed.grouped_position_weighted.GroupedPositionWeightedModule 方法) (torchrec.distributed.model_parallel.DistributedModelParallel 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEbcInputDist 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingDistWithPermute 方法) (torchrec.distributed.sharding.dp_sharding.DpPooledEmbeddingDist 方法) (torchrec.distributed.sharding.dp_sharding.DpSparseFeaturesDist 方法) (torchrec.distributed.sharding.rw_sharding.InferRwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.rw_sharding.InferRwSparseFeaturesDist 方法) (torchrec.distributed.sharding.rw_sharding.RwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.rw_sharding.RwSparseFeaturesDist 方法) (torchrec.distributed.sharding.tw_sharding.InferTwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.tw_sharding.InferTwSparseFeaturesDist 方法) (torchrec.distributed.sharding.tw_sharding.TwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.tw_sharding.TwSparseFeaturesDist 方法) (torchrec.distributed.sharding.twrw_sharding.TwRwPooledEmbeddingDist 方法) (torchrec.distributed.sharding.twrw_sharding.TwRwSparseFeaturesDist 方法) (torchrec.distributed.types.ShardedModule 方法) (torchrec.inference.modules.PredictModule 方法) (torchrec.models.deepfm.DenseArch 方法) (torchrec.models.deepfm.FMInteractionArch 方法) (torchrec.models.deepfm.OverArch 方法) (torchrec.models.deepfm.SimpleDeepFMNN 方法) (torchrec.models.deepfm.SparseArch 方法) (torchrec.modules.activation.SwishLayerNorm 方法) (torchrec.modules.crossnet.CrossNet 方法) (torchrec.modules.crossnet.LowRankCrossNet 方法) (torchrec.modules.crossnet.LowRankMixtureCrossNet 方法) (torchrec.modules.crossnet.VectorCrossNet 方法) (torchrec.modules.deepfm.DeepFM 方法) (torchrec.modules.deepfm.FactorizationMachine 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollectionInterface 方法) (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 方法) (torchrec.modules.feature_processor.BaseFeatureProcessor 方法) (torchrec.modules.feature_processor.BaseGroupedFeatureProcessor 方法) (torchrec.modules.feature_processor.PositionWeightedModule 方法) (torchrec.modules.feature_processor.PositionWeightedProcessor 方法) (torchrec.modules.mc_embedding_modules.BaseManagedCollisionEmbeddingCollection 方法) (torchrec.modules.mc_modules.ManagedCollisionCollection 方法) (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) (torchrec.modules.mlp.MLP 方法) (torchrec.modules.mlp.Perceptron 方法) (torchrec.quant.embedding_modules.EmbeddingBagCollection 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) (torchrec.quant.embedding_modules.FeatureProcessedEmbeddingBagCollection 方法) (torchrec.sparse.jagged_tensor.ComputeJTDictToKJT 方法) (torchrec.sparse.jagged_tensor.ComputeKJTToJTDict 方法) forward_recat_tensor (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性)，[1] fqn (torchrec.distributed.planner.types.ShardingOption 属性) from_dense() (torchrec.sparse.jagged_tensor.JaggedTensor 静态方法) from_dense_lengths() (torchrec.sparse.jagged_tensor.JaggedTensor 静态方法) from_float() (torchrec.quant.embedding_modules.EmbeddingBagCollection 类方法) (torchrec.quant.embedding_modules.EmbeddingCollection 类方法) (torchrec.quant.embedding_modules.FeatureProcessedEmbeddingBagCollection 类方法) from_jt_dict() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) from_lengths_sync() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) from_local() (torchrec.distributed.types.ShardingEnv 类方法) from_offsets_sync() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 静态方法) from_process_group() (torchrec.distributed.types.ShardingEnv 类方法) from_tensor_list() (torchrec.sparse.jagged_tensor.KeyedTensor 静态方法) FUSED (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) fused_optimizer (torchrec.distributed.embedding.ShardedEmbeddingCollection 属性) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 属性) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 属性) (torchrec.distributed.model_parallel.DistributedModelParallel 属性) (torchrec.optim.fused.FusedOptimizerModule 属性) fused_params (torchrec.distributed.embedding_sharding.EmbeddingShardingInfo 属性) (torchrec.distributed.embedding_types.BaseEmbeddingSharder 属性) (torchrec.distributed.embedding_types.BaseQuantEmbeddingSharder 属性) (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.distributed.embedding_types.ShardedEmbeddingTable 属性) FUSED_UVM (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) FUSED_UVM_CACHING (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) FusedKJTListSplitsAwaitable (torchrec.distributed.embedding_sharding 类) FusedOptimizer (torchrec.optim.fused 类) FusedOptimizerModule (torchrec.optim.fused 类) fwd_comms (torchrec.distributed.planner.types.Perf 属性) fwd_compute (torchrec.distributed.planner.types.Perf 属性) fx_wrap_tensor_view2d() (在 torchrec.distributed.embedding_lookup 模块中) G gb_to_bytes() (在 torchrec.distributed.planner.utils 模块中) GenericMeta (torchrec.distributed.types 类) get_block_sizes_runtime_device() (在 torchrec.distributed.sharding.rw_sharding 模块中) get_budget() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) get_cacheability() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) get_data() (torchrec.distributed.planner.types.CustomTopologyData 方法) get_device_from_parameter_sharding() (在 torchrec.distributed.embedding 模块中) (在模块 torchrec.distributed.embeddingbag 中) (在 torchrec.distributed.quant_embeddingbag 模块中) get_device_from_sharding_infos() (在 torchrec.distributed.quant_embeddingbag 模块中) get_ec_index_dedup() (在 torchrec.distributed.embedding 模块中) get_embedding_names_by_table() (在 torchrec.modules.embedding_modules 模块中) get_embedding_shard_metadata() (在 torchrec.distributed.sharding.rw_sharding 模块中) get_expected_lookups() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) get_gradient_division() (在 torchrec.distributed.comm_ops 模块中) get_group_rank() (在 torchrec.distributed.comm 模块中) get_local_rank() (在 torchrec.distributed.comm 模块中) get_local_size() (在 torchrec.distributed.comm 模块中) get_module() (在 torchrec.distributed.model_parallel 模块中) get_module_output_dimension() (在 torchrec.modules.utils 模块中) get_num_groups() (在 torchrec.distributed.comm 模块中) get_partition_by_type() (在 torchrec.distributed.planner.enumerators 模块中) get_plan_for_module() (torchrec.distributed.types.ShardingPlan 方法) get_tbes_to_register() (torchrec.distributed.embedding_lookup.InferCPUGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) get_tensor_size_bytes() (在模块 torchrec.distributed.types 中) get_unsharded_module_names() (在模块 torchrec.distributed.utils 中) get_unwrapped_module() (在模块 torchrec.distributed.model_parallel 中) get_use_sync_collectives() (在模块 torchrec.distributed.comm_ops 中) get_weight_init_max() (torchrec.modules.embedding_configs.BaseEmbeddingConfig 方法) get_weight_init_min() (torchrec.modules.embedding_configs.BaseEmbeddingConfig 方法) GetItemLazyAwaitable (torchrec.distributed.types 中的类) global_metadata (torchrec.distributed.embedding_types.ShardedMetaConfig 属性) GradientClipping (torchrec.optim.clipping 中的类) GradientClippingOptimizer (torchrec.optim.clipping 中的类) GreedyPerfPartitioner (torchrec.distributed.planner.partitioners 中的类) GreedyProposer (torchrec.distributed.planner.proposers 中的类) GridSearchProposer (torchrec.distributed.planner.proposers 中的类) group_tables() (在模块 torchrec.distributed.embedding_sharding 中) GroupedEmbeddingConfig (torchrec.distributed.embedding_types 中的类) GroupedEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) GroupedPooledEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) GroupedPositionWeightedModule (torchrec.distributed.grouped_position_weighted 中的类) H has_data() (torchrec.distributed.planner.types.CustomTopologyData 方法) has_feature_processor (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.modules.embedding_configs.EmbeddingTableConfig 属性) hbm (torchrec.distributed.planner.types.Storage 属性) HBM (torchrec.distributed.types.ParameterStorage 属性) hbm_mem_bw (torchrec.distributed.planner.types.Topology 属性) HeteroEmbeddingShardingPlanner (torchrec.distributed.planner.planners 中的类) HeuristicalStorageReservation (torchrec.distributed.planner.storage_reservations 中的类) HOST (torchrec.distributed.planner.types.PartitionByType 属性) I index_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) InferCPUGroupedEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) InferCwPooledEmbeddingDist (torchrec.distributed.sharding.cw_sharding 中的类) InferCwPooledEmbeddingDistWithPermute (torchrec.distributed.sharding.cw_sharding 中的类) InferCwPooledEmbeddingSharding (torchrec.distributed.sharding.cw_sharding 中的类) inference (torchrec.distributed.types.ObjectPoolShardingPlan 属性) InferenceStorageReservation (torchrec.distributed.planner.storage_reservations 中的类) InferGroupedEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) InferGroupedLookupMixin (torchrec.distributed.embedding_lookup 中的类) InferGroupedPooledEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) InferRwPooledEmbeddingDist (torchrec.distributed.sharding.rw_sharding 中的类) InferRwPooledEmbeddingSharding (torchrec.distributed.sharding.rw_sharding 中的类) InferRwSparseFeaturesDist (torchrec.distributed.sharding.rw_sharding 中的类) InferTwEmbeddingSharding (torchrec.distributed.sharding.tw_sharding 中的类) InferTwPooledEmbeddingDist (torchrec.distributed.sharding.tw_sharding 中的类) InferTwSparseFeaturesDist (torchrec.distributed.sharding.tw_sharding 中的类) init_data_parallel() (torchrec.distributed.model_parallel.DistributedModelParallel 方法) init_fn (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) init_mlp_weights_xavier_uniform() (在模块 torchrec.modules.utils 中) init_parameters() (在模块 torchrec.distributed.utils 中) init_state() (torchrec.optim.keyed.KeyedOptimizer 方法) input_dist() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.types.ShardedModule 方法) input_lengths (torchrec.distributed.planner.types.ShardingOption 属性) input_shape (torchrec.distributed.comm_ops.All2AllDenseInfo 属性) input_size (torchrec.distributed.comm_ops.AllGatherBaseInfo 属性), [1] input_size() (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) input_sizes (torchrec.distributed.comm_ops.ReduceScatterBaseInfo 属性), [1] (torchrec.distributed.comm_ops.ReduceScatterInfo 属性), [1] (torchrec.distributed.comm_ops.ReduceScatterVInfo 属性), [1] input_split_sizes (torchrec.distributed.comm_ops.All2AllVInfo 属性), [1] input_splits (torchrec.distributed.comm_ops.All2AllDenseInfo 属性) (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] (torchrec.distributed.comm_ops.ReduceScatterVInfo 属性), [1] (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性), [1] (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) input_tensors (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) InputDistOutputs (torchrec.distributed.embedding_types 中的类) INSUFFICIENT_STORAGE (torchrec.distributed.planner.types.PlannerErrorType 属性) inter_host_bw (torchrec.distributed.planner.types.Topology 属性) interaction() (torchrec.distributed.utils.ForkedPdb 方法) intra_and_cross_node_pg() (在模块 torchrec.distributed.comm 中) intra_host_bw (torchrec.distributed.planner.types.Topology 属性) inverse_indices (torchrec.distributed.embeddingbag.EmbeddingBagCollectionContext 属性) inverse_indices() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) inverse_indices_or_none() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) invoke_on_rank_and_broadcast_result() (在模块 torchrec.distributed.collective_utils 中) INVSQRT (torchrec.optim.warmup.WarmupPolicy 属性) is_fx_tracing() (在模块 torchrec.fx.tracer 中) is_history_metadata (torchrec.modules.mc_modules.MCHEvictionPolicyMetadataInfo 属性) is_leader() (在模块 torchrec.distributed.collective_utils 中) is_leaf_module() (torchrec.fx.tracer.Tracer 方法) is_mch_metadata (torchrec.modules.mc_modules.MCHEvictionPolicyMetadataInfo 属性) is_pooled (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.planner.types.ShardingOption 属性) is_weighted (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] (torchrec.modules.embedding_configs.EmbeddingTableConfig 属性) is_weighted() (torchrec.modules.embedding_modules.EmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollectionInterface 方法) (torchrec.quant.embedding_modules.EmbeddingBagCollection 方法) J jagged_index_select_with_empty() (在模块 torchrec.modules.utils 中) JaggedTensor (torchrec.sparse.jagged_tensor 中的类) JaggedTensorAllToAll (torchrec.distributed.dist_data 中的类), [1] JaggedTensorMeta (torchrec.sparse.jagged_tensor 中的类) jt_is_equal() (在模块 torchrec.sparse.jagged_tensor 中) K kernel_bw_lookup() (在模块 torchrec.distributed.planner.constants 中) key_dim() (torchrec.sparse.jagged_tensor.KeyedTensor 方法) KEY_VALUE (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) key_value_params (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) KeyedJaggedTensor (torchrec.sparse.jagged_tensor 中的类) KeyedOptimizer (torchrec.optim.keyed 中的类) KeyedOptimizerWrapper (torchrec.optim.keyed 中的类) KeyedTensor (torchrec.sparse.jagged_tensor 中的类) keys (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) keys() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) KeyValueParams (torchrec.distributed.types 中的类) kjt_is_equal() (在模块 torchrec.sparse.jagged_tensor 中) KJTAllToAll (torchrec.distributed.dist_data 中的类), [1] KJTAllToAllSplitsAwaitable (torchrec.distributed.dist_data 中的类), [1] KJTAllToAllTensorsAwaitable (torchrec.distributed.dist_data 中的类), [1] KJTList (torchrec.distributed.embedding_types 中的类) KJTListAwaitable (torchrec.distributed.embedding_sharding 中的类) KJTListSplitsAwaitable (torchrec.distributed.embedding_sharding 中的类) KJTOneToAll (torchrec.distributed.dist_data 中的类), [1] KJTSplitsAllToAllMeta (torchrec.distributed.embedding_sharding 中的类) L labels (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) LAMB (torchrec.distributed.embedding_types.OptimType 属性) LARS_SGD (torchrec.distributed.embedding_types.OptimType 属性) lazy_apply() (在模块 torchrec.modules.lazy_extension 中) LazyAwaitable (torchrec.distributed.types 中的类) LazyGetItemMixin (torchrec.distributed.types 中的类) LazyModuleExtensionMixin (torchrec.modules.lazy_extension 中的类) LazyNoWait (torchrec.distributed.types 中的类) length_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) length_per_key_or_none() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) lengths() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) lengths_after_sparse_data_all2all (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] lengths_offset_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) lengths_or_none() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) LFU_EvictionPolicy (torchrec.modules.mc_modules 中的类) LINEAR (torchrec.optim.warmup.WarmupPolicy 属性) LION (torchrec.distributed.embedding_types.OptimType 属性) ListOfKJTList (torchrec.distributed.embedding_types 中的类) ListOfKJTListAwaitable (torchrec.distributed.embedding_sharding 中的类) ListOfKJTListSplitsAwaitable (torchrec.distributed.embedding_sharding 中的类) load() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 方法) (torchrec.distributed.planner.proposers.GreedyProposer 方法) (torchrec.distributed.planner.proposers.GridSearchProposer 方法) (torchrec.distributed.planner.proposers.UniformProposer 方法) (torchrec.distributed.planner.types.Proposer 方法) load_config_text() (在模块 torchrec.inference.model_packager 中) load_factor (torchrec.distributed.types.CacheParams 属性), [1] load_pickle_config() (在模块 torchrec.inference.model_packager 中) load_state_dict() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedLookupMixin 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.model_parallel.DistributedModelParallel 方法) (torchrec.optim.keyed.KeyedOptimizer 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) local_cols (torchrec.distributed.embedding_types.ShardedConfig 属性) local_metadata (torchrec.distributed.embedding_types.ShardedMetaConfig 属性) local_rows (torchrec.distributed.embedding_types.ShardedConfig 属性) local_world_size (torchrec.distributed.planner.partitioners.OrderedDeviceHardware 属性) (torchrec.distributed.planner.types.Topology 属性) log() (torchrec.distributed.planner.stats.EmbeddingStats 方法) (torchrec.distributed.planner.stats.NoopEmbeddingStats 方法) (torchrec.distributed.planner.types.Stats 方法) LowRankCrossNet (torchrec.modules.crossnet 中的类) LowRankMixtureCrossNet (torchrec.modules.crossnet 中的类) lr_scale (torchrec.optim.warmup.WarmupStage 属性) LRU_EvictionPolicy (torchrec.modules.mc_modules 中的类) LuusJaakolaSearch (torchrec.distributed.planner.utils 中的类) M ManagedCollisionCollection (torchrec.modules.mc_modules 中的类) ManagedCollisionCollectionAwaitable (torchrec.distributed.mc_modules 中的类) ManagedCollisionCollectionContext (torchrec.distributed.mc_modules 中的类) ManagedCollisionCollectionSharder (torchrec.distributed.mc_modules 中的类) ManagedCollisionEmbeddingBagCollection (torchrec.modules.mc_embedding_modules 中的类) ManagedCollisionEmbeddingBagCollectionContext (torchrec.distributed.mc_embeddingbag 中的类) ManagedCollisionEmbeddingBagCollectionSharder (torchrec.distributed.mc_embeddingbag 中的类) ManagedCollisionEmbeddingCollection (torchrec.modules.mc_embedding_modules 中的类) ManagedCollisionEmbeddingCollectionContext (torchrec.distributed.mc_embedding 中的类) ManagedCollisionEmbeddingCollectionSharder (torchrec.distributed.mc_embedding 中的类) ManagedCollisionModule (torchrec.modules.mc_modules 中的类) max_iters (torchrec.optim.warmup.WarmupStage 属性) MCHEvictionPolicy (torchrec.modules.mc_modules 中的类) MCHEvictionPolicyMetadataInfo (torchrec.modules.mc_modules 中的类) MCHManagedCollisionModule (torchrec.modules.mc_modules 中的类) MEAN (torchrec.modules.embedding_configs.PoolingType 属性) MemoryBalancedPartitioner (torchrec.distributed.planner.partitioners 中的类) merge_fused_params() (在模块 torchrec.distributed.utils 中) MergePooledEmbeddingsModule (torchrec.distributed.dist_data 中的类), [1] mesh (torchrec.distributed.embedding_types.DTensorMetadata 属性) metadata_info (torchrec.modules.mc_modules.DistanceLFU_EvictionPolicy 属性) (torchrec.modules.mc_modules.LFU_EvictionPolicy 属性) (torchrec.modules.mc_modules.LRU_EvictionPolicy 属性) (torchrec.modules.mc_modules.MCHEvictionPolicy 属性) metadata_name (torchrec.modules.mc_modules.MCHEvictionPolicyMetadataInfo 属性) MetaInferGroupedEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) MetaInferGroupedPooledEmbeddingsLookup (torchrec.distributed.embedding_lookup 中的类) min_partition (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] MLP (torchrec.modules.mlp 中的类) model_inputs_data() (torchrec.inference.modules.PredictFactory 方法) module torchrec.distributed torchrec.distributed.collective_utils torchrec.distributed.comm torchrec.distributed.comm_ops torchrec.distributed.dist_data，[1] torchrec.distributed.embedding torchrec.distributed.embedding_lookup torchrec.distributed.embedding_sharding torchrec.distributed.embedding_types torchrec.distributed.embeddingbag torchrec.distributed.grouped_position_weighted torchrec.distributed.mc_embedding torchrec.distributed.mc_embeddingbag torchrec.distributed.mc_modules torchrec.distributed.model_parallel torchrec.distributed.planner torchrec.distributed.planner.constants torchrec.distributed.planner.enumerators torchrec.distributed.planner.partitioners torchrec.distributed.planner.perf_models torchrec.distributed.planner.planners torchrec.distributed.planner.proposers torchrec.distributed.planner.shard_estimators torchrec.distributed.planner.stats torchrec.distributed.planner.storage_reservations torchrec.distributed.planner.types torchrec.distributed.planner.utils torchrec.distributed.quant_embeddingbag torchrec.distributed.sharding torchrec.distributed.sharding.cw_sharding torchrec.distributed.sharding.dp_sharding torchrec.distributed.sharding.rw_sharding torchrec.distributed.sharding.tw_sharding torchrec.distributed.sharding.twcw_sharding torchrec.distributed.sharding.twrw_sharding torchrec.distributed.train_pipeline torchrec.distributed.types torchrec.distributed.utils torchrec.fx，[1] torchrec.fx.tracer torchrec.inference，[1] torchrec.inference.model_packager torchrec.inference.modules torchrec.models.deepfm torchrec.modules torchrec.modules.activation torchrec.modules.crossnet torchrec.modules.deepfm torchrec.modules.embedding_configs torchrec.modules.embedding_modules torchrec.modules.feature_processor torchrec.modules.lazy_extension torchrec.modules.mc_embedding_modules torchrec.modules.mc_modules torchrec.modules.mlp torchrec.modules.utils torchrec.optim，[1] torchrec.optim.clipping torchrec.optim.fused torchrec.optim.keyed torchrec.optim.warmup torchrec.quant，[1] torchrec.quant.embedding_modules torchrec.sparse，[1] torchrec.sparse.jagged_tensor module (torchrec.distributed.model_parallel.DistributedModelParallel 属性) (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.planner.types.ShardingOption 属性) module_pooled() (torchrec.distributed.planner.types.ShardingOption 静态方法) module_type (torchrec.distributed.embedding.EmbeddingCollectionSharder 属性) (torchrec.distributed.embeddingbag.EmbeddingBagCollectionSharder 属性) (torchrec.distributed.embeddingbag.EmbeddingBagSharder 属性) (torchrec.distributed.mc_embedding.ManagedCollisionEmbeddingCollectionSharder 属性) (torchrec.distributed.mc_embeddingbag.ManagedCollisionEmbeddingBagCollectionSharder 属性) (torchrec.distributed.mc_modules.ManagedCollisionCollectionSharder 属性) (torchrec.distributed.quant_embeddingbag.QuantEmbeddingBagCollectionSharder 属性) (torchrec.distributed.quant_embeddingbag.QuantFeatureProcessedEmbeddingBagCollectionSharder 属性) (torchrec.distributed.types.ModuleSharder 属性) ModuleSharder (torchrec.distributed.types 中的类) ModuleShardingMixIn (torchrec.distributed.embedding_types 中的类) ModuleShardingPlan (torchrec.distributed.types 中的类) multipass_prefetch_config (torchrec.distributed.types.CacheParams 属性) N name (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) named_buffers() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedLookupMixin 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.grouped_position_weighted.GroupedPositionWeightedModule 方法) (torchrec.distributed.model_parallel.DistributedModelParallel 方法) (torchrec.modules.feature_processor.PositionWeightedProcessor 方法) named_modules() (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) named_parameters() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedLookupMixin 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.grouped_position_weighted.GroupedPositionWeightedModule 方法) (torchrec.distributed.model_parallel.DistributedModelParallel 方法) named_parameters_by_table() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) need_indices() (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) need_pos (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) need_preproc (torchrec.inference.modules.QualNameMetadata 属性) next() (torchrec.distributed.planner.utils.BinarySearchPredicate 方法) (torchrec.distributed.planner.utils.LuusJaakolaSearch 方法) next_plan() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) NONE (torchrec.distributed.types.PipelineType 属性) (torchrec.modules.embedding_configs.PoolingType 属性) (torchrec.optim.clipping.GradientClipping 属性) (torchrec.optim.warmup.WarmupPolicy 属性) none_throws() (在模块 torchrec.distributed.utils 中) NoopEmbeddingStats (torchrec.distributed.planner.stats 中的类) NoopPerfModel (torchrec.distributed.planner.perf_models 中的类) NoOpQuantizedCommCodec (torchrec.distributed.types 中的类) NoopStorageModel (torchrec.distributed.planner.perf_models 中的类) NORM (torchrec.optim.clipping.GradientClipping 属性) NoWait (torchrec.distributed.types 中的类) NullShardedModuleContext (torchrec.distributed.types 中的类) NullShardingContext (torchrec.distributed.types 中的类) num_embeddings (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) (torchrec.modules.embedding_configs.EmbeddingConfig 属性) num_features() (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) (torchrec.modules.embedding_configs.BaseEmbeddingConfig 方法) num_inputs (torchrec.distributed.planner.types.ShardingOption 属性) num_poolings (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] num_shards (torchrec.distributed.planner.types.ShardingOption 属性) O ObjectPoolShardingPlan (torchrec.distributed.types 中的类) ObjectPoolShardingType (torchrec.distributed.types 中的类) offset (torchrec.distributed.planner.types.Shard 属性) offset_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) offset_per_key_or_none() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) offsets() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) offsets_or_none() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) offsets_to_range_traceble() (在模块 torchrec.modules.feature_processor 中) optimizer_type_to_emb_opt_type() (在模块 torchrec.distributed.utils 中) optimizers (torchrec.optim.keyed.CombinedOptimizer 属性) OptimizerWrapper (torchrec.optim.keyed 中的类) OptimType (torchrec.distributed.embedding_types 中的类) OrderedDeviceHardware (torchrec.distributed.planner.partitioners 中的类) OTHER (torchrec.distributed.planner.types.PlannerErrorType 属性) output_dist() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) (torchrec.distributed.types.ShardedModule 方法) output_dtype (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) output_dtype() (torchrec.quant.embedding_modules.EmbeddingBagCollection 方法) (torchrec.quant.embedding_modules.EmbeddingCollection 方法) output_size() (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) output_split_sizes (torchrec.distributed.comm_ops.All2AllVInfo 属性)，[1] output_splits (torchrec.distributed.comm_ops.All2AllDenseInfo 属性) (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性)，[1] (torchrec.distributed.comm_ops.VariableBatchAll2AllPooledInfo 属性)，[1] OverArch (torchrec.models.deepfm 中的类) P pad_vbe_kjt_lengths() (在模块 torchrec.distributed.embedding 中) param (torchrec.distributed.embedding_sharding.EmbeddingShardingInfo 属性) param_count (torchrec.distributed.planner.partitioners.ShardingOptionGroup 属性) param_groups (torchrec.optim.keyed.CombinedOptimizer 属性) param_sharding (torchrec.distributed.embedding_sharding.EmbeddingShardingInfo 属性) ParameterConstraints (torchrec.distributed.planner.types 中的类) ParameterSharding (torchrec.distributed.types 中的类) ParameterStorage (torchrec.distributed.types 中的类) params (torchrec.optim.keyed.CombinedOptimizer 属性) PARTIAL_ROWWISE_ADAM (torchrec.distributed.embedding_types.OptimType 属性) PARTIAL_ROWWISE_LAMB (torchrec.distributed.embedding_types.OptimType 属性) PARTITION (torchrec.distributed.planner.types.PlannerErrorType 属性) partition() (torchrec.distributed.planner.partitioners.GreedyPerfPartitioner 方法) (torchrec.distributed.planner.partitioners.MemoryBalancedPartitioner 方法) (torchrec.distributed.planner.types.Partitioner 方法) PartitionByType (torchrec.distributed.planner.types 中的类) Partitioner (torchrec.distributed.planner.types 中的类) path (torchrec.distributed.planner.types.ShardingOption 属性) path_of_module() (torchrec.fx.tracer.Tracer 方法) per_table_weight_dtype (torchrec.modules.embedding_configs.QuantConfig 属性) Perceptron (torchrec.modules.mlp 中的类) Perf (torchrec.distributed.planner.types 中的类) PERF (torchrec.distributed.planner.partitioners.SortBy 属性) perf (torchrec.distributed.planner.types.DeviceHardware 属性) (torchrec.distributed.planner.types.Shard 属性) perf_func_emb_wall_time() (torchrec.distributed.planner.shard_estimators.EmbeddingPerfEstimator 类方法) perf_sum (torchrec.distributed.planner.partitioners.ShardingOptionGroup 属性) PerfModel (torchrec.distributed.planner.types 中的类) permute() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) permute_multi_embedding() (在模块 torchrec.sparse.jagged_tensor 中) permuted_lengths_after_sparse_data_all2all (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性)，[1] pg (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) pg_name() (在模块 torchrec.distributed.comm_ops 中) pin_memory() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) pinned (torchrec.inference.modules.BatchingMetadata 属性) PipelineType (torchrec.distributed.types 中的类) placement() (在模块 torchrec.distributed.planner.utils 中) placements (torchrec.distributed.embedding_types.DTensorMetadata 属性) plan (torchrec.distributed.model_parallel.DistributedModelParallel 属性) (torchrec.distributed.types.ShardingPlan 属性)，[1] plan() (torchrec.distributed.planner.planners.EmbeddingShardingPlanner 方法) (torchrec.distributed.planner.planners.HeteroEmbeddingShardingPlanner 方法) (torchrec.distributed.types.ShardingPlanner 方法) PlannerError PlannerErrorType (torchrec.distributed.planner.types 中的类) policy (torchrec.optim.warmup.WarmupStage 属性) POLY (torchrec.optim.warmup.WarmupPolicy 属性) POOLED_EMBEDDINGS_ALL_TO_ALL (torchrec.distributed.types.CommOp 属性) POOLED_EMBEDDINGS_REDUCE_SCATTER (torchrec.distributed.types.CommOp 属性) PooledEmbeddingsAllGather (torchrec.distributed.dist_data 中的类)，[1] PooledEmbeddingsAllToAll (torchrec.distributed.dist_data 中的类)，[1] PooledEmbeddingsAwaitable (torchrec.distributed.dist_data 中的类)，[1] PooledEmbeddingsReduceScatter (torchrec.distributed.dist_data 中的类)，[1] pooling (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 属性) (torchrec.modules.embedding_configs.EmbeddingBagConfig 属性) (torchrec.modules.embedding_configs.EmbeddingTableConfig 属性) pooling_factors (torchrec.distributed.planner.types.ParameterConstraints 属性)，[1] pooling_type_to_pooling_mode() (在模块 torchrec.modules.embedding_configs 中) pooling_type_to_str() (在模块 torchrec.modules.embedding_configs 中) PoolingType (torchrec.modules.embedding_configs 中的类) populate_estimates() (torchrec.distributed.planner.enumerators.EmbeddingEnumerator 方法) (torchrec.distributed.planner.types.Enumerator 方法) position_weighted_module_update_features() (在模块 torchrec.modules.feature_processor 中) PositionWeightedModule (torchrec.modules.feature_processor 中的类) PositionWeightedProcessor (torchrec.modules.feature_processor 中的类) post_load_state_dict() (torchrec.optim.keyed.CombinedOptimizer 方法) (torchrec.optim.keyed.KeyedOptimizer 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) (torchrec.optim.warmup.WarmupOptimizer 方法) precision (torchrec.distributed.types.CacheParams 属性)，[1] predict_forward() (torchrec.inference.modules.PredictModule 方法) predict_module (torchrec.inference.modules.PredictModule 属性) PredictFactory (torchrec.inference.modules 中的类) PredictFactoryPackager (torchrec.inference.model_packager 中的类) PredictModule (torchrec.inference.modules 中的类) prefetch() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_types.ShardedEmbeddingModule 方法) prefetch_compute (torchrec.distributed.planner.types.Perf 属性) prefetch_pipeline (torchrec.distributed.types.CacheParams 属性)，[1] prepend_opt_key() (torchrec.optim.keyed.CombinedOptimizer 静态方法) preprocess() (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) probabilistic_threshold_filter() (在模块 torchrec.modules.mc_modules 中) process_pooled_embeddings() (在模块 torchrec.modules.embedding_modules 中) prod() (在模块 torchrec.distributed.planner.utils 中) profile() (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) promote_high_prefetch_overheaad_table_to_hbm() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 静态方法) propose() (torchrec.distributed.planner.proposers.EmbeddingOffloadScaleupProposer 方法) (torchrec.distributed.planner.proposers.GreedyProposer 方法) (torchrec.distributed.planner.proposers.GridSearchProposer 方法) (torchrec.distributed.planner.proposers.UniformProposer 方法) (torchrec.distributed.planner.types.Proposer 方法) Proposer (torchrec.distributed.planner.types 中的类) proposers_to_proposals_list() (在模块 torchrec.distributed.planner.proposers 中) pruned_num_embeddings() (在模块 torchrec.quant.embedding_modules 中) pruning_indices_remapping (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) ps_hosts (torchrec.distributed.types.KeyValueParams 属性), [1] purge() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) Q qcomm_codecs_registry (torchrec.distributed.embedding_sharding.EmbeddingSharding 属性) (torchrec.distributed.types.ModuleSharder 属性) (torchrec.distributed.types.ShardedModule 属性) qualname_metadata() (torchrec.inference.modules.PredictFactory 方法) qualname_metadata_json() (torchrec.inference.modules.PredictFactory 方法) QualNameMetadata (torchrec.inference.modules 中的类) QUANT (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) quant_prep_customize_row_alignment() (在模块 torchrec.quant.embedding_modules 中) quant_prep_enable_quant_state_dict_split_scale_bias() (在模块 torchrec.quant.embedding_modules 中) quant_prep_enable_quant_state_dict_split_scale_bias_for_types() (在模块 torchrec.quant.embedding_modules 中) quant_prep_enable_register_tbes() (在模块 torchrec.quant.embedding_modules 中) QUANT_UVM (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) QUANT_UVM_CACHING (torchrec.distributed.embedding_types.EmbeddingComputeKernel 属性) QuantConfig (torchrec.modules.embedding_configs 中的类) QuantEmbeddingBagCollectionSharder (torchrec.distributed.quant_embeddingbag 中的类) QuantFeatureProcessedEmbeddingBagCollectionSharder (torchrec.distributed.quant_embeddingbag 中的类) quantize_dense() (在模块 torchrec.inference.modules 中) quantize_embeddings() (在模块 torchrec.inference.modules 中) quantize_feature() (在模块 torchrec.inference.modules 中) quantize_inference_model() (在模块 torchrec.inference.modules 中) quantize_state_dict() (在模块 torchrec.quant.embedding_modules 中) quantized_dtype (torchrec.distributed.types.QuantizedCommCodec 属性) quantized_dtype() (torchrec.distributed.types.NoOpQuantizedCommCodec 方法) QuantizedCommCodec (torchrec.distributed.types 中的类) QuantizedCommCodecs (torchrec.distributed.types 中的类) R rank (torchrec.distributed.planner.types.DeviceHardware 属性) (torchrec.distributed.planner.types.Shard 属性) rank_device() (在模块 torchrec.distributed.types 中) ranks (torchrec.distributed.types.ParameterSharding 属性) rate() (torchrec.distributed.planner.perf_models.NoopPerfModel 方法) (torchrec.distributed.planner.perf_models.NoopStorageModel 方法) (torchrec.distributed.planner.types.PerfModel 方法) rebuild_with_output_id_range() (torchrec.modules.mc_modules.ManagedCollisionModule 方法) (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) recat (torchrec.modules.utils.SequenceVBEContext 属性) record_history_metadata() (torchrec.modules.mc_modules.DistanceLFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LRU_EvictionPolicy 方法) (torchrec.modules.mc_modules.MCHEvictionPolicy 方法) record_stream() (torchrec.distributed.embedding.EmbeddingCollectionContext 方法) (torchrec.distributed.embedding_sharding.EmbeddingShardingContext 方法) (torchrec.distributed.embedding_types.InputDistOutputs 方法) (torchrec.distributed.embedding_types.KJTList 方法) (torchrec.distributed.embedding_types.ListOfKJTList 方法) (torchrec.distributed.embeddingbag.EmbeddingBagCollectionContext 方法) (torchrec.distributed.mc_embedding.ManagedCollisionEmbeddingCollectionContext 方法) (torchrec.distributed.mc_embeddingbag.ManagedCollisionEmbeddingBagCollectionContext 方法) (torchrec.distributed.types.NullShardedModuleContext 方法) (torchrec.distributed.types.NullShardingContext 方法) (torchrec.modules.utils.SequenceVBEContext 方法) (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) reduce_scatter_base_pooled() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_base_sync() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_pooled() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_sync() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_tensor_backward() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_tensor_fake() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_tensor_setup_context() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_v_per_feature_pooled() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_v_pooled() (在模块 torchrec.distributed.comm_ops 中) reduce_scatter_v_sync() (在模块 torchrec.distributed.comm_ops 中) ReduceScatter_Req (torchrec.distributed.comm_ops 中的类) ReduceScatter_Wait (torchrec.distributed.comm_ops 中的类) ReduceScatterBase_Req (torchrec.distributed.comm_ops 中的类) ReduceScatterBase_Wait (torchrec.distributed.comm_ops 中的类) ReduceScatterBaseInfo (torchrec.distributed.comm_ops 中的类) ReduceScatterInfo (torchrec.distributed.comm_ops 中的类) ReduceScatterV_Req (torchrec.distributed.comm_ops 中的类) ReduceScatterV_Wait (torchrec.distributed.comm_ops 中的类) ReduceScatterVInfo (torchrec.distributed.comm_ops 中的类) regroup() (torchrec.sparse.jagged_tensor.KeyedTensor 静态方法) regroup_as_dict() (torchrec.sparse.jagged_tensor.KeyedTensor 静态方法) regroup_kts() (在模块 torchrec.sparse.jagged_tensor 中) reindexed_length_per_key (torchrec.modules.utils.SequenceVBEContext 属性) reindexed_lengths (torchrec.modules.utils.SequenceVBEContext 属性) reindexed_values (torchrec.modules.utils.SequenceVBEContext 属性) remap() (torchrec.modules.mc_modules.MCHManagedCollisionModule 方法) remapped_kjt (torchrec.distributed.mc_embeddingbag.ManagedCollisionEmbeddingBagCollectionContext 属性) reorder_inverse_indices() (在模块 torchrec.modules.embedding_modules 中) replace_placement_with_meta_device() (在模块 torchrec.distributed.embeddingbag 中) REPLICATED_ROW_WISE (torchrec.distributed.types.ObjectPoolShardingType 属性) Request (torchrec.distributed.comm_ops 中的类) reserve() (torchrec.distributed.planner.storage_reservations.FixedPercentageStorageReservation 方法) (torchrec.distributed.planner.storage_reservations.HeuristicalStorageReservation 方法) (torchrec.distributed.planner.storage_reservations.InferenceStorageReservation 方法) (torchrec.distributed.planner.types.StorageReservation 方法) reserved_memory (torchrec.distributed.types.CacheParams 属性), [1] reset_parameters() (torchrec.distributed.embedding.ShardedEmbeddingCollection 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingBagCollection 方法) (torchrec.modules.embedding_modules.EmbeddingCollection 方法) (torchrec.modules.feature_processor.PositionWeightedModule 方法) reset_shard_rank() (在模块 torchrec.distributed.planner.utils 中) result_metadata() (torchrec.inference.modules.PredictFactory 方法) round_to_one_sigfig() (在模块 torchrec.distributed.planner.stats 中) ROW_WISE (torchrec.distributed.types.ObjectPoolShardingType 属性) (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) ROWWISE_ADAGRAD (torchrec.distributed.embedding_types.OptimType 属性) run_on_leader() (在模块 torchrec.distributed.collective_utils 中) run_weights_dependent_transformations() (torchrec.inference.modules.PredictFactory 方法) run_weights_independent_tranformations() (torchrec.inference.modules.PredictFactory 方法) RwPooledEmbeddingDist (torchrec.distributed.sharding.rw_sharding 中的类) RwPooledEmbeddingSharding (torchrec.distributed.sharding.rw_sharding 中的类) RwSparseFeaturesDist (torchrec.distributed.sharding.rw_sharding 中的类) S save_param_groups() (torchrec.optim.keyed.CombinedOptimizer 方法) (torchrec.optim.keyed.KeyedOptimizer 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) save_predict_factory() (torchrec.inference.model_packager.PredictFactoryPackager 类方法) scope() (在模块 torchrec.distributed.types 中) SeqEmbeddingsAllToOne (torchrec.distributed.dist_data 中的类), [1] SEQUENCE_EMBEDDINGS_ALL_TO_ALL (torchrec.distributed.types.CommOp 属性) SequenceEmbeddingsAllToAll (torchrec.distributed.dist_data 中的类), [1] SequenceEmbeddingsAwaitable (torchrec.distributed.dist_data 中的类), [1] SequenceVBEContext (torchrec.modules.utils 中的类) set_device() (torchrec.distributed.dist_data.EmbeddingsAllToOne 方法), [1] (torchrec.distributed.dist_data.EmbeddingsAllToOneReduce 方法), [1] (torchrec.distributed.dist_data.MergePooledEmbeddingsModule 方法), [1] (torchrec.distributed.dist_data.SeqEmbeddingsAllToOne 方法), [1] set_ec_index_dedup() (在模块 torchrec.distributed.embedding 中) set_extern_modules() (torchrec.inference.model_packager.PredictFactoryPackager 类方法) set_gradient_division() (在模块 torchrec.distributed.comm_ops 中) set_hbm_per_device() (在模块 torchrec.distributed.planner.partitioners 中) set_mocked_modules() (torchrec.inference.model_packager.PredictFactoryPackager 类方法) set_use_sync_collectives() (在模块 torchrec.distributed.comm_ops 中) SGD (torchrec.distributed.embedding_types.OptimType 属性) sgdr_period (torchrec.optim.warmup.WarmupStage 属性) SHAMPOO (torchrec.distributed.embedding_types.OptimType 属性) SHAMPOO_V2 (torchrec.distributed.embedding_types.OptimType 属性) Shard (torchrec.distributed.planner.types 中的类) shard() (torchrec.distributed.embedding.EmbeddingCollectionSharder 方法) (torchrec.distributed.embeddingbag.EmbeddingBagCollectionSharder 方法) (torchrec.distributed.embeddingbag.EmbeddingBagSharder 方法) (torchrec.distributed.mc_embedding.ManagedCollisionEmbeddingCollectionSharder 方法) (torchrec.distributed.mc_embeddingbag.ManagedCollisionEmbeddingBagCollectionSharder 方法) (torchrec.distributed.mc_modules.ManagedCollisionCollectionSharder 方法) (torchrec.distributed.quant_embeddingbag.QuantEmbeddingBagCollectionSharder 方法) (torchrec.distributed.quant_embeddingbag.QuantFeatureProcessedEmbeddingBagCollectionSharder 方法) (torchrec.distributed.types.ModuleSharder 类方法) shard_quant_model() (在模块 torchrec.inference.modules 中) shardable_parameters() (torchrec.distributed.embedding.EmbeddingCollectionSharder 方法) (torchrec.distributed.embedding_types.BaseQuantEmbeddingSharder 方法) (torchrec.distributed.embeddingbag.EmbeddingBagCollectionSharder 方法) (torchrec.distributed.embeddingbag.EmbeddingBagSharder 方法) (torchrec.distributed.mc_modules.ManagedCollisionCollectionSharder 方法) (torchrec.distributed.types.ModuleSharder 方法) sharded_model_copy (torchrec.distributed.utils 中的类) sharded_parameter_names() (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 方法) (torchrec.distributed.types.ShardedModule 方法) ShardedConfig (torchrec.distributed.embedding_types 中的类) ShardedEmbeddingBag (torchrec.distributed.embeddingbag 中的类) ShardedEmbeddingBagCollection (torchrec.distributed.embeddingbag 中的类) ShardedEmbeddingCollection (torchrec.distributed.embedding 中的类) ShardedEmbeddingModule (torchrec.distributed.embedding_types 中的类) ShardedEmbeddingTable (torchrec.distributed.embedding_types 中的类) ShardedManagedCollisionCollection (torchrec.distributed.mc_modules 中的类) ShardedManagedCollisionEmbeddingBagCollection (torchrec.distributed.mc_embeddingbag 中的类) ShardedManagedCollisionEmbeddingCollection (torchrec.distributed.mc_embedding 中的类) ShardedMetaConfig (torchrec.distributed.embedding_types 中的类) ShardedModule (torchrec.distributed.types 中的类) ShardedQuantEbcInputDist (torchrec.distributed.quant_embeddingbag 中的类) ShardedQuantEmbeddingBagCollection (torchrec.distributed.quant_embeddingbag 中的类) ShardedQuantFeatureProcessedEmbeddingBagCollection (torchrec.distributed.quant_embeddingbag 中的类) sharder_name() (在模块 torchrec.distributed.planner.utils 中) ShardEstimator (torchrec.distributed.planner.types 中的类) sharding_contexts (torchrec.distributed.embeddingbag.EmbeddingBagCollectionContext 属性) sharding_options (torchrec.distributed.planner.partitioners.ShardingOptionGroup 属性) sharding_spec (torchrec.distributed.types.ParameterSharding 属性) sharding_type (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ObjectPoolShardingPlan 属性) (torchrec.distributed.types.ParameterSharding 属性) sharding_type_device_group_to_sharding_infos() (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) sharding_types (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] sharding_types() (torchrec.distributed.embedding.EmbeddingCollectionSharder 方法) (torchrec.distributed.embedding_types.BaseEmbeddingSharder 方法) (torchrec.distributed.embedding_types.BaseQuantEmbeddingSharder 方法) (torchrec.distributed.mc_modules.ManagedCollisionCollectionSharder 方法) (torchrec.distributed.quant_embeddingbag.QuantFeatureProcessedEmbeddingBagCollectionSharder 方法) (torchrec.distributed.types.ModuleSharder 方法) ShardingEnv (torchrec.distributed.types 中的类) ShardingOption (torchrec.distributed.planner.types 中的类) ShardingOptionGroup (torchrec.distributed.planner.partitioners 中的类) ShardingPlan (torchrec.distributed.types 中的类) ShardingPlanner (torchrec.distributed.types 中的类) shardings (torchrec.distributed.embedding_types.ModuleShardingMixIn 属性) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 属性) ShardingType (torchrec.distributed.types 中的类) (torchrec.modules.embedding_configs 中的类) shards (torchrec.distributed.planner.types.ShardingOption 属性) shrink_right() (torchrec.distributed.planner.utils.LuusJaakolaSearch 方法) SimpleDeepFMNN (torchrec.models.deepfm 中的类) size (torchrec.distributed.embedding_types.DTensorMetadata 属性) (torchrec.distributed.planner.types.Shard 属性) SortBy (torchrec.distributed.planner.partitioners 中的类) sparse_grad_parameter_names() (torchrec.distributed.model_parallel.DistributedModelParallel 方法) SparseArch (torchrec.models.deepfm 中的类) split() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) splits (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) splits_tensors (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) SplitsAllToAllAwaitable (torchrec.distributed.dist_data 中的类), [1] ssd_storage_directory (torchrec.distributed.types.KeyValueParams 属性), [1] stagger (torchrec.distributed.embedding_sharding.KJTSplitsAllToAllMeta 属性) state (torchrec.optim.keyed.CombinedOptimizer 属性) state_dict() (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.InferGroupedLookupMixin 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 方法) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 方法) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 方法) (torchrec.distributed.grouped_position_weighted.GroupedPositionWeightedModule 方法) (torchrec.distributed.model_parallel.DistributedModelParallel 方法) (torchrec.inference.modules.PredictModule 方法) (torchrec.modules.feature_processor.PositionWeightedProcessor 方法) (torchrec.optim.keyed.KeyedOptimizer 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) Stats (torchrec.distributed.planner.types 类) stats (torchrec.distributed.types.CacheParams 属性), [1] STEP (torchrec.optim.warmup.WarmupPolicy 属性) step() (torchrec.optim.clipping.GradientClippingOptimizer 方法) (torchrec.optim.fused.EmptyFusedOptimizer 方法) (torchrec.optim.fused.FusedOptimizer 方法) (torchrec.optim.keyed.CombinedOptimizer 方法) (torchrec.optim.keyed.KeyedOptimizerWrapper 方法) (torchrec.optim.keyed.OptimizerWrapper 方法) (torchrec.optim.warmup.WarmupOptimizer 方法) stochastic_rounding (torchrec.distributed.planner.types.ParameterConstraints 属性), [1] (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.types.ParameterSharding 属性) Storage (torchrec.distributed.planner.types 类) STORAGE (torchrec.distributed.planner.partitioners.SortBy 属性) storage (torchrec.distributed.planner.types.DeviceHardware 属性) (torchrec.distributed.planner.types.Shard 属性) storage_repr_in_gb() (torchrec.distributed.planner.utils 模块中) storage_sum (torchrec.distributed.planner.partitioners.ShardingOptionGroup 属性) storage_usage() (torchrec.distributed.embedding_types.BaseEmbeddingSharder 方法) (torchrec.distributed.embedding_types.BaseQuantEmbeddingSharder 方法) (torchrec.distributed.types.ModuleSharder 方法) StorageReservation (torchrec.distributed.planner.types 类) STRICT_CONSTRAINTS (torchrec.distributed.planner.types.PlannerErrorType 属性) stride (torchrec.distributed.embedding_types.DTensorMetadata 属性) stride() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) stride_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) stride_per_key_per_rank() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) SUM (torchrec.modules.embedding_configs.PoolingType 属性) supported_fields (torchrec.distributed.planner.types.CustomTopologyData 属性) SwishLayerNorm (torchrec.modules.activation 类) symbolic_trace() (torchrec.fx.tracer 模块中) sync() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) T TABLE_COLUMN_WISE (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) table_names() (torchrec.distributed.embedding_types.GroupedEmbeddingConfig 方法) TABLE_ROW_WISE (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) TABLE_WISE (torchrec.distributed.types.ShardingType 属性) (torchrec.modules.embedding_configs.ShardingType 属性) tbes (torchrec.distributed.quant_embeddingbag.ShardedQuantFeatureProcessedEmbeddingBagCollection 属性) (torchrec.quant.embedding_modules.FeatureProcessedEmbeddingBagCollection 属性) tbes_configs() (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 方法) tensor (torchrec.distributed.planner.types.ShardingOption 属性) (torchrec.distributed.planner.types.ShardingOption 属性) TensorAllToAll (torchrec.distributed.dist_data 类), [1] TensorAllToAllSplitsAwaitable (torchrec.distributed.dist_data 类), [1] TensorAllToAllValuesAwaitable (torchrec.distributed.dist_data 类), [1] TensorValuesAllToAll (torchrec.distributed.dist_data 类), [1] to() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) to_dense() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) to_dense_weights() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) to_dict() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) to_padded_dense() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) to_padded_dense_weights() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) Topology (torchrec.distributed.planner.types 类) torchrec.distributed module torchrec.distributed.collective_utils module torchrec.distributed.comm module torchrec.distributed.comm_ops module torchrec.distributed.dist_data 模块, [1] torchrec.distributed.embedding module torchrec.distributed.embedding_lookup module torchrec.distributed.embedding_sharding module torchrec.distributed.embedding_types module torchrec.distributed.embeddingbag module torchrec.distributed.grouped_position_weighted module torchrec.distributed.mc_embedding module torchrec.distributed.mc_embeddingbag module torchrec.distributed.mc_modules module torchrec.distributed.model_parallel module torchrec.distributed.planner module torchrec.distributed.planner.constants module torchrec.distributed.planner.enumerators module torchrec.distributed.planner.partitioners module torchrec.distributed.planner.perf_models module torchrec.distributed.planner.planners module torchrec.distributed.planner.proposers module torchrec.distributed.planner.shard_estimators module torchrec.distributed.planner.stats module torchrec.distributed.planner.storage_reservations module torchrec.distributed.planner.types module torchrec.distributed.planner.utils module torchrec.distributed.quant_embeddingbag module torchrec.distributed.sharding module torchrec.distributed.sharding.cw_sharding module torchrec.distributed.sharding.dp_sharding module torchrec.distributed.sharding.rw_sharding module torchrec.distributed.sharding.tw_sharding module torchrec.distributed.sharding.twcw_sharding module torchrec.distributed.sharding.twrw_sharding module torchrec.distributed.train_pipeline module torchrec.distributed.types module torchrec.distributed.utils module torchrec.fx 模块, [1] torchrec.fx.tracer module torchrec.inference 模块, [1] torchrec.inference.model_packager module torchrec.inference.modules module torchrec.models.deepfm module torchrec.modules module torchrec.modules.activation module torchrec.modules.crossnet module torchrec.modules.deepfm module torchrec.modules.embedding_configs module torchrec.modules.embedding_modules module torchrec.modules.feature_processor module torchrec.modules.lazy_extension module torchrec.modules.mc_embedding_modules module torchrec.modules.mc_modules module torchrec.modules.mlp module torchrec.modules.utils module torchrec.optim 模块, [1] torchrec.optim.clipping module torchrec.optim.fused module torchrec.optim.keyed module torchrec.optim.warmup module torchrec.quant 模块, [1] torchrec.quant.embedding_modules module torchrec.sparse 模块, [1] torchrec.sparse.jagged_tensor module torchrec_use_sync_collectives() (torchrec.distributed.comm_ops 模块中) total (torchrec.distributed.planner.types.Perf 属性) total_input_size (torchrec.distributed.comm_ops.ReduceScatterVInfo 属性), [1] total_perf (torchrec.distributed.planner.types.ShardingOption 属性) total_storage (torchrec.distributed.planner.types.ShardingOption 属性) trace() (torchrec.fx.tracer.Tracer 方法) Tracer (torchrec.fx.tracer 类) TRAIN_BASE (torchrec.distributed.types.PipelineType 属性) TRAIN_PREFETCH_SPARSE_DIST (torchrec.distributed.types.PipelineType 属性) TRAIN_SPARSE_DIST (torchrec.distributed.types.PipelineType 属性) training (torchrec.distributed.dist_data.EmbeddingsAllToOne 属性), [1] (torchrec.distributed.dist_data.EmbeddingsAllToOneReduce 属性), [1] (torchrec.distributed.dist_data.KJTAllToAll 属性), [1] (torchrec.distributed.dist_data.KJTOneToAll 属性), [1] (torchrec.distributed.dist_data.MergePooledEmbeddingsModule 属性), [1] (torchrec.distributed.dist_data.PooledEmbeddingsAllGather 属性), [1] (torchrec.distributed.dist_data.PooledEmbeddingsAllToAll 属性), [1] (torchrec.distributed.dist_data.PooledEmbeddingsReduceScatter 属性), [1] (torchrec.distributed.dist_data.SeqEmbeddingsAllToOne 属性), [1] (torchrec.distributed.dist_data.SequenceEmbeddingsAllToAll 属性), [1] (torchrec.distributed.dist_data.TensorAllToAll 属性), [1] (torchrec.distributed.dist_data.TensorValuesAllToAll 属性), [1] (torchrec.distributed.dist_data.VariableBatchPooledEmbeddingsAllToAll 属性), [1] (torchrec.distributed.dist_data.VariableBatchPooledEmbeddingsReduceScatter 属性), [1] (torchrec.distributed.embedding.ShardedEmbeddingCollection 属性) (torchrec.distributed.embedding_lookup.GroupedEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.GroupedPooledEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.InferCPUGroupedEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.InferGroupedEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.InferGroupedPooledEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.MetaInferGroupedEmbeddingsLookup 属性) (torchrec.distributed.embedding_lookup.MetaInferGroupedPooledEmbeddingsLookup 属性) (torchrec.distributed.embedding_sharding.BaseEmbeddingDist 属性) (torchrec.distributed.embedding_sharding.BaseSparseFeaturesDist 属性) (torchrec.distributed.embedding_types.BaseEmbeddingLookup 属性) (torchrec.distributed.embedding_types.BaseGroupedFeatureProcessor 属性) (torchrec.distributed.embedding_types.ShardedEmbeddingModule 属性) (torchrec.distributed.embeddingbag.ShardedEmbeddingBag 属性) (torchrec.distributed.embeddingbag.ShardedEmbeddingBagCollection 属性) (torchrec.distributed.grouped_position_weighted.GroupedPositionWeightedModule 属性) (torchrec.distributed.mc_embedding.ShardedManagedCollisionEmbeddingCollection 属性) (torchrec.distributed.mc_embeddingbag.ShardedManagedCollisionEmbeddingBagCollection 属性) (torchrec.distributed.mc_modules.ShardedManagedCollisionCollection 属性) (torchrec.distributed.model_parallel.DistributedModelParallel 属性) (torchrec.distributed.quant_embeddingbag.ShardedQuantEbcInputDist 属性) (torchrec.distributed.quant_embeddingbag.ShardedQuantEmbeddingBagCollection 属性) (torchrec.distributed.quant_embeddingbag.ShardedQuantFeatureProcessedEmbeddingBagCollection 属性) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.cw_sharding.InferCwPooledEmbeddingDistWithPermute 属性) (torchrec.distributed.sharding.dp_sharding.DpPooledEmbeddingDist 属性) (torchrec.distributed.sharding.dp_sharding.DpSparseFeaturesDist 属性) (torchrec.distributed.sharding.rw_sharding.InferRwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.rw_sharding.InferRwSparseFeaturesDist 属性) (torchrec.distributed.sharding.rw_sharding.RwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.rw_sharding.RwSparseFeaturesDist 属性) (torchrec.distributed.sharding.tw_sharding.InferTwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.tw_sharding.InferTwSparseFeaturesDist 属性) (torchrec.distributed.sharding.tw_sharding.TwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.tw_sharding.TwSparseFeaturesDist 属性) (torchrec.distributed.sharding.twrw_sharding.TwRwPooledEmbeddingDist 属性) (torchrec.distributed.sharding.twrw_sharding.TwRwSparseFeaturesDist 属性) (torchrec.distributed.types.ShardedModule 属性) (torchrec.distributed.utils.CopyableMixin 属性) (torchrec.inference.modules.PredictModule 属性) (torchrec.models.deepfm.DenseArch 属性) (torchrec.models.deepfm.FMInteractionArch 属性) (torchrec.models.deepfm.OverArch 属性) (torchrec.models.deepfm.SimpleDeepFMNN 属性) (torchrec.models.deepfm.SparseArch 属性) (torchrec.modules.activation.SwishLayerNorm 属性) (torchrec.modules.crossnet.CrossNet 属性) (torchrec.modules.crossnet.LowRankCrossNet 属性) (torchrec.modules.crossnet.LowRankMixtureCrossNet 属性) (torchrec.modules.crossnet.VectorCrossNet 属性) (torchrec.modules.deepfm.DeepFM 属性) (torchrec.modules.deepfm.FactorizationMachine 属性) (torchrec.modules.embedding_modules.EmbeddingBagCollection 属性) (torchrec.modules.embedding_modules.EmbeddingBagCollectionInterface 属性) (torchrec.modules.embedding_modules.EmbeddingCollection 属性) (torchrec.modules.embedding_modules.EmbeddingCollectionInterface 属性) (torchrec.modules.feature_processor.BaseFeatureProcessor 属性) (torchrec.modules.feature_processor.BaseGroupedFeatureProcessor 属性) (torchrec.modules.feature_processor.PositionWeightedModule 属性) (torchrec.modules.feature_processor.PositionWeightedProcessor 属性) (torchrec.modules.mc_embedding_modules.BaseManagedCollisionEmbeddingCollection 属性) (torchrec.modules.mc_embedding_modules.ManagedCollisionEmbeddingBagCollection 属性) (torchrec.modules.mc_embedding_modules.ManagedCollisionEmbeddingCollection 属性) (torchrec.modules.mc_modules.ManagedCollisionCollection 属性) (torchrec.modules.mc_modules.ManagedCollisionModule 属性) (torchrec.modules.mc_modules.MCHManagedCollisionModule 属性) (torchrec.modules.mlp.MLP 属性) (torchrec.modules.mlp.Perceptron 属性) (torchrec.quant.embedding_modules.EmbeddingBagCollection 属性) (torchrec.quant.embedding_modules.EmbeddingCollection 属性) (torchrec.quant.embedding_modules.FeatureProcessedEmbeddingBagCollection 属性) (torchrec.sparse.jagged_tensor.ComputeJTDictToKJT 属性) (torchrec.sparse.jagged_tensor.ComputeKJTToJTDict 属性) trim_torch_package_prefix_from_typename() (torchrec.inference.modules 模块中) TwCwPooledEmbeddingSharding (torchrec.distributed.sharding.twcw_sharding 类) TwPooledEmbeddingDist (torchrec.distributed.sharding.tw_sharding 类) TwPooledEmbeddingSharding (torchrec.distributed.sharding.tw_sharding 类) TwRwPooledEmbeddingDist (torchrec.distributed.sharding.twrw_sharding 类) TwRwPooledEmbeddingSharding (torchrec.distributed.sharding.twrw_sharding 类) TwRwSparseFeaturesDist (torchrec.distributed.sharding.twrw_sharding 类) TwSparseFeaturesDist (torchrec.distributed.sharding.tw_sharding 类) type (torchrec.inference.modules.BatchingMetadata 属性) U unbucketize_permute_tensor (torchrec.distributed.embedding_types.InputDistOutputs 属性) uncombined_embedding_dims() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.BaseCwEmbeddingSharding 方法) uncombined_embedding_names() (torchrec.distributed.embedding_sharding.EmbeddingSharding 方法) (torchrec.distributed.sharding.cw_sharding.BaseCwEmbeddingSharding 方法) uneven_sharding_perf_multiplier (torchrec.distributed.planner.types.Topology 属性) unflatten_kjt_list() (torchrec.sparse.jagged_tensor 模块中) UNIFORM (torchrec.distributed.planner.types.PartitionByType 属性) uniform() (torchrec.distributed.planner.utils.LuusJaakolaSearch 方法) UniformProposer (torchrec.distributed.planner.proposers 类) unpadded_lengths (torchrec.modules.utils.SequenceVBEContext 属性) unsync() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) update_metadata_and_generate_eviction_scores() (torchrec.modules.mc_modules.DistanceLFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LFU_EvictionPolicy 方法) (torchrec.modules.mc_modules.LRU_EvictionPolicy 方法) (torchrec.modules.mc_modules.MCHEvictionPolicy 方法) V VALUE (torchrec.optim.clipping.GradientClipping 属性) value (torchrec.optim.warmup.WarmupStage 属性) values() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedTensor 方法) Variable_Batch_All2All_Pooled_Req (torchrec.distributed.comm_ops 类) variable_batch_all2all_pooled_sync() (torchrec.distributed.comm_ops 模块中) Variable_Batch_All2All_Pooled_Wait (torchrec.distributed.comm_ops 类) variable_batch_alltoall_pooled() (torchrec.distributed.comm_ops 模块中) variable_batch_per_feature (torchrec.distributed.embeddingbag.EmbeddingBagCollectionContext 属性) variable_batch_size (torchrec.distributed.comm_ops.All2AllSequenceInfo 属性), [1] variable_stride_per_key() (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) VariableBatchAll2AllPooledInfo (torchrec.distributed.comm_ops 类) VariableBatchEmbeddingBagCollectionAwaitable (torchrec.distributed.embeddingbag 类) VariableBatchPooledEmbeddingsAllToAll (torchrec.distributed.dist_data 类), [1] VariableBatchPooledEmbeddingsReduceScatter (torchrec.distributed.dist_data 类), [1] VectorCrossNet (torchrec.modules.crossnet 类) W wait() (torchrec.distributed.types.Awaitable 方法) WarmupOptimizer (torchrec.optim.warmup 类) WarmupPolicy (torchrec.optim.warmup 类) WarmupStage (torchrec.optim.warmup 类) weight (torchrec.modules.embedding_configs.QuantConfig 属性) weight_init_max (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) weight_init_min (torchrec.modules.embedding_configs.BaseEmbeddingConfig 属性) weighted_feature_bwd_compute_multiplier (torchrec.distributed.planner.types.Topology 属性) weights() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) weights_or_none() (torchrec.sparse.jagged_tensor.JaggedTensor 方法) (torchrec.sparse.jagged_tensor.KeyedJaggedTensor 方法) world_size (torchrec.distributed.planner.types.Topology 属性) wrap() (torchrec.distributed.model_parallel.DataParallelWrapper 方法) (torchrec.distributed.model_parallel.DefaultDataParallelWrapper 方法) Z zero_grad() (torchrec.optim.fused.EmptyFusedOptimizer 方法) (torchrec.optim.fused.FusedOptimizer 方法) (torchrec.optim.keyed.CombinedOptimizer 方法) (torchrec.optim.keyed.KeyedOptimizerWrapper 方法) (torchrec.optim.keyed.OptimizerWrapper 方法)
