Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z _ __create_chunk_list__() (torch.distributed.tensor.DTensor method) __getstate__() (torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) __init__() (torch.distributed.FileStore method) (torch.distributed.HashStore method) (torch.distributed.PrefixStore method) (torch.distributed.Store method) (torch.distributed.TCPStore method) (torch.fx.Graph method) (torch.fx.GraphModule method) (torch.monitor.Event method) (torch.monitor.Stat method) (torch.monitor.TensorboardEventHandler method) (torch.package.PackageExporter method) (torch.package.PackageImporter method) (torch.Tensor method) (torch.utils.tensorboard.writer.SummaryWriter method) __setstate__() (torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) _assert() (in module torch) _assign_worker_ranks() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _create_jit_fn() (in module torch.cuda.jiterator) _create_multi_output_jit_fn() (in module torch.cuda.jiterator) _dump_snapshot() (in module torch.cuda.memory) _exit_barrier() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _foreach_abs() (in module torch) _foreach_abs_() (in module torch) _foreach_acos() (in module torch) _foreach_acos_() (in module torch) _foreach_asin() (in module torch) _foreach_asin_() (in module torch) _foreach_atan() (in module torch) _foreach_atan_() (in module torch) _foreach_ceil() (in module torch) _foreach_ceil_() (in module torch) _foreach_cos() (in module torch) _foreach_cos_() (in module torch) _foreach_cosh() (in module torch) _foreach_cosh_() (in module torch) _foreach_erf() (in module torch) _foreach_erf_() (in module torch) _foreach_erfc() (in module torch) _foreach_erfc_() (in module torch) _foreach_exp() (in module torch) _foreach_exp_() (in module torch) _foreach_expm1() (in module torch) _foreach_expm1_() (in module torch) _foreach_floor() (in module torch) _foreach_floor_() (in module torch) _foreach_frac() (in module torch) _foreach_frac_() (in module torch) _foreach_lgamma() (in module torch) _foreach_lgamma_() (in module torch) _foreach_log() (in module torch) _foreach_log10() (in module torch) _foreach_log10_() (in module torch) _foreach_log1p() (in module torch) _foreach_log1p_() (in module torch) _foreach_log2() (in module torch) _foreach_log2_() (in module torch) _foreach_log_() (in module torch) _foreach_neg() (in module torch) _foreach_neg_() (in module torch) _foreach_reciprocal() (in module torch) _foreach_reciprocal_() (in module torch) _foreach_round() (in module torch) _foreach_round_() (in module torch) _foreach_sigmoid() (in module torch) _foreach_sigmoid_() (in module torch) _foreach_sin() (in module torch) _foreach_sin_() (in module torch) _foreach_sinh() (in module torch) _foreach_sinh_() (in module torch) _foreach_sqrt() (in module torch) _foreach_sqrt_() (in module torch) _foreach_tan() (in module torch) _foreach_tan_() (in module torch) _foreach_trunc() (in module torch) _foreach_trunc_() (in module torch) _foreach_zero_() (in module torch) _initialize_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _KinetoProfile (class in torch.profiler) _monitor_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _record_memory_history() (in module torch.cuda.memory) _rendezvous() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _restart_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _shutdown() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _snapshot() (in module torch.cuda.memory) _start_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _stop_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) A abs() (in module torch) (torch.Tensor method) abs_() (torch.Tensor method) absolute() (in module torch) (torch.Tensor method) absolute_() (torch.Tensor method) AbsTransform (class in torch.distributions.transforms) acos() (in module torch) (torch.Tensor method) acos_() (torch.Tensor method) acosh() (in module torch) (torch.Tensor method) acosh_() (torch.Tensor method) acquire() (torch.distributed.elastic.timer.TimerClient method) active_pool() (torch.cuda.MemPoolContext static method) Adadelta (class in torch.optim) Adafactor (class in torch.optim) Adagrad (class in torch.optim) Adam (class in torch.optim) Adamax (class in torch.optim) AdamW (class in torch.optim) adapt() (torch.export.unflatten.FlatArgsAdapter method) adaptive_autorange() (torch.utils.benchmark.Timer method) adaptive_avg_pool1d() (in module torch.nn.functional) adaptive_avg_pool2d (class in torch.ao.nn.quantized.functional) adaptive_avg_pool2d() (in module torch.nn.functional) adaptive_avg_pool3d (class in torch.ao.nn.quantized.functional) adaptive_avg_pool3d() (in module torch.nn.functional) adaptive_max_pool1d() (in module torch.nn.functional) adaptive_max_pool2d() (in module torch.nn.functional) adaptive_max_pool3d() (in module torch.nn.functional) AdaptiveAvgPool1d (class in torch.nn) AdaptiveAvgPool2d (class in torch.nn) AdaptiveAvgPool3d (class in torch.nn) AdaptiveLogSoftmaxWithLoss (class in torch.nn) AdaptiveMaxPool1d (class in torch.nn) AdaptiveMaxPool2d (class in torch.nn) AdaptiveMaxPool3d (class in torch.nn) add() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) (torch.export.dynamic_shapes.AdditionalInputs method) (torch.fx.experimental.symbolic_shapes.DimConstraints method) (torch.monitor.Stat method) (torch.Tensor method) add_() (torch.Tensor method) add_audio() (torch.utils.tensorboard.writer.SummaryWriter method) add_custom_scalars() (torch.utils.tensorboard.writer.SummaryWriter method) add_dependency() (torch.package.PackageExporter method) add_done_callback() (torch.futures.Future method) add_dtype_config() (torch.ao.quantization.backend_config.BackendPatternConfig method) add_embedding() (torch.utils.tensorboard.writer.SummaryWriter method) add_equality() (torch.fx.experimental.symbolic_shapes.DimConstraints method) add_figure() (torch.utils.tensorboard.writer.SummaryWriter method) add_graph() (torch.utils.tensorboard.writer.SummaryWriter method) add_histogram() (torch.utils.tensorboard.writer.SummaryWriter method) add_hparams() (torch.utils.tensorboard.writer.SummaryWriter method) add_image() (torch.utils.tensorboard.writer.SummaryWriter method) add_images() (torch.utils.tensorboard.writer.SummaryWriter method) add_loggers() (in module torch.ao.ns._numeric_suite_fx) add_mesh() (torch.utils.tensorboard.writer.SummaryWriter method) add_metadata() (torch.profiler._KinetoProfile method) add_metadata_json() (torch.profiler._KinetoProfile method) add_module() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) add_param_group() (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) add_pr_curve() (torch.utils.tensorboard.writer.SummaryWriter method) add_pruning_method() (torch.nn.utils.prune.PruningContainer method) add_quant_dequant (class in torch.ao.quantization) add_relu() (torch.ao.ns._numeric_suite.Shadow method) add_safe_globals() (in module torch.serialization) add_scalar() (torch.ao.ns._numeric_suite.Shadow method) (torch.utils.tensorboard.writer.SummaryWriter method) add_scalars() (torch.utils.tensorboard.writer.SummaryWriter method) add_shadow_loggers() (in module torch.ao.ns._numeric_suite_fx) add_submodule() (torch.fx.GraphModule method) add_text() (torch.utils.tensorboard.writer.SummaryWriter method) add_var_to_val() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) add_video() (torch.utils.tensorboard.writer.SummaryWriter method) addbmm() (in module torch) (torch.Tensor method) addbmm_() (torch.Tensor method) addcdiv() (in module torch) (torch.Tensor method) addcdiv_() (torch.Tensor method) addcmul() (in module torch) (torch.Tensor method) addcmul_() (torch.Tensor method) AdditionalInputs (class in torch.export.dynamic_shapes) addmm() (in module torch) (in module torch.sparse) (torch.Tensor method) addmm_() (torch.Tensor method) addmv() (in module torch) (torch.Tensor method) addmv_() (torch.Tensor method) addr() (in module torch) (torch.Tensor method) addr_() (torch.Tensor method) adjoint() (in module torch) (torch.Tensor method) affine_grid() (in module torch.nn.functional) AffineQuantizedObserverBase (class in torch.ao.quantization.observer) AffineTransform (class in torch.distributions.transforms) Aggregation (class in torch.monitor) airy_ai() (in module torch.special) align_as() (torch.Tensor method) align_to() (torch.Tensor method) all() (in module torch) (torch.Tensor method) all_gather() (in module torch.distributed) all_gather_into_tensor() (in module torch.distributed) all_gather_object() (in module torch.distributed) all_input_nodes (torch.fx.Node property) all_paths() (torch.package.PackageExporter method) all_reduce() (in module torch.distributed) all_to_all() (in module torch.distributed) all_to_all_single() (in module torch.distributed) allclose() (in module torch) (torch.Tensor method) allocator (torch.cuda.MemPool property) allow_bf16_reduced_precision_reduction (in module torch.backends.cuda.matmul) allow_fp16_bf16_reduction_math_sdp() (in module torch.backends.cuda) allow_fp16_reduced_precision_reduction (in module torch.backends.cuda.matmul) allow_in_graph() (in module torch.compiler) allow_mutation_on_saved_tensors (class in torch.autograd.graph) allow_tf32 (in module torch.backends.cuda.matmul) (in module torch.backends.cudnn) allreduce_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) alpha_dropout() (in module torch.nn.functional) AlphaDropout (class in torch.nn) amax() (in module torch) (torch.Tensor method) amin() (in module torch) (torch.Tensor method) aminmax() (in module torch) (torch.Tensor method) and_masks() (in module torch.nn.attention.flex_attention) angle() (in module torch) (torch.Tensor method) annotate() (in module torch.jit) any() (in module torch) (torch.Tensor method) aoti_compile_and_package() (in module torch._inductor) aoti_load_package() (in module torch._inductor) append() (torch.distributed.Store method) (torch.fx.Node method) (torch.nn.ModuleList method) (torch.nn.ParameterList method) (torch.nn.Sequential method) apply() (torch.autograd.function.BackwardCFunction method) (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.nn.utils.prune.BasePruningMethod class method) (torch.nn.utils.prune.CustomFromMask class method) (torch.nn.utils.prune.Identity class method) (torch.nn.utils.prune.L1Unstructured class method) (torch.nn.utils.prune.LnStructured class method) (torch.nn.utils.prune.PruningContainer class method) (torch.nn.utils.prune.RandomStructured class method) (torch.nn.utils.prune.RandomUnstructured class method) (torch.optim.swa_utils.AveragedModel method) apply_() (torch.Tensor method) apply_jvp() (torch.autograd.function.BackwardCFunction method) apply_mask() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) apply_weights() (torch.onnx.ONNXProgram method) arange() (in module torch) arccos() (in module torch) (torch.Tensor method) arccos_() (torch.Tensor method) arccosh() (in module torch) (torch.Tensor method) arccosh_() (torch.Tensor method) arcsin() (in module torch) (torch.Tensor method) arcsin_() (torch.Tensor method) arcsinh() (in module torch) (torch.Tensor method) arcsinh_() (torch.Tensor method) arctan() (in module torch) (torch.Tensor method) arctan2() (in module torch) (torch.Tensor method) arctan2_() (torch.Tensor method) arctan_() (torch.Tensor method) arctanh() (in module torch) (torch.Tensor method) arctanh_() (torch.Tensor method) are_deterministic_algorithms_enabled() (in module torch) arg_constraints (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.beta.Beta attribute) (torch.distributions.binomial.Binomial attribute) (torch.distributions.categorical.Categorical attribute) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.chi2.Chi2 attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto attribute) (torch.distributions.geometric.Geometric attribute) (torch.distributions.gumbel.Gumbel attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent attribute) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.lkj_cholesky.LKJCholesky attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily attribute) (torch.distributions.multinomial.Multinomial attribute) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.negative_binomial.NegativeBinomial attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) (torch.distributions.pareto.Pareto attribute) (torch.distributions.poisson.Poisson attribute) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution attribute) (torch.distributions.uniform.Uniform attribute) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.weibull.Weibull attribute) (torch.distributions.wishart.Wishart attribute) argmax() (in module torch) (torch.Tensor method) argmin() (in module torch) (torch.Tensor method) args (torch.fx.Node property) argsort() (in module torch) (torch.Tensor method) argwhere() (in module torch) (torch.Tensor method) as_integer_ratio() (torch.SymFloat method) (torch.SymInt method) as_nested_tensor() (in module torch.nested) as_sparse_gradcheck() (in module torch.sparse) as_standardized() (torch.utils.benchmark.CallgrindStats method) as_strided() (in module torch) (torch.Tensor method) as_subclass() (torch.Tensor method) as_tensor() (in module torch) as_tuple() (torch.nn.attention.flex_attention.BlockMask method) asarray() (in module torch) asdict() (torch.onnx.verification.VerificationInfo method) ASGD (class in torch.optim) asin() (in module torch) (torch.Tensor method) asin_() (torch.Tensor method) asinh() (in module torch) (torch.Tensor method) asinh_() (torch.Tensor method) assert_allclose() (in module torch.testing) assert_close() (in module torch.testing) assume_constant_result() (in module torch.compiler) async_execution() (in module torch.distributed.rpc.functions) async_save() (in module torch.distributed.checkpoint.state_dict_saver) AsyncCheckpointerType (class in torch.distributed.checkpoint.state_dict_saver) AsyncStager (class in torch.distributed.checkpoint.staging) atan() (in module torch) (torch.Tensor method) atan2() (in module torch) (torch.Tensor method) atan2_() (torch.Tensor method) atan_() (torch.Tensor method) atanh() (in module torch) (torch.Tensor method) atanh_() (torch.Tensor method) atleast_1d() (in module torch) atleast_2d() (in module torch) atleast_3d() (in module torch) attach_out_of_memory_observer() (in module torch.mtia) Attribute (class in torch.jit) autocast (class in torch) (class in torch.cpu.amp) (class in torch.cuda.amp) AveragedModel (class in torch.optim.swa_utils) avg_pool1d() (in module torch.nn.functional) avg_pool2d (class in torch.ao.nn.quantized.functional) avg_pool2d() (in module torch.nn.functional) avg_pool3d (class in torch.ao.nn.quantized.functional) avg_pool3d() (in module torch.nn.functional) AvgPool1d (class in torch.nn) AvgPool2d (class in torch.nn) AvgPool3d (class in torch.nn) B Backend (class in torch.distributed) BackendConfig (class in torch.ao.quantization.backend_config) BackendPatternConfig (class in torch.ao.quantization.backend_config) BackendType (class in torch.distributed.rpc) backward() (in module torch.autograd) (in module torch.distributed.autograd) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction method) (torch.distributed.rpc.PyRRef method) (torch.Tensor method) backward_extended() (torch.autograd.function.NestedIOFunction method) BackwardCFunction (class in torch.autograd.function) BackwardPrefetch (class in torch.distributed.fsdp) baddbmm() (in module torch) (torch.Tensor method) baddbmm_() (torch.Tensor method) barrier() (in module torch.distributed) bartlett() (in module torch.signal.windows) bartlett_window() (in module torch) base_dist (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent attribute) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) BasePruningMethod (class in torch.nn.utils.prune) batch_isend_irecv() (in module torch.distributed) batch_norm() (in module torch.nn.functional) batch_shape (torch.distributions.distribution.Distribution property) batch_sizes (torch.nn.utils.rnn.PackedSequence attribute) batched_powerSGD_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) BatchNorm1d (class in torch.nn) BatchNorm2d (class in torch.ao.nn.quantized) (class in torch.nn) BatchNorm3d (class in torch.ao.nn.quantized) (class in torch.nn) BatchSampler (class in torch.utils.data) BCELoss (class in torch.nn) BCEWithLogitsLoss (class in torch.nn) benchmark (in module torch.backends.cudnn) benchmark_limit (in module torch.backends.cudnn) Bernoulli (class in torch.distributions.bernoulli) bernoulli() (in module torch) (torch.Tensor method) bernoulli_() (torch.Tensor method) bessel_j0() (in module torch.special) bessel_j1() (in module torch.special) Beta (class in torch.distributions.beta) bf16_compress_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) bf16_compress_wrapper() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) bfloat16() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BFloat16Storage (class in torch) Bilinear (class in torch.nn) bilinear() (in module torch.nn.functional) binary_cross_entropy() (in module torch.nn.functional) binary_cross_entropy_with_logits() (in module torch.nn.functional) bincount() (in module torch) (torch.Tensor method) bind_symbols() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) Binomial (class in torch.distributions.binomial) bitwise_and() (in module torch) (torch.Tensor method) bitwise_and_() (torch.Tensor method) bitwise_left_shift() (in module torch) (torch.Tensor method) bitwise_left_shift_() (torch.Tensor method) bitwise_not() (in module torch) (torch.Tensor method) bitwise_not_() (torch.Tensor method) bitwise_or() (in module torch) (torch.Tensor method) bitwise_or_() (torch.Tensor method) bitwise_right_shift() (in module torch) (torch.Tensor method) bitwise_right_shift_() (torch.Tensor method) bitwise_xor() (in module torch) (torch.Tensor method) bitwise_xor_() (torch.Tensor method) blackman() (in module torch.signal.windows) blackman_window() (in module torch) block_diag() (in module torch) BLOCK_SIZE (torch.nn.attention.flex_attention.BlockMask attribute) blocked_autorange() (torch.utils.benchmark.Timer method) BlockingAsyncStager (class in torch.distributed.checkpoint.staging) BlockMask (class in torch.nn.attention.flex_attention) bmm() (in module torch) (torch.Tensor method) BNReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) BNReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) bool() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BoolStorage (class in torch) bound_sympy() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) boxed() (torch.distributed.Work method) boxed_run() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) breakpoint() (in module torch.distributed) broadcast() (in module torch.cuda.comm) (in module torch.distributed) broadcast_coalesced() (in module torch.cuda.comm) broadcast_object_list() (in module torch.distributed) broadcast_shapes() (in module torch) broadcast_tensors() (in module torch) broadcast_to() (in module torch) (torch.Tensor method) BroadcastingTorchSaveReader (class in torch.distributed.checkpoint.format_utils) bucketize() (in module torch) Buffer (class in torch.nn.parameter) buffer() (in module torch.distributed.GradBucket) buffers() (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) build() (torch.distributed.elastic.rendezvous.api.RendezvousStoreInfo static method) build_stage() (in module torch.distributed.pipelining.stage) BuildExtension() (in module torch.utils.cpp_extension) byte() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BytesIOWriteData (class in torch.distributed.checkpoint.planner) ByteStorage (class in torch) byteswap() (torch.UntypedStorage method) C C10dRendezvousBackend (class in torch.distributed.elastic.rendezvous.c10d_rendezvous_backend) cached() (in module torch.nn.utils.parametrize) caching_allocator_alloc() (in module torch.cuda) caching_allocator_delete() (in module torch.cuda) caching_allocator_enable() (in module torch.cuda.memory) calculate_gain() (in module torch.nn.init) calculate_qparams() (torch.ao.quantization.observer.AffineQuantizedObserverBase method) (torch.ao.quantization.observer.MinMaxObserver method) call_function() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) call_method() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) call_module() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Tracer method) (torch.fx.Transformer method) CallgrindStats (class in torch.utils.benchmark) CallMethodKey (class in torch.fx.experimental.symbolic_shapes) can_cast() (in module torch) can_device_access_peer() (in module torch.cuda) can_use_cudnn_attention() (in module torch.backends.cuda) can_use_efficient_attention() (in module torch.backends.cuda) can_use_flash_attention() (in module torch.backends.cuda) canonicalize_bool_expr() (in module torch.fx.experimental.symbolic_shapes) capture_begin() (torch.cuda.CUDAGraph method) capture_end() (torch.cuda.CUDAGraph method) cartesian_prod() (in module torch) cat (in module torch.distributions.constraints) cat() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) Categorical (class in torch.distributions.categorical) CatTransform (class in torch.distributions.transforms) Cauchy (class in torch.distributions.cauchy) cauchy_() (torch.Tensor method) causal_lower_right() (in module torch.nn.attention.bias) causal_upper_left() (in module torch.nn.attention.bias) CausalBias (class in torch.nn.attention.bias) CausalVariant (class in torch.nn.attention.bias) ccol_indices() (torch.Tensor method) cdf() (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.laplace.Laplace method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.normal.Normal method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) cdist() (in module torch) cdouble() (torch.Tensor method) ceil() (in module torch) (torch.Tensor method) ceil_() (torch.Tensor method) celu (class in torch.ao.nn.quantized.functional) CELU (class in torch.nn) celu() (in module torch.nn.functional) cfloat() (torch.Tensor method) chain_matmul() (in module torch) ChainDataset (class in torch.utils.data) ChainedScheduler (class in torch.optim.lr_scheduler) chalf() (torch.Tensor method) change_current_allocator() (in module torch.cuda) ChannelShuffle (class in torch.nn) char() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CharStorage (class in torch) check() (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) (torch.distributions.constraints.Constraint method) check_consistent() (in module torch.fx.experimental.symbolic_shapes) check_equal() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) check_export_model_diff (class in torch.onnx.verification) check_is_root() (torch.distributed.fsdp.FullyShardedDataParallel method) check_sparse_tensor_invariants (class in torch.sparse) checkpoint() (in module torch.utils.checkpoint) checkpoint_id (torch.distributed.checkpoint.FileSystemReader property) checkpoint_sequential() (in module torch.utils.checkpoint) CheckpointPolicy (class in torch.utils.checkpoint) Chi2 (class in torch.distributions.chi2) ChildFailedError (class in torch.distributed.elastic.multiprocessing.errors) children() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) cholesky() (in module torch) (in module torch.linalg) (torch.Tensor method) cholesky_ex() (in module torch.linalg) cholesky_inverse() (in module torch) (torch.Tensor method) cholesky_solve() (in module torch) (torch.Tensor method) chunk() (in module torch) (torch.Tensor method) CircularPad1d (class in torch.nn) CircularPad2d (class in torch.nn) CircularPad3d (class in torch.nn) clamp (class in torch.ao.nn.quantized.functional) clamp() (in module torch) (torch.Tensor method) clamp_() (torch.Tensor method) cleanup() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) clear() (in module torch.backends.cuda.cufft_plan_cache) (torch.autograd.profiler_util.StringTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) clear_safe_globals() (in module torch.serialization) clear_timers() (torch.distributed.elastic.timer.TimerServer method) clip() (in module torch) (torch.Tensor method) clip_() (torch.Tensor method) clip_grad_norm() (in module torch.nn.utils) clip_grad_norm_() (in module torch.nn.utils) (torch.distributed.fsdp.FullyShardedDataParallel method) clip_grad_value_() (in module torch.nn.utils) clip_grads_with_norm_() (in module torch.nn.utils) clock_rate() (in module torch.cuda) clone() (in module torch) (torch.autograd.grad_mode.inference_mode method) (torch.autograd.grad_mode.set_grad_enabled method) (torch.autograd.grad_mode.set_multithreading_enabled method) (torch.distributed.Store method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) clone_state() (torch.Generator method) close (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) close() (torch.package.PackageExporter method) (torch.utils.tensorboard.writer.SummaryWriter method) cls_to_become (torch.nn.LazyBatchNorm1d attribute) (torch.nn.LazyBatchNorm2d attribute) (torch.nn.LazyBatchNorm3d attribute) (torch.nn.LazyConv1d attribute) (torch.nn.LazyConv2d attribute) (torch.nn.LazyConv3d attribute) (torch.nn.LazyConvTranspose1d attribute) (torch.nn.LazyConvTranspose2d attribute) (torch.nn.LazyConvTranspose3d attribute) (torch.nn.LazyInstanceNorm1d attribute) (torch.nn.LazyInstanceNorm2d attribute) (torch.nn.LazyInstanceNorm3d attribute) (torch.nn.LazyLinear attribute) (torch.nn.parameter.UninitializedParameter attribute) coalesce() (torch.Tensor method) code (torch.fx.GraphModule property) (torch.jit.ScriptModule property) code_with_constants (torch.jit.ScriptModule property) col_indices() (torch.Tensor method) collate() (in module torch.utils.data._utils.collate) collect_all() (in module torch.futures) collect_callgrind() (torch.utils.benchmark.Timer method) colorize() (torch.utils.benchmark.Compare method) column_stack() (in module torch) ColwiseParallel (class in torch.distributed.tensor.parallel) combinations() (in module torch) CommDebugMode (class in torch.distributed.tensor.debug) commit_tensor() (torch.distributed.checkpoint.LoadPlanner method) Compare (class in torch.utils.benchmark) compare_model_outputs() (in module torch.ao.ns._numeric_suite) compare_model_stub() (in module torch.ao.ns._numeric_suite) compare_results (class in torch.ao.quantization) compare_set() (torch.distributed.Store method) compare_weights() (in module torch.ao.ns._numeric_suite) compile() (in module torch) (in module torch.compiler) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) compile_shader() (in module torch.mps) compiled_with_cxx11_abi() (in module torch) complex() (in module torch) complex_double() (torch.TypedStorage method) (torch.UntypedStorage method) complex_float() (torch.TypedStorage method) (torch.UntypedStorage method) ComplexDoubleStorage (class in torch) ComplexFloatStorage (class in torch) component_distribution (torch.distributions.mixture_same_family.MixtureSameFamily property) ComposeTransform (class in torch.distributions.transforms) compute_cosine_similarity() (in module torch.ao.ns.fx.utils) compute_mask() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) compute_normalized_l2_error() (in module torch.ao.ns.fx.utils) compute_sqnr() (in module torch.ao.ns.fx.utils) compute_unbacked_bindings() (in module torch.fx.experimental.symbolic_shapes) compute_values() (torch.onnx.ONNXProgram method) concat() (in module torch) ConcatDataset (class in torch.utils.data) concatenate() (in module torch) concentration (torch.distributions.inverse_gamma.InverseGamma property) concentration0 (torch.distributions.beta.Beta property) concentration1 (torch.distributions.beta.Beta property) cond() (in module torch) (in module torch._higher_order_ops.cond) (in module torch.linalg) configs (torch.ao.quantization.backend_config.BackendConfig property) configure() (in module torch.distributed.elastic.metrics) (in module torch.distributed.elastic.timer) confirmed_by_owner() (torch.distributed.rpc.PyRRef method) conj() (in module torch) (torch.Tensor method) conj_physical() (in module torch) (torch.Tensor method) conj_physical_() (torch.Tensor method) conjugate() (torch.SymFloat method) ConsoleMetricHandler (class in torch.distributed.elastic.metrics.api) consolidate_state_dict() (torch.distributed.optim.ZeroRedundancyOptimizer method) constant_() (in module torch.nn.init) ConstantLR (class in torch.optim.lr_scheduler) ConstantPad1d (class in torch.nn) ConstantPad2d (class in torch.nn) ConstantPad3d (class in torch.nn) constrain_range() (in module torch.fx.experimental.symbolic_shapes) constrain_unify() (in module torch.fx.experimental.symbolic_shapes) Constraint (class in torch.distributions.constraints) (in module torch.export) ConstraintRegistry (class in torch.distributions.constraint_registry) construct_and_record_rdzv_event() (in module torch.distributed.elastic.events) context (class in torch.distributed.autograd) context_parallel() (in module torch.distributed.tensor.experimental) contiguous() (torch.Tensor method) ContinuousBernoulli (class in torch.distributions.continuous_bernoulli) Conv1d (class in torch.ao.nn.quantized) conv1d (class in torch.ao.nn.quantized.functional) Conv1d (class in torch.nn) conv1d() (in module torch.nn.functional) Conv2d (class in torch.ao.nn.qat) (class in torch.ao.nn.quantized) conv2d (class in torch.ao.nn.quantized.functional) Conv2d (class in torch.nn) conv2d() (in module torch.nn.functional) Conv3d (class in torch.ao.nn.qat) (class in torch.ao.nn.quantized) conv3d (class in torch.ao.nn.quantized.functional) Conv3d (class in torch.nn) conv3d() (in module torch.nn.functional) conv_transpose1d() (in module torch.nn.functional) conv_transpose2d() (in module torch.nn.functional) conv_transpose3d() (in module torch.nn.functional) ConvBn1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBn2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBn3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) convert (class in torch.ao.quantization) convert() (torch.ao.quantization.observer.AffineQuantizedObserverBase method) convert_conv2d_weight_memory_format() (in module torch.nn.utils) convert_conv3d_weight_memory_format() (in module torch.nn.utils) convert_fx (class in torch.ao.quantization.quantize_fx) convert_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) convert_sync_batchnorm() (torch.nn.SyncBatchNorm class method) ConvertCustomConfig (class in torch.ao.quantization.fx.custom_config) ConvertIntKey (class in torch.fx.experimental.symbolic_shapes) ConvReLU1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) ConvReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) ConvReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) ConvTranspose1d (class in torch.ao.nn.quantized) (class in torch.nn) ConvTranspose2d (class in torch.ao.nn.quantized) (class in torch.nn) ConvTranspose3d (class in torch.ao.nn.quantized) (class in torch.nn) copy() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ParameterDict method) copy_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) copysign() (in module torch) (torch.Tensor method) copysign_() (torch.Tensor method) CorrCholeskyTransform (class in torch.distributions.transforms) corrcoef() (in module torch) (torch.Tensor method) cos() (in module torch) (torch.Tensor method) cos_() (torch.Tensor method) cosh() (in module torch) (torch.Tensor method) cosh_() (torch.Tensor method) cosine() (in module torch.signal.windows) cosine_embedding_loss() (in module torch.nn.functional) cosine_similarity() (in module torch.nn.functional) CosineAnnealingLR (class in torch.optim.lr_scheduler) CosineAnnealingWarmRestarts (class in torch.optim.lr_scheduler) CosineEmbeddingLoss (class in torch.nn) CosineSimilarity (class in torch.nn) count (torch.monitor.Stat property) count() (torch.autograd.forward_ad.UnpackedDualTensor method) (torch.autograd.profiler_util.Kernel method) (torch.jit.Attribute method) (torch.nn.utils.rnn.PackedSequence method) (torch.Size method) count_nonzero() (in module torch) (torch.Tensor method) counts() (torch.utils.benchmark.CallgrindStats method) cov() (in module torch) (torch.Tensor method) covariance_matrix (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) CppExtension() (in module torch.utils.cpp_extension) cpu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CPUOffload (class in torch.distributed.fsdp) CPUOffloadPolicy (class in torch.distributed.fsdp) create_arg() (torch.fx.Tracer method) create_args_for_root() (torch.fx.Tracer method) create_backend() (in module torch.distributed.elastic.rendezvous.c10d_rendezvous_backend) (in module torch.distributed.elastic.rendezvous.etcd_rendezvous_backend) create_block_mask() (in module torch.nn.attention.flex_attention) create_global_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) create_handler() (in module torch.distributed.elastic.rendezvous.dynamic_rendezvous) create_healthcheck_server() (in module torch.distributed.elastic.agent.server.health_check_server) create_local_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) create_mask() (in module torch.nn.attention.flex_attention) create_nested_block_mask() (in module torch.nn.attention.flex_attention) create_node() (torch.fx.Graph method) (torch.fx.Tracer method) create_proxy() (torch.fx.Tracer method) create_selective_checkpoint_contexts() (in module torch.utils.checkpoint) create_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symbolic_sizes_strides_storage_offset() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symboolnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symfloatnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symintnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symbool() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symfloat() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unspecified_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unspecified_symint_and_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) cross() (in module torch) (in module torch.linalg) (torch.Tensor method) cross_entropy() (in module torch.nn.functional) CrossEntropyLoss (class in torch.nn) crow_indices() (torch.Tensor method) ctc_loss() (in module torch.nn.functional) CTCLoss (class in torch.nn) cuda() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CUDAExtension() (in module torch.utils.cpp_extension) CUDAGraph (class in torch.cuda) cudagraph_mark_step_begin() (in module torch.compiler) CUDAPluggableAllocator (class in torch.cuda) cudart() (in module torch.cuda) cudnn_sdp_enabled() (in module torch.backends.cuda) cufft_plan_cache (in module torch.backends.cuda) cummax() (in module torch) (torch.Tensor method) cummin() (in module torch) (torch.Tensor method) cumprod() (in module torch) (torch.Tensor method) cumprod_() (torch.Tensor method) cumsum() (in module torch) (torch.Tensor method) cumsum_() (torch.Tensor method) cumulative_trapezoid() (in module torch) CumulativeDistributionTransform (class in torch.distributions.transforms) current_accelerator() (in module torch.accelerator) current_allocated_memory() (in module torch.mps) current_blas_handle() (in module torch.cuda) current_device() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) current_device_idx() (in module torch.accelerator) current_device_index() (in module torch.accelerator) current_step() (torch.autograd.profiler.KinetoStepTracker class method) current_stream() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) custom_bwd() (in module torch.amp) (in module torch.cuda.amp) custom_from_mask() (in module torch.nn.utils.prune) custom_fwd() (in module torch.amp) (in module torch.cuda.amp) CUSTOM_KEY (in module torch.ao.quantization) custom_op() (in module torch.library) CustomDecompTable (class in torch.export.decomp_utils) CustomFromMask (class in torch.nn.utils.prune) CustomObjArgument (class in torch.export.graph_signature) CustomOpDef (class in torch._library.custom_ops) CyclicLR (class in torch.optim.lr_scheduler) D data (torch.monitor.Event property) (torch.nn.utils.rnn.PackedSequence attribute) data_parallel() (in module torch.nn.parallel) data_ptr() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) data_value_t (class in torch.monitor) DataLoader (class in torch.utils.data) DataParallel (class in torch.nn) Dataset (class in torch.utils.data) dcp_to_torch_save() (in module torch.distributed.checkpoint.format_utils) dde_suppressed() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) debug_dump() (torch.cuda.CUDAGraph method) debug_unwrap() (in module torch.func) default_activation_only_qconfig (in module torch.ao.quantization.qconfig) default_collate() (in module torch.utils.data) default_convert() (in module torch.utils.data) default_debug_observer (in module torch.ao.quantization.observer) default_debug_qconfig (in module torch.ao.quantization.qconfig) default_decompositions() (in module torch.export.exported_program) default_dynamic_qconfig (in module torch.ao.quantization.qconfig) default_dynamic_quant_observer (in module torch.ao.quantization.observer) default_eval_fn (class in torch.ao.quantization) default_factory (torch.autograd.profiler_util.StringTable attribute) default_fake_quant (in module torch.ao.quantization.fake_quantize) default_float_qparams_observer (in module torch.ao.quantization.observer) default_fused_act_fake_quant (in module torch.ao.quantization.fake_quantize) default_fused_per_channel_wt_fake_quant (in module torch.ao.quantization.fake_quantize) default_fused_wt_fake_quant (in module torch.ao.quantization.fake_quantize) default_generator (torch.torch attribute) default_histogram_fake_quant (in module torch.ao.quantization.fake_quantize) default_histogram_observer (in module torch.ao.quantization.observer) default_observer (in module torch.ao.quantization.observer) default_per_channel_qconfig (in module torch.ao.quantization.qconfig) default_per_channel_weight_fake_quant (in module torch.ao.quantization.fake_quantize) default_per_channel_weight_observer (in module torch.ao.quantization.observer) default_placeholder_observer (in module torch.ao.quantization.observer) default_qat_qconfig (in module torch.ao.quantization.qconfig) default_qat_qconfig_v2 (in module torch.ao.quantization.qconfig) default_qconfig (in module torch.ao.quantization.qconfig) default_stream() (in module torch.cuda) (in module torch.mtia) default_weight_fake_quant (in module torch.ao.quantization.fake_quantize) default_weight_observer (in module torch.ao.quantization.observer) default_weight_only_qconfig (in module torch.ao.quantization.qconfig) DefaultLoadPlanner (class in torch.distributed.checkpoint) DefaultLogsSpecs (class in torch.distributed.elastic.multiprocessing.api) DefaultSavePlanner (class in torch.distributed.checkpoint) defer_runtime_assert() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) DeferredMtiaCallError define() (in module torch.library) (torch.library.Library method) deg2rad() (in module torch) (torch.Tensor method) delete_all_unused_submodules() (torch.fx.GraphModule method) delete_key() (torch.distributed.Store method) delete_submodule() (torch.fx.GraphModule method) delta() (torch.utils.benchmark.CallgrindStats method) denied_modules() (torch.package.PackageExporter method) denoise() (torch.utils.benchmark.FunctionCounts method) dense_dim() (torch.Tensor method) deny() (torch.package.PackageExporter method) dependency_graph_string() (torch.package.PackageExporter method) dependent_property (in module torch.distributions.constraints) dequantize() (in module torch) (torch.ao.nn.quantizable.MultiheadAttention method) (torch.Tensor method) DeQuantStub (class in torch.ao.quantization) deregister_handle() (torch.cuda.gds.GdsFile method) deserialize_symexpr() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) det() (in module torch) (in module torch.linalg) (torch.Tensor method) detach() (torch.Tensor method) detach_() (torch.Tensor method) detect_anomaly (class in torch.autograd) deterministic (in module torch.backends.cudnn) device (class in torch) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) (torch.autograd.profiler_util.Kernel attribute) (torch.Generator attribute) (torch.Tensor attribute) (torch.TypedStorage property) (torch.UntypedStorage attribute) device_count() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) device_index (class in torch.accelerator) device_maps (torch.distributed.rpc.TensorPipeRpcBackendOptions property) device_memory_used() (in module torch.cuda) device_mesh (torch.distributed.tensor.DTensor property) device_of (class in torch.cuda) (class in torch.xpu) DeviceMesh (class in torch.distributed.device_mesh) devices (torch.distributed.rpc.TensorPipeRpcBackendOptions property) df (torch.distributions.chi2.Chi2 property) diag() (in module torch) (torch.Tensor method) diag_embed() (in module torch) (torch.Tensor method) diagflat() (in module torch) (torch.Tensor method) diagonal() (in module torch) (in module torch.linalg) (torch.Tensor method) diagonal_scatter() (in module torch) (torch.Tensor method) diff() (in module torch) (torch.Tensor method) digamma() (in module torch) (in module torch.special) (torch.Tensor method) digamma_() (torch.Tensor method) Dim (class in torch.export.dynamic_shapes) dim (torch.distributed.tensor.placement_types.Shard attribute) dim() (torch.Tensor method) dim_order() (torch.Tensor method) DimConstraints (class in torch.fx.experimental.symbolic_shapes) DimDynamic (class in torch.fx.experimental.symbolic_shapes) dims() (in module torch.export) dirac_() (in module torch.nn.init) Directory (class in torch.package) Dirichlet (class in torch.distributions.dirichlet) disable() (in module torch.compiler) (torch.sparse.check_sparse_tensor_invariants static method) disable_fake_quant (class in torch.ao.quantization.fake_quantize) disable_observer (class in torch.ao.quantization.fake_quantize) disable_saved_tensors_hooks (class in torch.autograd.graph) dist() (in module torch) (torch.Tensor method) DistBackendError (class in torch.distributed) DistError (class in torch.distributed) DistNetworkError (class in torch.distributed) distribute_module() (in module torch.distributed.tensor) distribute_tensor() (in module torch.distributed.tensor) DistributedDataParallel (class in torch.nn.parallel) DistributedOptimizer (class in torch.distributed.optim) DistributedSampler (class in torch.utils.data.distributed) Distribution (class in torch.distributions.distribution) DistStoreError (class in torch.distributed) div() (in module torch) (torch.Tensor method) div_() (torch.Tensor method) divide() (in module torch) (torch.Tensor method) divide_() (torch.Tensor method) DivideByKey (class in torch.fx.experimental.symbolic_shapes) done() (torch.futures.Future method) dot() (in module torch) (torch.Tensor method) double() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) DoubleStorage (class in torch) download_url_to_file() (in module torch.hub) draw() (torch.quasirandom.SobolEngine method) draw_base2() (torch.quasirandom.SobolEngine method) driver_allocated_memory() (in module torch.mps) Dropout (class in torch.nn) dropout() (in module torch.nn.functional) Dropout1d (class in torch.nn) dropout1d() (in module torch.nn.functional) Dropout2d (class in torch.nn) dropout2d() (in module torch.nn.functional) Dropout3d (class in torch.nn) dropout3d() (in module torch.nn.functional) dsplit() (in module torch) (torch.Tensor method) dstack() (in module torch) DTensor (class in torch.distributed.tensor) dtype (class in torch) (torch.BFloat16Storage attribute) (torch.BoolStorage attribute) (torch.ByteStorage attribute) (torch.CharStorage attribute) (torch.ComplexDoubleStorage attribute) (torch.ComplexFloatStorage attribute) (torch.DoubleStorage attribute) (torch.FloatStorage attribute) (torch.HalfStorage attribute) (torch.IntStorage attribute) (torch.LongStorage attribute) (torch.QInt32Storage attribute) (torch.QInt8Storage attribute) (torch.QUInt2x4Storage attribute) (torch.QUInt4x2Storage attribute) (torch.QUInt8Storage attribute) (torch.ShortStorage attribute) (torch.TypedStorage attribute) dtype() (torch.onnx.JitScalarType method) DTypeConfig (class in torch.ao.quantization.backend_config) DTypeWithConstraints (class in torch.ao.quantization.backend_config) dual_level (class in torch.autograd.forward_ad) duration (torch.autograd.profiler_util.Kernel attribute) dynamic_shapes() (torch.export.dynamic_shapes.AdditionalInputs method) (torch.export.dynamic_shapes.ShapesCollection method) DynamicMetaLoadPlanner (class in torch.distributed.checkpoint.format_utils) DynamicRendezvousHandler (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) dynamo_export() (in module torch.onnx) E eig() (in module torch.linalg) eigh() (in module torch.linalg) eigvals() (in module torch.linalg) eigvalsh() (in module torch.linalg) einsum() (in module torch) elapsed_time() (torch.cuda.Event method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) elapsed_us() (torch.autograd.profiler_util.Interval method) ElasticAgent (class in torch.distributed.elastic.agent.server) element_size() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) eliminate_dead_code() (torch.fx.Graph method) ELU (class in torch.ao.nn.quantized) elu (class in torch.ao.nn.quantized.functional) ELU (class in torch.nn) elu() (in module torch.nn.functional) elu_() (in module torch.nn.functional) Embedding (class in torch.ao.nn.quantized) (class in torch.nn) embedding() (in module torch.nn.functional) embedding_bag() (in module torch.nn.functional) EmbeddingBag (class in torch.ao.nn.quantized) (class in torch.nn) emit_itt (class in torch.autograd.profiler) emit_nvtx (class in torch.autograd.profiler) empty() (in module torch) (in module torch.distributed.tensor) empty_cache() (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) empty_like() (in module torch) empty_strided() (in module torch) EmptyMatchError (class in torch.package) enable() (in module torch.cuda.tunable) (torch.sparse.check_sparse_tensor_invariants static method) enable_cuda_sanitizer() (in module torch.cuda._sanitizer) enable_cudnn_sdp() (in module torch.backends.cuda) enable_debug_mode() (torch.cuda.CUDAGraph method) enable_fake_mode() (in module torch.onnx) enable_fake_quant (class in torch.ao.quantization.fake_quantize) enable_flash_sdp() (in module torch.backends.cuda) enable_grad (class in torch) enable_math_sdp() (in module torch.backends.cuda) enable_mem_efficient_sdp() (in module torch.backends.cuda) enable_observer (class in torch.ao.quantization.fake_quantize) enable_onednn_fusion() (in module torch.jit) enabled (in module torch.backends.cudnn) (in module torch.backends.opt_einsum) EnforceUnique (class in torch.autograd.profiler) enter_dual_level() (in module torch.autograd.forward_ad) entr() (in module torch.special) entropy() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exp_family.ExponentialFamily method) (torch.distributions.exponential.Exponential method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.inverse_gamma.InverseGamma method) (torch.distributions.kumaraswamy.Kumaraswamy method) (torch.distributions.laplace.Laplace method) (torch.distributions.log_normal.LogNormal method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.pareto.Pareto method) (torch.distributions.studentT.StudentT method) (torch.distributions.uniform.Uniform method) (torch.distributions.weibull.Weibull method) (torch.distributions.wishart.Wishart method) enumerate_support() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.distribution.Distribution method) (torch.distributions.independent.Independent method) (torch.distributions.one_hot_categorical.OneHotCategorical method) environment variable PYTORCH_JIT TORCH_COMPILE_JOB_ID eq() (in module torch) (torch.Tensor method) eq_() (torch.Tensor method) equal() (in module torch) (torch.Tensor method) EqualityConstraint (class in torch.fx.experimental.symbolic_shapes) erase_node() (torch.fx.Graph method) erase_step_count() (torch.autograd.profiler.KinetoStepTracker class method) erf() (in module torch) (in module torch.special) (torch.Tensor method) erf_() (torch.Tensor method) erfc() (in module torch) (in module torch.special) (torch.Tensor method) erfc_() (torch.Tensor method) erfcx() (in module torch.special) erfinv() (in module torch) (in module torch.special) (torch.Tensor method) erfinv_() (torch.Tensor method) ErrorHandler (class in torch.distributed.elastic.multiprocessing.errors) EtcdRendezvousBackend (class in torch.distributed.elastic.rendezvous.etcd_rendezvous_backend) EtcdRendezvousHandler (class in torch.distributed.elastic.rendezvous.etcd_rendezvous) EtcdServer (class in torch.distributed.elastic.rendezvous.etcd_server) EtcdStore (class in torch.distributed.elastic.rendezvous.etcd_store) eval() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) evaluate_guards_expression() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_guards_for_args() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_sym_node() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_symexpr() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) Event (class in torch) (class in torch.cuda) (class in torch.distributed.elastic.events.api) (class in torch.monitor) (class in torch.mps.event) (class in torch.mtia) (class in torch.xpu) event_shape (torch.distributions.distribution.Distribution property) EventHandlerHandle (class in torch.monitor) EventMetadataValue (in module torch.distributed.elastic.events.api) events() (torch.profiler._KinetoProfile method) EventSource (class in torch.distributed.elastic.events.api) exception() (torch.distributed.Work method) exit_dual_level() (in module torch.autograd.forward_ad) exp() (in module torch) (torch.Tensor method) exp2() (in module torch) (in module torch.special) exp_() (torch.Tensor method) expand() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.chi2.Chi2 method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.inverse_gamma.InverseGamma method) (torch.distributions.kumaraswamy.Kumaraswamy method) (torch.distributions.laplace.Laplace method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.log_normal.LogNormal method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.pareto.Pareto method) (torch.distributions.poisson.Poisson method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli method) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.von_mises.VonMises method) (torch.distributions.weibull.Weibull method) (torch.distributions.wishart.Wishart method) (torch.Tensor method) expand_as() (torch.Tensor method) expires() (in module torch.distributed.elastic.timer) expit() (in module torch.special) expm1() (in module torch) (in module torch.special) (torch.Tensor method) expm1_() (torch.Tensor method) Exponential (class in torch.distributions.exponential) exponential() (in module torch.signal.windows) exponential_() (torch.Tensor method) ExponentialFamily (class in torch.distributions.exp_family) ExponentialLR (class in torch.optim.lr_scheduler) export() (in module torch.export) (in module torch.jit) (in module torch.onnx) export_chrome_trace() (torch.autograd.profiler.profile method) (torch.profiler._KinetoProfile method) export_memory_timeline() (torch.profiler._KinetoProfile method) export_stacks() (torch.profiler._KinetoProfile method) ExportBackwardSignature (class in torch.export) ExportedProgram (class in torch.export) ExportGraphSignature (class in torch.export) (class in torch.export.graph_signature) ExportOptions (class in torch.onnx) ExpTransform (class in torch.distributions.transforms) extend() (torch.nn.ModuleList method) (torch.nn.ParameterList method) (torch.nn.Sequential method) extend_logger_results_with_comparison() (in module torch.ao.ns._numeric_suite_fx) extend_results() (torch.utils.benchmark.Compare method) extern() (torch.package.PackageExporter method) ExternalStream (class in torch.cuda) externed_modules() (torch.package.PackageExporter method) extra_repr() (torch.ao.quantization.fake_quantize.FixedQParamsFakeQuantize method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.nn.modules.normalization.RMSNorm method) (torch.nn.RMSNorm method) (torch.optim.swa_utils.AveragedModel method) extract_logger_info() (in module torch.ao.ns._numeric_suite_fx) extract_results_from_loggers (class in torch.ao.quantization) extract_results_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) extract_shadow_logger_info() (in module torch.ao.ns._numeric_suite_fx) extract_weights() (in module torch.ao.ns._numeric_suite_fx) eye() (in module torch) eye_() (in module torch.nn.init) F fake_quantize_per_channel_affine() (in module torch) fake_quantize_per_tensor_affine() (in module torch) FakeQuantize (class in torch.ao.quantization.fake_quantize) FakeQuantizeBase (class in torch.ao.quantization.fake_quantize) fallback() (torch.library.Library method) fallthrough_kernel() (in module torch.library) fast_forward() (torch.quasirandom.SobolEngine method) feature_alpha_dropout() (in module torch.nn.functional) FeatureAlphaDropout (class in torch.nn) fetch_args_kwargs_from_env() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) fetch_attr() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) fft() (in module torch.fft) fft2() (in module torch.fft) fftfreq() (in module torch.fft) fftn() (in module torch.fft) fftshift() (in module torch.fft) file_structure() (torch.package.PackageImporter method) filename (torch.TypedStorage property) (torch.UntypedStorage property) FileStore (class in torch.distributed) FileSystemReader (class in torch.distributed.checkpoint) FileSystemWriter (class in torch.distributed.checkpoint) FileTimerClient (class in torch.distributed.elastic.timer) FileTimerServer (class in torch.distributed.elastic.timer) fill_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) fill_diagonal_() (torch.Tensor method) fill_uninitialized_memory (in module torch.utils.deterministic) filter() (torch.utils.benchmark.FunctionCounts method) find_mismatch() (in module torch.onnx.verification) find_nodes() (torch.fx.Graph method) finish() (torch.distributed.checkpoint.StorageWriter method) finish_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) FisherSnedecor (class in torch.distributions.fishersnedecor) fix() (in module torch) (torch.Tensor method) fix_() (torch.Tensor method) FixedQParamsFakeQuantize (class in torch.ao.quantization.fake_quantize) flags() (in module torch.backends.nnpack) flash_sdp_enabled() (in module torch.backends.cuda) FlatArgsAdapter (class in torch.export.unflatten) Flatten (class in torch.nn) flatten() (in module torch) (torch.Tensor method) flatten_parameters() (torch.nn.RNNBase method) flatten_sharded_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) flex_attention() (in module torch.nn.attention.flex_attention) flip() (in module torch) (torch.Tensor method) fliplr() (in module torch) (torch.Tensor method) flipud() (in module torch) (torch.Tensor method) float() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) float16_dynamic_qconfig (in module torch.ao.quantization.qconfig) float16_static_qconfig (in module torch.ao.quantization.qconfig) float8_e4m3fn() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e4m3fnuz() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e5m2() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e5m2fnuz() (torch.TypedStorage method) (torch.UntypedStorage method) float_power() (in module torch) (torch.Tensor method) float_power_() (torch.Tensor method) float_qparams_weight_only_qconfig (in module torch.ao.quantization.qconfig) FloatFunctional (class in torch.ao.nn.quantized) FloatStorage (class in torch) floor() (in module torch) (torch.Tensor method) floor_() (torch.Tensor method) floor_divide() (in module torch) (torch.Tensor method) floor_divide_() (torch.Tensor method) flush() (torch.utils.tensorboard.writer.SummaryWriter method) fmax() (in module torch) (torch.Tensor method) fmin() (in module torch) (torch.Tensor method) fmod() (in module torch) (torch.Tensor method) fmod_() (torch.Tensor method) Fold (class in torch.nn) fold() (in module torch.nn.functional) forced_specializations() (torch.fx.experimental.symbolic_shapes.DimConstraints method) fork() (in module torch.jit) fork_rng() (in module torch.random) format_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) format_node() (torch.fx.Node method) forward() (torch.ao.nn.quantizable.MultiheadAttention method) (torch.ao.ns._numeric_suite.Logger method) (torch.ao.ns._numeric_suite.OutputLogger method) (torch.ao.ns._numeric_suite.Shadow method) (torch.ao.ns._numeric_suite.ShadowLogger method) (torch.ao.ns._numeric_suite_fx.OutputComparisonLogger method) (torch.ao.ns._numeric_suite_fx.OutputLogger method) (torch.ao.quantization.observer.AffineQuantizedObserverBase method) (torch.ao.quantization.observer.MinMaxObserver method) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction method) (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.ConvTranspose2d method) (torch.nn.EmbeddingBag method) (torch.nn.Module method) (torch.nn.modules.normalization.RMSNorm method) (torch.nn.MultiheadAttention method) (torch.nn.RMSNorm method) (torch.nn.Transformer method) (torch.nn.TransformerDecoder method) (torch.nn.TransformerDecoderLayer method) (torch.nn.TransformerEncoder method) (torch.nn.TransformerEncoderLayer method) (torch.optim.swa_utils.AveragedModel method) forward_extended() (torch.autograd.function.NestedIOFunction method) forward_shape() (torch.distributions.transforms.Transform method) fp16_bf16_reduction_math_sdp_allowed() (in module torch.backends.cuda) fp16_compress_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) fp16_compress_wrapper() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) frac() (in module torch) (torch.Tensor method) frac_() (torch.Tensor method) fractional_max_pool2d() (in module torch.nn.functional) fractional_max_pool3d() (in module torch.nn.functional) FractionalMaxPool2d (class in torch.nn) FractionalMaxPool3d (class in torch.nn) freeze() (in module torch.jit) (torch.fx.experimental.symbolic_shapes.ShapeEnv method) freeze_bn_stats (class in torch.ao.nn.intrinsic.qat) freeze_runtime_asserts() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) frexp() (in module torch) (torch.Tensor method) from_backend() (torch.distributed.elastic.rendezvous.dynamic_rendezvous.DynamicRendezvousHandler class method) from_buffer() (torch.TypedStorage class method) (torch.UntypedStorage static method) from_dict() (torch.ao.quantization.backend_config.BackendConfig class method) (torch.ao.quantization.backend_config.BackendPatternConfig class method) (torch.ao.quantization.backend_config.DTypeConfig class method) (torch.ao.quantization.fx.custom_config.ConvertCustomConfig class method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig class method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig class method) (torch.ao.quantization.qconfig_mapping.QConfigMapping class method) from_dlpack() (in module torch) (in module torch.utils.dlpack) from_dtype() (torch.onnx.JitScalarType class method) from_file() (in module torch) (torch.TypedStorage class method) (torch.UntypedStorage static method) from_float() (torch.ao.nn.qat.Linear class method) (torch.ao.nn.quantized.Conv1d class method) (torch.ao.nn.quantized.Conv2d class method) (torch.ao.nn.quantized.Conv3d class method) (torch.ao.nn.quantized.dynamic.Linear class method) (torch.ao.nn.quantized.Embedding class method) (torch.ao.nn.quantized.EmbeddingBag class method) (torch.ao.nn.quantized.Linear class method) from_group() (torch.distributed.device_mesh.DeviceMesh static method) from_ipc_handle() (torch.cuda.Event class method) from_kv_blocks() (torch.nn.attention.flex_attention.BlockMask class method) from_local() (torch.distributed.tensor.DTensor static method) from_numpy() (in module torch) from_onnx_type() (torch.onnx.JitScalarType class method) from_pretrained() (torch.nn.Embedding class method) (torch.nn.EmbeddingBag class method) from_reference() (torch.ao.nn.quantized.dynamic.Linear class method) (torch.ao.nn.quantized.Linear class method) from_tensors() (torch.onnx.verification.VerificationInfo class method) from_value() (torch.onnx.JitScalarType class method) frombuffer() (in module torch) fromkeys() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) fsdp_modules() (torch.distributed.fsdp.FullyShardedDataParallel static method) FSDPModule (class in torch.distributed.fsdp) full() (in module torch) (in module torch.distributed.tensor) full_kv_indices (torch.nn.attention.flex_attention.BlockMask attribute) full_kv_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) full_like() (in module torch) full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) full_q_indices (torch.nn.attention.flex_attention.BlockMask attribute) full_q_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) full_tensor() (torch.distributed.tensor.DTensor method) FullOptimStateDictConfig (class in torch.distributed.fsdp) FullStateDictConfig (class in torch.distributed.fsdp) fully_shard() (in module torch.distributed.fsdp) FullyShardedDataParallel (class in torch.distributed.fsdp) Function (class in torch.autograd) functional_call() (in module torch.func) (in module torch.nn.utils.stateless) functionalize() (in module torch.func) FunctionCounts (class in torch.utils.benchmark) fuse_conv_bn_eval() (in module torch.nn.utils) fuse_conv_bn_weights() (in module torch.nn.utils) fuse_fx (class in torch.ao.quantization.quantize_fx) fuse_linear_bn_eval() (in module torch.nn.utils) fuse_linear_bn_weights() (in module torch.nn.utils) fuse_modules (class in torch.ao.quantization.fuse_modules) FuseCustomConfig (class in torch.ao.quantization.fx.custom_config) FusedMovingAvgObsFakeQuantize (class in torch.ao.quantization.fake_quantize) Future (class in torch.futures) FXFloatFunctional (class in torch.ao.nn.quantized) G Gamma (class in torch.distributions.gamma) gammainc() (in module torch.special) gammaincc() (in module torch.special) gammaln() (in module torch.special) gather() (in module torch) (in module torch.cuda.comm) (in module torch.distributed) (torch.Tensor method) gather_object() (in module torch.distributed) gaussian() (in module torch.signal.windows) gaussian_nll_loss() (in module torch.nn.functional) GaussianNLLLoss (class in torch.nn) gcd() (in module torch) (torch.Tensor method) gcd_() (torch.Tensor method) gds_deregister_buffer() (in module torch.cuda.gds) gds_register_buffer() (in module torch.cuda.gds) GdsFile (class in torch.cuda.gds) ge() (in module torch) (torch.Tensor method) ge_() (torch.Tensor method) GELU (class in torch.nn) gelu() (in module torch.nn.functional) general_cosine() (in module torch.signal.windows) general_hamming() (in module torch.signal.windows) GeneralizedPareto (class in torch.distributions.generalized_pareto) generate_comm_debug_tracing_table() (torch.distributed.tensor.debug.CommDebugMode method) generate_json_dump() (torch.distributed.tensor.debug.CommDebugMode method) generate_methods_for_privateuse1_backend() (in module torch.utils) generate_numeric_debug_handle (class in torch.ao.quantization) generate_square_subsequent_mask() (torch.nn.Transformer static method) Generator (class in torch) Geometric (class in torch.distributions.geometric) geometric_() (torch.Tensor method) geqrf() (in module torch) (torch.Tensor method) ger() (in module torch) (torch.Tensor method) get() (torch.autograd.profiler_util.StringTable method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.elastic.rendezvous.RendezvousParameters method) (torch.distributed.Store method) (torch.fx.experimental.symbolic_shapes.CallMethodKey method) (torch.fx.experimental.symbolic_shapes.ConvertIntKey method) (torch.fx.experimental.symbolic_shapes.DivideByKey method) (torch.fx.experimental.symbolic_shapes.InnerTensorKey method) (torch.monitor.Stat method) (torch.nn.ParameterDict method) get_all_groups() (torch.distributed.device_mesh.DeviceMesh method) get_all_sharing_strategies() (in module torch.multiprocessing) get_allocator_backend() (in module torch.cuda) get_arch_list() (in module torch.cuda) (in module torch.xpu) get_as_bool() (torch.distributed.elastic.rendezvous.RendezvousParameters method) get_as_int() (torch.distributed.elastic.rendezvous.RendezvousParameters method) get_attr() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) get_axioms() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_backend() (in module torch.distributed) (torch.distributed.elastic.rendezvous.RendezvousHandler method) get_block_size (class in torch.ao.quantization.observer) get_buffer() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_comm_counts() (torch.distributed.tensor.debug.CommDebugMode method) get_compiler_abi_compatibility_and_version() (in module torch.utils.cpp_extension) get_coordinate() (torch.distributed.device_mesh.DeviceMesh method) get_cpp_backtrace() (in module torch.utils) get_cpu_capability() (in module torch.backends.cpu) get_crc32_options() (in module torch.serialization) get_ctx() (in module torch.library) get_debug_state() (torch.jit.ScriptFunction method) get_default_device() (in module torch) get_default_dtype() (in module torch) get_default_load_endianness() (in module torch.serialization) get_default_mmap_options() (in module torch.serialization) get_default_qat_qconfig_mapping (class in torch.ao.quantization.qconfig_mapping) get_default_qconfig_mapping (class in torch.ao.quantization.qconfig_mapping) get_deterministic_debug_mode() (in module torch) get_device() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) get_device_capability() (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) get_device_module() (in module torch) get_device_name() (in module torch.cuda) (in module torch.xpu) get_device_properties() (in module torch.cuda) (in module torch.xpu) get_dir() (in module torch.hub) get_ema_multi_avg_fn() (in module torch.optim.swa_utils) get_entrypoint_name() (torch.distributed.elastic.agent.server.WorkerSpec method) get_expired_timers() (torch.distributed.elastic.timer.TimerServer method) get_extra_state() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_fastpath_enabled() (in module torch.backends.mha) get_filename() (in module torch.cuda.tunable) get_float32_matmul_precision() (in module torch) get_fresh_qualname() (torch.fx.Tracer method) get_future() (torch.distributed.Work method) get_future_result() (torch.distributed.Work method) get_gencode_flags() (in module torch.cuda) (in module torch.xpu) get_global_rank() (in module torch.distributed) get_gradient_edge() (in module torch.autograd.graph) get_gradients() (in module torch.distributed.autograd) get_group() (torch.distributed.device_mesh.DeviceMesh method) get_group_rank() (in module torch.distributed) get_ignored_functions() (in module torch.overrides) get_implications() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_last_lr() (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.swa_utils.SWALR method) get_local_rank() (torch.distributed.device_mesh.DeviceMesh method) get_logger_dict() (in module torch.ao.ns._numeric_suite) get_logging_handler() (in module torch.distributed.elastic.events) get_lr() (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.swa_utils.SWALR method) get_matching_activations() (in module torch.ao.ns._numeric_suite) get_max_tuning_duration() (in module torch.cuda.tunable) get_max_tuning_iterations() (in module torch.cuda.tunable) get_model_state_dict() (in module torch.distributed.checkpoint.state_dict) get_module_rref() (torch.distributed.nn.api.remote_module.RemoteModule method) get_nontrivial_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_num_interop_threads() (in module torch) get_num_threads() (in module torch) get_observer_state_dict (class in torch.ao.quantization.observer) get_opt_einsum() (in module torch.backends.opt_einsum) get_optimizer_state_dict() (in module torch.distributed.checkpoint.state_dict) get_overridable_functions() (in module torch.overrides) get_overwrite_module_params_on_conversion() (in module torch.__future__) get_parameter() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_parameter_info() (torch.distributed.tensor.debug.CommDebugMode method) get_per_process_memory_fraction() (in module torch.cuda) get_process_group_ranks() (in module torch.distributed) get_proxy_mode() (in module torch.fx.experimental.proxy_tensor) get_pruned_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_rank() (in module torch.distributed) (torch.distributed.device_mesh.DeviceMesh method) get_rdeps() (torch.package.PackageExporter method) get_replace_hook() (torch.export.graph_signature.ExportGraphSignature method) get_results() (in module torch.cuda.tunable) get_rng_state() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.random) (in module torch.xpu) get_rng_state_all() (in module torch.cuda) (in module torch.xpu) get_rotating_buffer_size() (in module torch.cuda.tunable) get_run_id() (torch.distributed.elastic.rendezvous.RendezvousHandler method) get_safe_globals() (in module torch.serialization) get_sharding_info() (torch.distributed.tensor.debug.CommDebugMode method) get_sharing_strategy() (in module torch.multiprocessing) get_state() (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend method) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend method) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend method) (torch.Generator method) get_state_dict() (in module torch.distributed.checkpoint.state_dict) get_state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) get_stream_from_external() (in module torch.cuda) (in module torch.xpu) get_submodule() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_subprocess_handler() (in module torch.distributed.elastic.multiprocessing.subprocess_handler.handlers) get_swap_module_params_on_conversion() (in module torch.__future__) get_sync_debug_mode() (in module torch.cuda) get_testing_overrides() (in module torch.overrides) get_total_counts() (torch.distributed.tensor.debug.CommDebugMode method) get_total_norm() (in module torch.nn.utils) get_trace_id() (torch.profiler.profile method) get_unique_id() (torch.package.PackageExporter method) get_unsafe_globals_in_checkpoint() (in module torch.serialization) get_validators() (in module torch.cuda.tunable) get_worker_group() (torch.distributed.elastic.agent.server.ElasticAgent method) get_worker_info() (in module torch.distributed.rpc) (in module torch.utils.data) get_world_size() (in module torch.distributed) getattr() (torch.fx.Tracer method) global_unstructured() (in module torch.nn.utils.prune) GLU (class in torch.nn) glu() (in module torch.nn.functional) grad (torch.Tensor attribute) grad() (in module torch.autograd) (in module torch.func) grad_and_value() (in module torch.func) GradBucket (class in torch.distributed) gradcheck() (in module torch.autograd.gradcheck) GradcheckError gradgradcheck() (in module torch.autograd.gradcheck) gradient() (in module torch) GradientEdge (class in torch.autograd.graph) gradients() (in module torch.distributed.GradBucket) GradScaler (class in torch.cpu.amp) (class in torch.cuda.amp) Granularity (class in torch.ao.quantization.observer) graph (class in torch.cuda) Graph (class in torch.fx) graph (torch.fx.GraphModule property) (torch.jit.ScriptModule property) graph_copy() (torch.fx.Graph method) graph_pool_handle() (in module torch.cuda) GraphInfo (class in torch.onnx.verification) GraphInfoPrettyPrinter (class in torch.onnx.verification) GraphModule (class in torch.fx) graphsafe_get_state() (torch.Generator method) graphsafe_set_state() (torch.Generator method) greater() (in module torch) (torch.Tensor method) greater_() (torch.Tensor method) greater_equal() (in module torch) (torch.Tensor method) greater_equal_() (torch.Tensor method) greater_than (in module torch.distributions.constraints) greater_than_eq (in module torch.distributions.constraints) grid_sample() (in module torch.nn.functional) group_norm() (in module torch.nn.functional) GroupNorm (class in torch.ao.nn.quantized) (class in torch.nn) GRU (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) GRUCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) gt() (in module torch) (torch.Tensor method) gt_() (torch.Tensor method) guard_or_false() (in module torch.fx.experimental.symbolic_shapes) guard_or_true() (in module torch.fx.experimental.symbolic_shapes) guard_size_oblivious() (in module torch.fx.experimental.symbolic_shapes) Gumbel (class in torch.distributions.gumbel) gumbel_softmax() (in module torch.nn.functional) H H (torch.Tensor attribute) half() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) half_open_interval (in module torch.distributions.constraints) HalfCauchy (class in torch.distributions.half_cauchy) HalfNormal (class in torch.distributions.half_normal) HalfStorage (class in torch) hamming() (in module torch.signal.windows) hamming_window() (in module torch) handle_sym_dispatch() (in module torch.fx.experimental.proxy_tensor) handle_torch_function() (in module torch.overrides) hann() (in module torch.signal.windows) hann_window() (in module torch) Hardshrink (class in torch.nn) hardshrink() (in module torch.nn.functional) (torch.Tensor method) hardsigmoid (class in torch.ao.nn.quantized.functional) Hardsigmoid (class in torch.nn) hardsigmoid() (in module torch.nn.functional) Hardswish (class in torch.ao.nn.quantized) hardswish (class in torch.ao.nn.quantized.functional) Hardswish (class in torch.nn) hardswish() (in module torch.nn.functional) hardtanh (class in torch.ao.nn.quantized.functional) Hardtanh (class in torch.nn) hardtanh() (in module torch.nn.functional) hardtanh_() (in module torch.nn.functional) has_enumerate_support (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.binomial.Binomial attribute) (torch.distributions.categorical.Categorical attribute) (torch.distributions.independent.Independent property) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) has_extended_api() (torch.distributed.Store method) has_file() (torch.package.Directory method) has_free_symbols() (in module torch.fx.experimental.symbolic_shapes) has_free_unbacked_symbols() (in module torch.fx.experimental.symbolic_shapes) has_rsample (torch.distributions.beta.Beta attribute) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily attribute) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution property) (torch.distributions.uniform.Uniform attribute) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.wishart.Wishart attribute) has_static_value() (in module torch.fx.experimental.symbolic_shapes) has_torch_function() (in module torch.overrides) has_uninitialized_params() (torch.nn.modules.lazy.LazyModuleMixin method) HashStore (class in torch.distributed) HealthCheckServer (class in torch.distributed.elastic.agent.server.health_check_server) heartbeat (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) heaviside() (in module torch) (torch.Tensor method) help() (in module torch.hub) hessian() (in module torch.autograd.functional) (in module torch.func) hex() (torch.SymFloat method) hfft() (in module torch.fft) hfft2() (in module torch.fft) hfftn() (in module torch.fft) highlight_warnings() (torch.utils.benchmark.Compare method) hinge_embedding_loss() (in module torch.nn.functional) HingeEmbeddingLoss (class in torch.nn) hint_int() (in module torch.fx.experimental.symbolic_shapes) histc() (in module torch) (torch.Tensor method) histogram() (in module torch) (torch.Tensor method) histogramdd() (in module torch) HistogramObserver (class in torch.ao.quantization.observer) host (torch.distributed.TCPStore property) host_memory_stats() (in module torch.cuda) householder_product() (in module torch.linalg) hpu() (torch.TypedStorage method) (torch.UntypedStorage method) hsplit() (in module torch) (torch.Tensor method) hspmm() (in module torch) hstack() (in module torch) huber_loss() (in module torch.nn.functional) HuberLoss (class in torch.nn) hvp() (in module torch.autograd.functional) hypot() (in module torch) (torch.Tensor method) hypot_() (torch.Tensor method) I i0() (in module torch) (in module torch.special) (torch.Tensor method) i0_() (torch.Tensor method) i0e() (in module torch.special) i1() (in module torch.special) i1e() (in module torch.special) icdf() (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.laplace.Laplace method) (torch.distributions.normal.Normal method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) id (torch.cuda.MemPool property) (torch.distributed.rpc.WorkerInfo property) id() (torch.package.PackageImporter method) Identity (class in torch.nn) (class in torch.nn.utils.prune) identity() (in module torch.nn.utils.prune) ifft() (in module torch.fft) ifft2() (in module torch.fft) ifftn() (in module torch.fft) ifftshift() (in module torch.fft) igamma() (in module torch) (torch.Tensor method) igamma_() (torch.Tensor method) igammac() (in module torch) (torch.Tensor method) igammac_() (torch.Tensor method) ignore() (in module torch.jit) ignore_fresh_unbacked_symbols() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) ihfft() (in module torch.fft) ihfft2() (in module torch.fft) ihfftn() (in module torch.fft) imag (torch.Tensor attribute) imag() (in module torch) impl() (in module torch.library) (torch.library.Library method) impl_abstract() (in module torch.library) import_module() (torch.package.PackageImporter method) in_interval() (torch.autograd.profiler_util.MemRecordsAcc method) include_paths() (in module torch.utils.cpp_extension) increment_step() (torch.autograd.profiler.KinetoStepTracker class method) increment_version() (in module torch.autograd.graph) Independent (class in torch.distributions.independent) independent (in module torch.distributions.constraints) IndependentTransform (class in torch.distributions.transforms) index() (in module torch.distributed.GradBucket) (torch.autograd.forward_ad.UnpackedDualTensor method) (torch.autograd.profiler_util.Kernel method) (torch.jit.Attribute method) (torch.nn.utils.rnn.PackedSequence method) (torch.Size method) index_add() (in module torch) (torch.Tensor method) index_add_() (torch.Tensor method) index_copy() (in module torch) (torch.Tensor method) index_copy_() (torch.Tensor method) index_fill() (torch.Tensor method) index_fill_() (torch.Tensor method) index_put() (torch.Tensor method) index_put_() (torch.Tensor method) index_reduce() (in module torch) (torch.Tensor method) index_reduce_() (torch.Tensor method) index_select() (in module torch) (torch.Tensor method) indices() (torch.Tensor method) infer_schema() (in module torch.library) inference_mode (class in torch.autograd.grad_mode) init() (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) init_device_mesh() (in module torch.distributed.device_mesh) init_method (torch.distributed.rpc.RpcBackendOptions property) (torch.distributed.rpc.TensorPipeRpcBackendOptions property) init_process_group() (in module torch.distributed) init_rpc() (in module torch.distributed.rpc) init_step_count() (torch.autograd.profiler.KinetoStepTracker class method) initial_seed() (in module torch) (in module torch.cuda) (in module torch.random) (in module torch.xpu) (torch.Generator method) initialize_inference_session() (torch.onnx.ONNXProgram method) initialize_parameters() (torch.nn.modules.lazy.LazyModuleMixin method) inlined_graph (torch.jit.ScriptModule property) inner() (in module torch) (torch.Tensor method) InnerTensorKey (class in torch.fx.experimental.symbolic_shapes) InplaceFunction (class in torch.autograd.function) INPUT_OUTPUT_NOT_OBSERVED (torch.ao.quantization.backend_config.ObservationType attribute) InputKind (class in torch.export.graph_signature) InputSpec (class in torch.export.graph_signature) insert() (torch.nn.ModuleList method) (torch.nn.Sequential method) insert_arg() (torch.fx.Node method) inserting_after() (torch.fx.Graph method) inserting_before() (torch.fx.Graph method) instance_norm() (in module torch.nn.functional) InstanceNorm1d (class in torch.ao.nn.quantized) (class in torch.nn) InstanceNorm2d (class in torch.ao.nn.quantized) (class in torch.nn) InstanceNorm3d (class in torch.ao.nn.quantized) (class in torch.nn) int() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) int_repr() (torch.Tensor method) integer_interval (in module torch.distributions.constraints) interface() (in module torch.jit) intern() (torch.package.PackageExporter method) interned_modules() (torch.package.PackageExporter method) interpolate (class in torch.ao.nn.quantized.functional) interpolate() (in module torch.nn.functional) Interpreter (class in torch.fx) InterpreterModule (class in torch.export.unflatten) InterpreterModuleDispatcher (class in torch.export.unflatten) Interval (class in torch.autograd.profiler_util) interval (in module torch.distributions.constraints) IntStorage (class in torch) inv (torch.distributions.transforms.Transform property) inv() (in module torch.linalg) inv_ex() (in module torch.linalg) inverse() (in module torch) (torch.Tensor method) inverse_shape() (torch.distributions.transforms.Transform method) InverseGamma (class in torch.distributions.inverse_gamma) ipc_collect() (in module torch.cuda) ipc_handle() (torch.cuda.Event method) ipu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) irecv() (in module torch.distributed) irfft() (in module torch.fft) irfft2() (in module torch.fft) irfftn() (in module torch.fft) is_accessor_node() (in module torch.fx.experimental.symbolic_shapes) is_autocast_available() (in module torch.amp.autocast_mode) is_available() (in module torch.accelerator) (in module torch.backends.cudnn) (in module torch.backends.cusparselt) (in module torch.backends.mkl) (in module torch.backends.mkldnn) (in module torch.backends.mps) (in module torch.backends.nnpack) (in module torch.backends.openmp) (in module torch.backends.opt_einsum) (in module torch.cpu) (in module torch.cuda) (in module torch.distributed) (in module torch.mtia) (in module torch.profiler.itt) (in module torch.xpu) is_built() (in module torch.backends.cuda) (in module torch.backends.mps) is_capturing_metal() (in module torch.mps.profiler) is_closed() (torch.distributed.elastic.rendezvous.RendezvousHandler method) is_coalesced() (torch.Tensor method) is_compiling() (in module torch.compiler) is_completed() (torch.distributed.Work method) is_complex() (in module torch) (torch.Tensor method) is_concrete_bool() (in module torch.fx.experimental.symbolic_shapes) is_concrete_float() (in module torch.fx.experimental.symbolic_shapes) is_concrete_int() (in module torch.fx.experimental.symbolic_shapes) is_conj() (in module torch) (torch.Tensor method) is_contiguous() (torch.Tensor method) is_cuda (torch.nn.utils.rnn.PackedSequence property) (torch.Tensor attribute) (torch.TypedStorage property) (torch.UntypedStorage property) is_current_stream_capturing() (in module torch.cuda) is_dependent() (in module torch.distributions.constraints) is_deterministic_algorithms_warn_only_enabled() (in module torch) is_dynamo_compiling() (in module torch.compiler) is_enabled() (in module torch.cuda.tunable) (torch.sparse.check_sparse_tensor_invariants static method) is_exporting() (in module torch.compiler) is_flash_attention_available() (in module torch.backends.cuda) is_floating_point() (in module torch) (torch.Tensor method) is_gloo_available() (in module torch.distributed) is_grad_enabled() (in module torch) is_hpu (torch.TypedStorage property) (torch.UntypedStorage property) is_impure() (torch.fx.Node method) is_in_onnx_export() (in module torch.onnx) is_inference() (torch.Tensor method) is_inference_mode_enabled() (in module torch) is_initialized() (in module torch.cuda) (in module torch.distributed) (in module torch.mtia) (in module torch.xpu) is_integer() (torch.SymFloat method) is_last() (in module torch.distributed.GradBucket) is_leaf (torch.Tensor attribute) is_leaf_module() (torch.ao.ns._numeric_suite_fx.NSTracer method) (torch.fx.Tracer method) is_meta (torch.Tensor attribute) is_metal_capture_enabled() (in module torch.mps.profiler) is_mpi_available() (in module torch.distributed) is_nccl_available() (in module torch.distributed) is_ninja_available() (in module torch.utils.cpp_extension) is_nonzero() (in module torch) is_onnxrt_backend_supported() (in module torch.onnx) is_owner() (torch.distributed.rpc.PyRRef method) is_parametrized() (in module torch.nn.utils.parametrize) is_partial() (torch.distributed.tensor.placement_types.Placement method) is_pinned() (torch.nn.utils.rnn.PackedSequence method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) is_pruned() (in module torch.nn.utils.prune) is_quantized (torch.Tensor attribute) is_replicate() (torch.distributed.tensor.placement_types.Placement method) is_running() (torch.distributed.elastic.agent.server.WorkerState static method) is_scripting() (in module torch.jit) is_set_to() (torch.Tensor method) is_shard() (torch.distributed.tensor.placement_types.Placement method) is_shared() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) is_signed() (torch.Tensor method) is_sparse (torch.Tensor attribute) (torch.TypedStorage attribute) (torch.UntypedStorage attribute) is_sparse_csr (torch.Tensor attribute) (torch.UntypedStorage attribute) is_storage() (in module torch) is_success() (torch.distributed.Work method) is_tensor() (in module torch) is_tensor_like() (in module torch.overrides) is_tensor_method_or_property() (in module torch.overrides) is_tf32_supported() (in module torch.cuda) is_torchelastic_launched() (in module torch.distributed) is_tracing() (in module torch.jit) is_unbacked_symint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) is_warn_always_enabled() (in module torch) is_xccl_available() (in module torch.distributed.distributed_c10d) isclose() (in module torch) (torch.Tensor method) isend() (in module torch.distributed) isfinite() (in module torch) (torch.Tensor method) isin() (in module torch) isinf() (in module torch) (torch.Tensor method) isinstance() (in module torch.jit) isnan() (in module torch) (torch.Tensor method) isneginf() (in module torch) (torch.Tensor method) isposinf() (in module torch) (torch.Tensor method) isreal() (in module torch) (torch.Tensor method) istft() (in module torch) (torch.Tensor method) item() (torch.Tensor method) items() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) itemsize (torch.Tensor attribute) iter() (torch.fx.Tracer method) IterableDataset (class in torch.utils.data) J jacfwd() (in module torch.func) jacobian() (in module torch.autograd.functional) jacrev() (in module torch.func) JitScalarType (class in torch.onnx) job_id (in module torch.compiler.config) Join (class in torch.distributed.algorithms) join (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) join() (torch.multiprocessing.SpawnContext method) (torch.nn.parallel.DistributedDataParallel method) join_device (torch.distributed.algorithms.Joinable property) (torch.distributed.optim.ZeroRedundancyOptimizer property) join_hook() (torch.distributed.algorithms.Joinable method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.nn.parallel.DistributedDataParallel method) join_process_group (torch.distributed.algorithms.Joinable property) (torch.distributed.optim.ZeroRedundancyOptimizer property) Joinable (class in torch.distributed.algorithms) JoinHook (class in torch.distributed.algorithms) jvp() (in module torch.autograd.functional) (in module torch.func) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) K kaiming_normal_() (in module torch.nn.init) kaiming_uniform_() (in module torch.nn.init) kaiser() (in module torch.signal.windows) kaiser_window() (in module torch) Kernel (class in torch.autograd.profiler_util) key_averages() (torch.autograd.profiler.profile method) (torch.profiler._KinetoProfile method) keys() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.fx.Tracer method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) KinetoStepTracker (class in torch.autograd.profiler) kl_div() (in module torch.nn.functional) kl_divergence() (in module torch.distributions.kl) KLDivLoss (class in torch.nn) kron() (in module torch) kthvalue() (in module torch) (torch.Tensor method) Kumaraswamy (class in torch.distributions.kumaraswamy) kv_indices (torch.nn.attention.flex_attention.BlockMask attribute) kv_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) kwargs (torch.fx.Node property) L l1_loss() (in module torch.nn.functional) l1_unstructured() (in module torch.nn.utils.prune) L1Loss (class in torch.nn) L1Unstructured (class in torch.nn.utils.prune) LambdaLR (class in torch.optim.lr_scheduler) Laplace (class in torch.distributions.laplace) last_call (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) layer_norm() (in module torch.nn.functional) LayerNorm (class in torch.ao.nn.quantized) (class in torch.nn) layout (class in torch) LazyBatchNorm1d (class in torch.nn) LazyBatchNorm2d (class in torch.nn) LazyBatchNorm3d (class in torch.nn) LazyConv1d (class in torch.nn) LazyConv2d (class in torch.nn) LazyConv3d (class in torch.nn) LazyConvTranspose1d (class in torch.nn) LazyConvTranspose2d (class in torch.nn) LazyConvTranspose3d (class in torch.nn) LazyInstanceNorm1d (class in torch.nn) LazyInstanceNorm2d (class in torch.nn) LazyInstanceNorm3d (class in torch.nn) LazyLinear (class in torch.nn) LazyModuleMixin (class in torch.nn.modules.lazy) LBFGS (class in torch.optim) lcm() (in module torch) (torch.Tensor method) lcm_() (torch.Tensor method) ldexp() (in module torch) (torch.Tensor method) ldexp_() (torch.Tensor method) ldl_factor() (in module torch.linalg) ldl_factor_ex() (in module torch.linalg) ldl_solve() (in module torch.linalg) le() (in module torch) (torch.Tensor method) le_() (torch.Tensor method) leaky_relu (class in torch.ao.nn.quantized.functional) leaky_relu() (in module torch.nn.functional) leaky_relu_() (in module torch.nn.functional) LeakyReLU (class in torch.ao.nn.quantized) (class in torch.nn) lerp() (in module torch) (torch.Tensor method) lerp_() (torch.Tensor method) less() (in module torch) (torch.Tensor method) less_() (torch.Tensor method) less_equal() (in module torch) (torch.Tensor method) less_equal_() (torch.Tensor method) less_than (in module torch.distributions.constraints) lgamma() (in module torch) (torch.Tensor method) lgamma_() (torch.Tensor method) Library (class in torch.library) libuvBackend (torch.distributed.TCPStore property) Linear (class in torch.ao.nn.qat) (class in torch.ao.nn.qat.dynamic) (class in torch.ao.nn.quantized) (class in torch.ao.nn.quantized.dynamic) linear (class in torch.ao.nn.quantized.functional) Linear (class in torch.nn) linear() (in module torch.nn.functional) linearize() (in module torch.func) LinearLR (class in torch.optim.lr_scheduler) LinearReLU (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) (class in torch.ao.nn.intrinsic.quantized.dynamic) linspace() (in module torch) lint() (torch.fx.Graph method) list() (in module torch.hub) list_backends() (in module torch.compiler) list_gpu_processes() (in module torch.cuda) LKJCholesky (class in torch.distributions.lkj_cholesky) ln_structured() (in module torch.nn.utils.prune) LnStructured (class in torch.nn.utils.prune) load() (in module torch) (in module torch.distributed.checkpoint.state_dict_loader) (in module torch.export) (in module torch.hub) (in module torch.jit) (in module torch.utils.cpp_extension) load_binary() (torch.package.PackageImporter method) load_bytes() (torch.distributed.checkpoint.LoadPlanner method) load_inline() (in module torch.utils.cpp_extension) load_nvprof() (in module torch.autograd.profiler) load_observer_state_dict (class in torch.ao.quantization.observer) load_pickle() (torch.package.PackageImporter method) load_state_dict() (in module torch.distributed.checkpoint.state_dict_loader) (torch.distributed.checkpoint.stateful.Stateful method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) (torch.optim.swa_utils.SWALR method) load_state_dict_from_url() (in module torch.hub) load_storage() (torch.cuda.gds.GdsFile method) load_text() (torch.package.PackageImporter method) load_url() (in module torch.utils.model_zoo) LoadPlan (class in torch.distributed.checkpoint) LoadPlanner (class in torch.distributed.checkpoint) lobpcg() (in module torch) loc (torch.distributions.log_normal.LogNormal property) local_map() (in module torch.distributed.tensor.experimental) local_response_norm() (in module torch.nn.functional) local_value() (torch.distributed.rpc.PyRRef method) LocalElasticAgent (class in torch.distributed.elastic.agent.server.local_elastic_agent) LocalOptimStateDictConfig (class in torch.distributed.fsdp) LocalResponseNorm (class in torch.nn) LocalStateDictConfig (class in torch.distributed.fsdp) LocalTimerClient (class in torch.distributed.elastic.timer) LocalTimerServer (class in torch.distributed.elastic.timer) log() (in module torch) (torch.Tensor method) log10() (in module torch) (torch.Tensor method) log10_() (torch.Tensor method) log1p() (in module torch) (in module torch.special) (torch.Tensor method) log1p_() (torch.Tensor method) log2() (in module torch) (torch.Tensor method) log2_() (torch.Tensor method) log_() (torch.Tensor method) log_abs_det_jacobian() (torch.distributions.transforms.Transform method) log_cdf() (torch.distributions.generalized_pareto.GeneralizedPareto method) log_comm_debug_tracing_table_to_file() (torch.distributed.tensor.debug.CommDebugMode method) log_debug_info_for_expired_timers() (in module torch.distributed.elastic.timer.debug_info_logging) log_event() (in module torch.monitor) log_ndtr() (in module torch.special) log_normal_() (torch.Tensor method) log_prob() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.laplace.Laplace method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.poisson.Poisson method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.von_mises.VonMises method) (torch.distributions.wishart.Wishart method) (torch.nn.AdaptiveLogSoftmaxWithLoss method) log_softmax() (in module torch.nn.functional) (in module torch.sparse) (in module torch.special) log_survival_function() (torch.distributions.generalized_pareto.GeneralizedPareto method) logaddexp() (in module torch) (torch.Tensor method) logaddexp2() (in module torch) (torch.Tensor method) logcumsumexp() (in module torch) (torch.Tensor method) logdet() (in module torch) (torch.Tensor method) Logger (class in torch.ao.ns._numeric_suite) loggers_set_enabled() (in module torch.ao.ns._numeric_suite_fx) loggers_set_save_activations() (in module torch.ao.ns._numeric_suite_fx) logical_and() (in module torch) (torch.Tensor method) logical_and_() (torch.Tensor method) logical_not() (in module torch) (torch.Tensor method) logical_not_() (torch.Tensor method) logical_or() (in module torch) (torch.Tensor method) logical_or_() (torch.Tensor method) logical_xor() (in module torch) (torch.Tensor method) logical_xor_() (torch.Tensor method) logit() (in module torch) (in module torch.special) (torch.Tensor method) logit_() (torch.Tensor method) LogitRelaxedBernoulli (class in torch.distributions.relaxed_bernoulli) logits (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.geometric.Geometric property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) LogNormal (class in torch.distributions.log_normal) LogsDest (class in torch.distributed.elastic.multiprocessing.api) LogSigmoid (class in torch.nn) logsigmoid() (in module torch.nn.functional) LogSoftmax (class in torch.nn) logspace() (in module torch) LogsSpecs (class in torch.distributed.elastic.multiprocessing.api) logsumexp() (in module torch) (in module torch.special) (torch.Tensor method) long() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) LongStorage (class in torch) lookup_object() (torch.distributed.checkpoint.DefaultSavePlanner method) lookup_tensor() (torch.distributed.checkpoint.DefaultLoadPlanner method) loss_parallel() (in module torch.distributed.tensor.parallel) lower_pt2e_quantized_to_x86 (class in torch.ao.quantization.pt2e.lowering) LowerCholeskyTransform (class in torch.distributions.transforms) LowRankMultivariateNormal (class in torch.distributions.lowrank_multivariate_normal) lp_pool1d() (in module torch.nn.functional) lp_pool2d() (in module torch.nn.functional) lp_pool3d() (in module torch.nn.functional) LPPool1d (class in torch.nn) LPPool2d (class in torch.nn) LPPool3d (class in torch.nn) LRScheduler (class in torch.optim.lr_scheduler) lru_cache() (in module torch.fx.experimental.symbolic_shapes) LSTM (class in torch.ao.nn.quantizable) (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) LSTMCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) lstsq() (in module torch.linalg) lt() (in module torch) (torch.Tensor method) lt_() (torch.Tensor method) lu() (in module torch) (in module torch.linalg) (torch.Tensor method) lu_factor() (in module torch.linalg) lu_factor_ex() (in module torch.linalg) lu_solve() (in module torch) (in module torch.linalg) (torch.Tensor method) lu_unpack() (in module torch) M main_hook() (torch.distributed.algorithms.JoinHook method) make_dual() (in module torch.autograd.forward_ad) make_fx() (in module torch.fx.experimental.proxy_tensor) make_graphed_callables() (in module torch.cuda) make_tensor() (in module torch.testing) manual_seed() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.random) (in module torch.xpu) (torch.Generator method) manual_seed_all() (in module torch.cuda) (in module torch.xpu) map_() (torch.Tensor method) map_nodes_to_values() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) MappingType (class in torch.ao.quantization.observer) margin_ranking_loss() (in module torch.nn.functional) MarginRankingLoss (class in torch.nn) mark() (in module torch.cuda.nvtx) (in module torch.profiler.itt) mark_dirty() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) mark_non_differentiable() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) mask_mod (torch.nn.attention.flex_attention.BlockMask attribute) masked_fill() (torch.Tensor method) masked_fill_() (torch.Tensor method) masked_scatter() (torch.Tensor method) masked_scatter_() (torch.Tensor method) masked_select() (in module torch) (in module torch.nested) (torch.Tensor method) materialize() (torch.export.decomp_utils.CustomDecompTable method) math_sdp_enabled() (in module torch.backends.cuda) matmul() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_exp() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_norm() (in module torch.linalg) matrix_power() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_rank() (in module torch.linalg) max() (in module torch) (torch.Tensor method) max_memory_allocated() (in module torch.cuda) (in module torch.xpu) max_memory_cached() (in module torch.cuda) max_memory_reserved() (in module torch.cuda) (in module torch.xpu) max_pool1d (class in torch.ao.nn.quantized.functional) max_pool1d() (in module torch.nn.functional) max_pool2d (class in torch.ao.nn.quantized.functional) max_pool2d() (in module torch.nn.functional) max_pool3d() (in module torch.nn.functional) max_size (in module torch.backends.cuda.cufft_plan_cache) max_unpool1d() (in module torch.nn.functional) max_unpool2d() (in module torch.nn.functional) max_unpool3d() (in module torch.nn.functional) maximum() (in module torch) (torch.Tensor method) MaxPool1d (class in torch.nn) MaxPool2d (class in torch.nn) MaxPool3d (class in torch.nn) MaxUnpool1d (class in torch.nn) MaxUnpool2d (class in torch.nn) MaxUnpool3d (class in torch.nn) maybe_disable_thunkify() (in module torch.fx.experimental.proxy_tensor) maybe_enable_thunkify() (in module torch.fx.experimental.proxy_tensor) mean (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) mean() (in module torch) (torch.Tensor method) Measurement (class in torch.utils.benchmark) median() (in module torch) (torch.Tensor method) mem_efficient_sdp_enabled() (in module torch.backends.cuda) mem_get_info() (in module torch.cuda) (in module torch.xpu) memory_allocated() (in module torch.cuda) (in module torch.xpu) memory_cached() (in module torch.cuda) memory_format (class in torch) memory_reserved() (in module torch.cuda) (in module torch.xpu) memory_snapshot() (in module torch.cuda) memory_stats() (in module torch.cuda) (in module torch.mtia) (in module torch.mtia.memory) (in module torch.xpu) memory_stats_as_nested_dict() (in module torch.xpu) memory_summary() (in module torch.cuda) memory_usage() (in module torch.cuda) MemPool (class in torch.cuda) MemPoolContext (class in torch.cuda) MemRecordsAcc (class in torch.autograd.profiler_util) merge() (torch.utils.benchmark.Measurement static method) merge_chunks() (in module torch.distributed.pipelining.microbatch) merge_masks() (torch.nn.MultiheadAttention method) meshgrid() (in module torch) metadata() (torch.autograd.graph.Node method) metal_capture() (in module torch.mps.profiler) MetricHandler (class in torch.distributed.elastic.metrics.api) mgpu_tune_gemm_in_file() (in module torch.cuda.tunable) mH (torch.Tensor attribute) min() (in module torch) (torch.Tensor method) minimum() (in module torch) (torch.Tensor method) MinMaxObserver (class in torch.ao.quantization.observer) Mish (class in torch.nn) mish() (in module torch.nn.functional) MixedPrecision (class in torch.distributed.fsdp) MixedPrecisionPolicy (class in torch.distributed.fsdp) mixture_distribution (torch.distributions.mixture_same_family.MixtureSameFamily property) MixtureSameFamily (class in torch.distributions.mixture_same_family) mm() (in module torch) (in module torch.sparse) (torch.Tensor method) mock() (torch.package.PackageExporter method) mocked_modules() (torch.package.PackageExporter method) mode (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) mode() (in module torch) (torch.Tensor method) model_is_exported (class in torch.ao.quantization.pt2e.export_utils) model_proto (torch.onnx.ONNXProgram property) module torch torch.__config__ torch.__future__ torch._logging torch.accelerator torch.amp torch.amp.autocast_mode torch.amp.grad_scaler torch.ao torch.ao.nn torch.ao.nn.intrinsic torch.ao.nn.intrinsic.modules torch.ao.nn.intrinsic.modules.fused torch.ao.nn.intrinsic.qat torch.ao.nn.intrinsic.qat.modules torch.ao.nn.intrinsic.qat.modules.conv_fused torch.ao.nn.intrinsic.qat.modules.linear_fused torch.ao.nn.intrinsic.qat.modules.linear_relu torch.ao.nn.intrinsic.quantized torch.ao.nn.intrinsic.quantized.dynamic torch.ao.nn.intrinsic.quantized.dynamic.modules torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu torch.ao.nn.intrinsic.quantized.modules torch.ao.nn.intrinsic.quantized.modules.bn_relu torch.ao.nn.intrinsic.quantized.modules.conv_add torch.ao.nn.intrinsic.quantized.modules.conv_relu torch.ao.nn.intrinsic.quantized.modules.linear_relu torch.ao.nn.qat torch.ao.nn.qat.dynamic torch.ao.nn.qat.dynamic.modules torch.ao.nn.qat.dynamic.modules.linear torch.ao.nn.qat.modules torch.ao.nn.qat.modules.conv torch.ao.nn.qat.modules.embedding_ops torch.ao.nn.qat.modules.linear torch.ao.nn.quantizable torch.ao.nn.quantizable.modules torch.ao.nn.quantizable.modules.activation torch.ao.nn.quantizable.modules.rnn torch.ao.nn.quantized torch.ao.nn.quantized.dynamic torch.ao.nn.quantized.dynamic.modules torch.ao.nn.quantized.dynamic.modules.conv torch.ao.nn.quantized.dynamic.modules.linear torch.ao.nn.quantized.dynamic.modules.rnn torch.ao.nn.quantized.functional torch.ao.nn.quantized.modules torch.ao.nn.quantized.modules.activation torch.ao.nn.quantized.modules.batchnorm torch.ao.nn.quantized.modules.conv torch.ao.nn.quantized.modules.dropout torch.ao.nn.quantized.modules.embedding_ops torch.ao.nn.quantized.modules.functional_modules torch.ao.nn.quantized.modules.linear torch.ao.nn.quantized.modules.normalization torch.ao.nn.quantized.modules.rnn torch.ao.nn.quantized.modules.utils torch.ao.nn.quantized.reference torch.ao.nn.quantized.reference.modules torch.ao.nn.quantized.reference.modules.conv torch.ao.nn.quantized.reference.modules.linear torch.ao.nn.quantized.reference.modules.rnn torch.ao.nn.quantized.reference.modules.sparse torch.ao.nn.quantized.reference.modules.utils torch.ao.nn.sparse torch.ao.nn.sparse.quantized torch.ao.nn.sparse.quantized.dynamic torch.ao.nn.sparse.quantized.dynamic.linear torch.ao.nn.sparse.quantized.linear torch.ao.nn.sparse.quantized.utils torch.ao.ns torch.ao.ns._numeric_suite torch.ao.ns._numeric_suite_fx torch.ao.ns.fx torch.ao.ns.fx.graph_matcher torch.ao.ns.fx.graph_passes torch.ao.ns.fx.mappings torch.ao.ns.fx.n_shadows_utils torch.ao.ns.fx.ns_types torch.ao.ns.fx.pattern_utils torch.ao.ns.fx.qconfig_multi_mapping torch.ao.ns.fx.utils torch.ao.ns.fx.weight_utils torch.ao.pruning torch.ao.pruning.scheduler torch.ao.pruning.scheduler.base_scheduler torch.ao.pruning.scheduler.cubic_scheduler torch.ao.pruning.scheduler.lambda_scheduler torch.ao.pruning.sparsifier torch.ao.pruning.sparsifier.base_sparsifier torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier torch.ao.pruning.sparsifier.utils torch.ao.pruning.sparsifier.weight_norm_sparsifier torch.ao.quantization torch.ao.quantization.backend_config torch.ao.quantization.backend_config.backend_config torch.ao.quantization.backend_config.executorch torch.ao.quantization.backend_config.fbgemm torch.ao.quantization.backend_config.native torch.ao.quantization.backend_config.observation_type torch.ao.quantization.backend_config.onednn torch.ao.quantization.backend_config.qnnpack torch.ao.quantization.backend_config.tensorrt torch.ao.quantization.backend_config.utils torch.ao.quantization.backend_config.x86 torch.ao.quantization.fake_quantize torch.ao.quantization.fuse_modules torch.ao.quantization.fuser_method_mappings torch.ao.quantization.fx torch.ao.quantization.fx.convert torch.ao.quantization.fx.custom_config torch.ao.quantization.fx.fuse torch.ao.quantization.fx.fuse_handler torch.ao.quantization.fx.graph_module torch.ao.quantization.fx.lower_to_fbgemm torch.ao.quantization.fx.lower_to_qnnpack torch.ao.quantization.fx.lstm_utils torch.ao.quantization.fx.match_utils torch.ao.quantization.fx.pattern_utils torch.ao.quantization.fx.prepare torch.ao.quantization.fx.qconfig_mapping_utils torch.ao.quantization.fx.quantize_handler torch.ao.quantization.fx.tracer torch.ao.quantization.fx.utils torch.ao.quantization.observer torch.ao.quantization.pt2e torch.ao.quantization.pt2e.duplicate_dq_pass torch.ao.quantization.pt2e.export_utils torch.ao.quantization.pt2e.graph_utils torch.ao.quantization.pt2e.lowering torch.ao.quantization.pt2e.port_metadata_pass torch.ao.quantization.pt2e.prepare torch.ao.quantization.pt2e.qat_utils torch.ao.quantization.pt2e.representation torch.ao.quantization.pt2e.representation.rewrite torch.ao.quantization.pt2e.utils torch.ao.quantization.qconfig torch.ao.quantization.qconfig_mapping torch.ao.quantization.quant_type torch.ao.quantization.quantization_mappings torch.ao.quantization.quantize_fx torch.ao.quantization.quantize_jit torch.ao.quantization.quantize_pt2e torch.ao.quantization.quantizer torch.ao.quantization.quantizer.composable_quantizer torch.ao.quantization.quantizer.embedding_quantizer torch.ao.quantization.quantizer.quantizer torch.ao.quantization.quantizer.utils torch.ao.quantization.quantizer.x86_inductor_quantizer torch.ao.quantization.quantizer.xnnpack_quantizer torch.ao.quantization.quantizer.xnnpack_quantizer_utils torch.ao.quantization.quantizer.xpu_inductor_quantizer torch.ao.quantization.stubs torch.ao.quantization.utils torch.autograd torch.autograd.anomaly_mode torch.autograd.forward_ad torch.autograd.function torch.autograd.functional torch.autograd.grad_mode torch.autograd.gradcheck torch.autograd.graph torch.autograd.profiler torch.autograd.profiler_legacy torch.autograd.profiler_util torch.autograd.variable torch.backends torch.backends.cpu torch.backends.cuda torch.backends.cudnn torch.backends.cudnn.rnn torch.backends.cusparselt torch.backends.kleidiai torch.backends.mha torch.backends.mkl torch.backends.mkldnn torch.backends.mps torch.backends.nnpack torch.backends.openmp torch.backends.opt_einsum torch.backends.quantized torch.backends.xeon torch.backends.xeon.run_cpu torch.backends.xnnpack torch.compiler torch.compiler.config torch.contrib torch.cpu torch.cpu.amp torch.cpu.amp.autocast_mode torch.cpu.amp.grad_scaler torch.cuda torch.cuda._sanitizer torch.cuda.amp torch.cuda.amp.autocast_mode torch.cuda.amp.common torch.cuda.amp.grad_scaler torch.cuda.comm torch.cuda.error torch.cuda.gds torch.cuda.graphs torch.cuda.jiterator torch.cuda.memory torch.cuda.nccl torch.cuda.nvtx torch.cuda.profiler torch.cuda.random torch.cuda.sparse torch.cuda.streams torch.cuda.tunable torch.distributed torch.distributed.algorithms torch.distributed.algorithms.ddp_comm_hooks torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks torch.distributed.algorithms.ddp_comm_hooks.default_hooks torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks torch.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks torch.distributed.algorithms.join torch.distributed.algorithms.model_averaging torch.distributed.algorithms.model_averaging.averagers torch.distributed.algorithms.model_averaging.hierarchical_model_averager torch.distributed.algorithms.model_averaging.utils torch.distributed.argparse_util torch.distributed.autograd torch.distributed.c10d_logger torch.distributed.checkpoint torch.distributed.checkpoint.api torch.distributed.checkpoint.default_planner torch.distributed.checkpoint.filesystem torch.distributed.checkpoint.format_utils torch.distributed.checkpoint.logger torch.distributed.checkpoint.logging_handlers torch.distributed.checkpoint.metadata torch.distributed.checkpoint.optimizer torch.distributed.checkpoint.planner torch.distributed.checkpoint.planner_helpers torch.distributed.checkpoint.resharding torch.distributed.checkpoint.staging torch.distributed.checkpoint.state_dict torch.distributed.checkpoint.state_dict_loader torch.distributed.checkpoint.state_dict_saver torch.distributed.checkpoint.stateful torch.distributed.checkpoint.storage torch.distributed.checkpoint.utils torch.distributed.collective_utils torch.distributed.constants torch.distributed.device_mesh torch.distributed.distributed_c10d torch.distributed.elastic torch.distributed.elastic.agent torch.distributed.elastic.agent.server torch.distributed.elastic.agent.server.api torch.distributed.elastic.agent.server.health_check_server torch.distributed.elastic.agent.server.local_elastic_agent torch.distributed.elastic.control_plane torch.distributed.elastic.events torch.distributed.elastic.events.api torch.distributed.elastic.events.handlers torch.distributed.elastic.metrics torch.distributed.elastic.metrics.api torch.distributed.elastic.multiprocessing torch.distributed.elastic.multiprocessing.api torch.distributed.elastic.multiprocessing.errors torch.distributed.elastic.multiprocessing.errors.error_handler torch.distributed.elastic.multiprocessing.errors.handlers torch.distributed.elastic.multiprocessing.redirects torch.distributed.elastic.multiprocessing.subprocess_handler torch.distributed.elastic.multiprocessing.subprocess_handler.handlers torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler torch.distributed.elastic.multiprocessing.tail_log torch.distributed.elastic.rendezvous torch.distributed.elastic.rendezvous.api torch.distributed.elastic.rendezvous.c10d_rendezvous_backend torch.distributed.elastic.rendezvous.dynamic_rendezvous torch.distributed.elastic.rendezvous.etcd_rendezvous torch.distributed.elastic.rendezvous.etcd_rendezvous_backend torch.distributed.elastic.rendezvous.etcd_server torch.distributed.elastic.rendezvous.etcd_store torch.distributed.elastic.rendezvous.registry torch.distributed.elastic.rendezvous.static_tcp_rendezvous torch.distributed.elastic.rendezvous.utils torch.distributed.elastic.timer torch.distributed.elastic.timer.api torch.distributed.elastic.timer.debug_info_logging torch.distributed.elastic.timer.file_based_local_timer torch.distributed.elastic.timer.local_timer torch.distributed.elastic.utils torch.distributed.elastic.utils.api torch.distributed.elastic.utils.data torch.distributed.elastic.utils.data.cycling_iterator torch.distributed.elastic.utils.data.elastic_distributed_sampler torch.distributed.elastic.utils.distributed torch.distributed.elastic.utils.log_level torch.distributed.elastic.utils.logging torch.distributed.elastic.utils.store torch.distributed.fsdp torch.distributed.fsdp.api torch.distributed.fsdp.fully_sharded_data_parallel torch.distributed.fsdp.sharded_grad_scaler torch.distributed.fsdp.wrap torch.distributed.launch torch.distributed.launcher torch.distributed.launcher.api torch.distributed.logging_handlers torch.distributed.nn torch.distributed.nn.api torch.distributed.nn.api.remote_module torch.distributed.nn.functional torch.distributed.nn.jit torch.distributed.nn.jit.instantiator torch.distributed.nn.jit.templates torch.distributed.nn.jit.templates.remote_module_template torch.distributed.optim torch.distributed.optim.apply_optimizer_in_backward torch.distributed.optim.functional_adadelta torch.distributed.optim.functional_adagrad torch.distributed.optim.functional_adam torch.distributed.optim.functional_adamax torch.distributed.optim.functional_adamw torch.distributed.optim.functional_rmsprop torch.distributed.optim.functional_rprop torch.distributed.optim.functional_sgd torch.distributed.optim.named_optimizer torch.distributed.optim.optimizer torch.distributed.optim.post_localSGD_optimizer torch.distributed.optim.utils torch.distributed.optim.zero_redundancy_optimizer torch.distributed.pipelining torch.distributed.pipelining.microbatch torch.distributed.pipelining.schedules torch.distributed.pipelining.stage torch.distributed.remote_device torch.distributed.rendezvous torch.distributed.rpc torch.distributed.rpc.api torch.distributed.rpc.backend_registry torch.distributed.rpc.constants torch.distributed.rpc.functions torch.distributed.rpc.internal torch.distributed.rpc.options torch.distributed.rpc.rref_proxy torch.distributed.rpc.server_process_global_profiler torch.distributed.run torch.distributed.tensor torch.distributed.tensor.debug torch.distributed.tensor.device_mesh torch.distributed.tensor.experimental torch.distributed.tensor.parallel torch.distributed.tensor.parallel.api torch.distributed.tensor.parallel.ddp torch.distributed.tensor.parallel.fsdp torch.distributed.tensor.parallel.input_reshard torch.distributed.tensor.parallel.loss torch.distributed.tensor.parallel.style torch.distributed.tensor.placement_types torch.distributed.utils torch.distributions torch.distributions.bernoulli torch.distributions.beta torch.distributions.binomial torch.distributions.categorical torch.distributions.cauchy torch.distributions.chi2 torch.distributions.constraint_registry torch.distributions.constraints torch.distributions.continuous_bernoulli torch.distributions.dirichlet torch.distributions.distribution torch.distributions.exp_family torch.distributions.exponential torch.distributions.fishersnedecor torch.distributions.gamma torch.distributions.generalized_pareto torch.distributions.geometric torch.distributions.gumbel torch.distributions.half_cauchy torch.distributions.half_normal torch.distributions.independent torch.distributions.inverse_gamma torch.distributions.kl torch.distributions.kumaraswamy torch.distributions.laplace torch.distributions.lkj_cholesky torch.distributions.log_normal torch.distributions.logistic_normal torch.distributions.lowrank_multivariate_normal torch.distributions.mixture_same_family torch.distributions.multinomial torch.distributions.multivariate_normal torch.distributions.negative_binomial torch.distributions.normal torch.distributions.one_hot_categorical torch.distributions.pareto torch.distributions.poisson torch.distributions.relaxed_bernoulli torch.distributions.relaxed_categorical torch.distributions.studentT torch.distributions.transformed_distribution torch.distributions.transforms torch.distributions.uniform torch.distributions.utils torch.distributions.von_mises torch.distributions.weibull torch.distributions.wishart torch.export torch.export.custom_obj torch.export.custom_ops torch.export.decomp_utils torch.export.dynamic_shapes torch.export.experimental torch.export.exported_program torch.export.graph_signature torch.export.passes torch.export.unflatten torch.fft torch.func torch.functional torch.futures torch.fx torch.fx.annotate torch.fx.config torch.fx.experimental torch.fx.experimental.accelerator_partitioner torch.fx.experimental.const_fold torch.fx.experimental.debug torch.fx.experimental.graph_gradual_typechecker torch.fx.experimental.merge_matmul torch.fx.experimental.meta_tracer torch.fx.experimental.migrate_gradual_types torch.fx.experimental.migrate_gradual_types.constraint torch.fx.experimental.migrate_gradual_types.constraint_generator torch.fx.experimental.migrate_gradual_types.constraint_transformation torch.fx.experimental.migrate_gradual_types.operation torch.fx.experimental.migrate_gradual_types.transform_to_z3 torch.fx.experimental.migrate_gradual_types.util torch.fx.experimental.migrate_gradual_types.z3_types torch.fx.experimental.normalize torch.fx.experimental.optimization torch.fx.experimental.partitioner_utils torch.fx.experimental.proxy_tensor torch.fx.experimental.recording torch.fx.experimental.refinement_types torch.fx.experimental.rewriter torch.fx.experimental.schema_type_annotation torch.fx.experimental.sym_node torch.fx.experimental.symbolic_shapes torch.fx.experimental.unification torch.fx.experimental.unification.core torch.fx.experimental.unification.dispatch torch.fx.experimental.unification.match torch.fx.experimental.unification.more torch.fx.experimental.unification.multipledispatch torch.fx.experimental.unification.multipledispatch.conflict torch.fx.experimental.unification.multipledispatch.core torch.fx.experimental.unification.multipledispatch.dispatcher torch.fx.experimental.unification.multipledispatch.utils torch.fx.experimental.unification.multipledispatch.variadic torch.fx.experimental.unification.unification_tools torch.fx.experimental.unification.utils torch.fx.experimental.unification.variable torch.fx.experimental.unify_refinements torch.fx.experimental.validator torch.fx.graph torch.fx.graph_module torch.fx.immutable_collections torch.fx.interpreter torch.fx.node torch.fx.operator_schemas torch.fx.passes torch.fx.passes.annotate_getitem_nodes torch.fx.passes.backends torch.fx.passes.backends.cudagraphs torch.fx.passes.dialect torch.fx.passes.dialect.common torch.fx.passes.dialect.common.cse_pass torch.fx.passes.fake_tensor_prop torch.fx.passes.graph_drawer torch.fx.passes.graph_manipulation torch.fx.passes.graph_transform_observer torch.fx.passes.infra torch.fx.passes.infra.partitioner torch.fx.passes.infra.pass_base torch.fx.passes.infra.pass_manager torch.fx.passes.net_min_base torch.fx.passes.operator_support torch.fx.passes.param_fetch torch.fx.passes.pass_manager torch.fx.passes.reinplace torch.fx.passes.runtime_assert torch.fx.passes.shape_prop torch.fx.passes.split_module torch.fx.passes.split_utils torch.fx.passes.splitter_base torch.fx.passes.tests torch.fx.passes.tests.test_pass_manager torch.fx.passes.tools_common torch.fx.passes.utils torch.fx.passes.utils.common torch.fx.passes.utils.fuser_utils torch.fx.passes.utils.matcher_utils torch.fx.passes.utils.matcher_with_name_node_map_utils torch.fx.passes.utils.source_matcher_utils torch.fx.proxy torch.fx.subgraph_rewriter torch.fx.tensor_type torch.fx.traceback torch.hub torch.jit torch.jit.annotations torch.jit.frontend torch.jit.generate_bytecode torch.jit.mobile torch.jit.quantized torch.jit.supported_ops torch.jit.unsupported_tensor_ops torch.library torch.linalg torch.masked torch.masked.maskedtensor torch.masked.maskedtensor.binary torch.masked.maskedtensor.core torch.masked.maskedtensor.creation torch.masked.maskedtensor.passthrough torch.masked.maskedtensor.reductions torch.masked.maskedtensor.unary torch.monitor torch.mps torch.mps.event torch.mps.profiler torch.mtia torch.mtia.memory torch.multiprocessing torch.multiprocessing.pool torch.multiprocessing.queue torch.multiprocessing.reductions torch.multiprocessing.spawn torch.nested torch.nn torch.nn.attention torch.nn.attention.bias torch.nn.attention.experimental torch.nn.attention.flex_attention torch.nn.backends torch.nn.backends.thnn torch.nn.common_types torch.nn.cpp torch.nn.functional torch.nn.grad torch.nn.init torch.nn.intrinsic torch.nn.intrinsic.modules torch.nn.intrinsic.modules.fused torch.nn.intrinsic.qat torch.nn.intrinsic.qat.modules torch.nn.intrinsic.qat.modules.conv_fused torch.nn.intrinsic.qat.modules.linear_fused torch.nn.intrinsic.qat.modules.linear_relu torch.nn.intrinsic.quantized torch.nn.intrinsic.quantized.dynamic torch.nn.intrinsic.quantized.dynamic.modules torch.nn.intrinsic.quantized.dynamic.modules.linear_relu torch.nn.intrinsic.quantized.modules torch.nn.intrinsic.quantized.modules.bn_relu torch.nn.intrinsic.quantized.modules.conv_relu torch.nn.intrinsic.quantized.modules.linear_relu torch.nn.modules torch.nn.modules.activation torch.nn.modules.adaptive torch.nn.modules.batchnorm torch.nn.modules.channelshuffle torch.nn.modules.container torch.nn.modules.conv torch.nn.modules.distance torch.nn.modules.dropout torch.nn.modules.flatten torch.nn.modules.fold torch.nn.modules.instancenorm torch.nn.modules.lazy torch.nn.modules.linear torch.nn.modules.loss torch.nn.modules.module torch.nn.modules.normalization torch.nn.modules.padding torch.nn.modules.pixelshuffle torch.nn.modules.pooling torch.nn.modules.rnn torch.nn.modules.sparse torch.nn.modules.transformer torch.nn.modules.upsampling torch.nn.modules.utils torch.nn.parallel torch.nn.parallel.comm torch.nn.parallel.distributed torch.nn.parallel.parallel_apply torch.nn.parallel.replicate torch.nn.parallel.scatter_gather torch.nn.parameter torch.nn.qat torch.nn.qat.dynamic torch.nn.qat.dynamic.modules torch.nn.qat.dynamic.modules.linear torch.nn.qat.modules torch.nn.qat.modules.conv torch.nn.qat.modules.embedding_ops torch.nn.qat.modules.linear torch.nn.quantizable torch.nn.quantizable.modules torch.nn.quantizable.modules.activation torch.nn.quantizable.modules.rnn torch.nn.quantized torch.nn.quantized.dynamic torch.nn.quantized.dynamic.modules torch.nn.quantized.dynamic.modules.conv torch.nn.quantized.dynamic.modules.linear torch.nn.quantized.dynamic.modules.rnn torch.nn.quantized.functional torch.nn.quantized.modules torch.nn.quantized.modules.activation torch.nn.quantized.modules.batchnorm torch.nn.quantized.modules.conv torch.nn.quantized.modules.dropout torch.nn.quantized.modules.embedding_ops torch.nn.quantized.modules.functional_modules torch.nn.quantized.modules.linear torch.nn.quantized.modules.normalization torch.nn.quantized.modules.rnn torch.nn.quantized.modules.utils torch.nn.utils torch.nn.utils.clip_grad torch.nn.utils.convert_parameters torch.nn.utils.fusion torch.nn.utils.init torch.nn.utils.memory_format torch.nn.utils.parametrizations torch.nn.utils.parametrize torch.nn.utils.prune torch.nn.utils.rnn torch.nn.utils.stateless torch.onnx torch.onnx.errors torch.onnx.operators torch.onnx.ops torch.onnx.symbolic_caffe2 torch.onnx.symbolic_helper torch.onnx.symbolic_opset10 torch.onnx.symbolic_opset11 torch.onnx.symbolic_opset12 torch.onnx.symbolic_opset13 torch.onnx.symbolic_opset14 torch.onnx.symbolic_opset15 torch.onnx.symbolic_opset16 torch.onnx.symbolic_opset17 torch.onnx.symbolic_opset18 torch.onnx.symbolic_opset19 torch.onnx.symbolic_opset20 torch.onnx.symbolic_opset7 torch.onnx.symbolic_opset8 torch.onnx.symbolic_opset9 torch.onnx.utils torch.onnx.verification torch.optim torch.optim.adadelta torch.optim.adagrad torch.optim.adam torch.optim.adamax torch.optim.adamw torch.optim.asgd torch.optim.lbfgs torch.optim.lr_scheduler torch.optim.nadam torch.optim.optimizer torch.optim.radam torch.optim.rmsprop torch.optim.rprop torch.optim.sgd torch.optim.sparse_adam torch.optim.swa_utils torch.overrides torch.package torch.package.analyze torch.package.analyze.find_first_use_of_broken_modules torch.package.analyze.is_from_package torch.package.analyze.trace_dependencies torch.package.file_structure_representation torch.package.find_file_dependencies torch.package.glob_group torch.package.importer torch.package.package_exporter torch.package.package_importer torch.profiler torch.profiler.itt torch.profiler.profiler torch.profiler.python_tracer torch.quantization torch.quantization.fake_quantize torch.quantization.fuse_modules torch.quantization.fuser_method_mappings torch.quantization.fx torch.quantization.fx.convert torch.quantization.fx.fuse torch.quantization.fx.fusion_patterns torch.quantization.fx.graph_module torch.quantization.fx.match_utils torch.quantization.fx.pattern_utils torch.quantization.fx.prepare torch.quantization.fx.quantization_patterns torch.quantization.fx.quantization_types torch.quantization.fx.utils torch.quantization.observer torch.quantization.qconfig torch.quantization.quant_type torch.quantization.quantization_mappings torch.quantization.quantize torch.quantization.quantize_fx torch.quantization.quantize_jit torch.quantization.stubs torch.quantization.utils torch.quasirandom torch.random torch.return_types torch.serialization torch.signal torch.signal.windows torch.signal.windows.windows torch.sparse torch.sparse.semi_structured torch.special torch.storage torch.testing torch.torch_version torch.types torch.utils torch.utils.backcompat torch.utils.backend_registration torch.utils.benchmark torch.utils.benchmark.examples torch.utils.benchmark.examples.compare torch.utils.benchmark.examples.fuzzer torch.utils.benchmark.examples.op_benchmark torch.utils.benchmark.examples.simple_timeit torch.utils.benchmark.examples.spectral_ops_fuzz_test torch.utils.benchmark.op_fuzzers torch.utils.benchmark.op_fuzzers.binary torch.utils.benchmark.op_fuzzers.sparse_binary torch.utils.benchmark.op_fuzzers.sparse_unary torch.utils.benchmark.op_fuzzers.spectral torch.utils.benchmark.op_fuzzers.unary torch.utils.benchmark.utils torch.utils.benchmark.utils.common torch.utils.benchmark.utils.compare torch.utils.benchmark.utils.compile torch.utils.benchmark.utils.cpp_jit torch.utils.benchmark.utils.fuzzer torch.utils.benchmark.utils.sparse_fuzzer torch.utils.benchmark.utils.timer torch.utils.benchmark.utils.valgrind_wrapper torch.utils.benchmark.utils.valgrind_wrapper.timer_interface torch.utils.bottleneck torch.utils.bundled_inputs torch.utils.checkpoint torch.utils.collect_env torch.utils.cpp_backtrace torch.utils.cpp_extension torch.utils.data torch.utils.data.backward_compatibility torch.utils.data.dataloader torch.utils.data.datapipes torch.utils.data.datapipes.dataframe torch.utils.data.datapipes.dataframe.dataframe_wrapper torch.utils.data.datapipes.dataframe.dataframes torch.utils.data.datapipes.dataframe.datapipes torch.utils.data.datapipes.dataframe.structures torch.utils.data.datapipes.datapipe torch.utils.data.datapipes.gen_pyi torch.utils.data.datapipes.iter torch.utils.data.datapipes.iter.callable torch.utils.data.datapipes.iter.combinatorics torch.utils.data.datapipes.iter.combining torch.utils.data.datapipes.iter.filelister torch.utils.data.datapipes.iter.fileopener torch.utils.data.datapipes.iter.grouping torch.utils.data.datapipes.iter.routeddecoder torch.utils.data.datapipes.iter.selecting torch.utils.data.datapipes.iter.sharding torch.utils.data.datapipes.iter.streamreader torch.utils.data.datapipes.iter.utils torch.utils.data.datapipes.map torch.utils.data.datapipes.map.callable torch.utils.data.datapipes.map.combinatorics torch.utils.data.datapipes.map.combining torch.utils.data.datapipes.map.grouping torch.utils.data.datapipes.map.utils torch.utils.data.datapipes.utils torch.utils.data.datapipes.utils.common torch.utils.data.datapipes.utils.decoder torch.utils.data.datapipes.utils.snapshot torch.utils.data.dataset torch.utils.data.distributed torch.utils.data.graph torch.utils.data.graph_settings torch.utils.data.sampler torch.utils.deterministic torch.utils.dlpack torch.utils.file_baton torch.utils.flop_counter torch.utils.hipify torch.utils.hipify.constants torch.utils.hipify.cuda_to_hip_mappings torch.utils.hipify.hipify_python torch.utils.hipify.version torch.utils.hooks torch.utils.jit torch.utils.jit.log_extract torch.utils.mkldnn torch.utils.mobile_optimizer torch.utils.model_dump torch.utils.model_zoo torch.utils.module_tracker torch.utils.serialization torch.utils.serialization.config torch.utils.show_pickle torch.utils.tensorboard torch.utils.tensorboard.summary torch.utils.tensorboard.writer torch.utils.throughput_benchmark torch.utils.viz torch.utils.weak torch.version torch.xpu torch.xpu.memory torch.xpu.random torch.xpu.streams Module (class in torch.nn) module (torch.distributed.fsdp.FullyShardedDataParallel property) module() (torch.export.ExportedProgram method) module_load() (torch.Tensor method) ModuleCallEntry (class in torch.export) ModuleCallSignature (class in torch.export) ModuleDict (class in torch.nn) ModuleList (class in torch.nn) modules() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) ModuleTracker (class in torch.utils.module_tracker) monitored_barrier() (in module torch.distributed) move_to_device_pass() (in module torch.export.passes) moveaxis() (in module torch) (torch.Tensor method) movedim() (in module torch) (torch.Tensor method) MovingAverageMinMaxObserver (class in torch.ao.quantization.observer) MovingAveragePerChannelMinMaxObserver (class in torch.ao.quantization.observer) mps() (torch.UntypedStorage method) mse_loss() (in module torch.nn.functional) MSELoss (class in torch.nn) msort() (in module torch) (torch.Tensor method) mT (torch.Tensor attribute) mtia() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) mul() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) (torch.Tensor method) mul_() (torch.Tensor method) mul_scalar() (torch.ao.ns._numeric_suite.Shadow method) multi_dot() (in module torch.linalg) multi_get() (torch.distributed.Store method) multi_margin_loss() (in module torch.nn.functional) multi_set() (torch.distributed.Store method) multigammaln() (in module torch.special) MultiheadAttention (class in torch.ao.nn.quantizable) (class in torch.nn) multilabel_margin_loss() (in module torch.nn.functional) multilabel_soft_margin_loss() (in module torch.nn.functional) MultiLabelMarginLoss (class in torch.nn) MultiLabelSoftMarginLoss (class in torch.nn) MultiMarginLoss (class in torch.nn) Multinomial (class in torch.distributions.multinomial) multinomial (in module torch.distributions.constraints) multinomial() (in module torch) (torch.Tensor method) MultiplicativeLR (class in torch.optim.lr_scheduler) multiply() (in module torch) (torch.Tensor method) multiply_() (torch.Tensor method) MultiprocessContext (class in torch.distributed.elastic.multiprocessing.api) MultiStepLR (class in torch.optim.lr_scheduler) MultivariateNormal (class in torch.distributions.multivariate_normal) mv() (in module torch) (torch.Tensor method) mvlgamma() (in module torch) (torch.Tensor method) mvlgamma_() (torch.Tensor method) N NAdam (class in torch.optim) name (torch.autograd.profiler_util.Kernel attribute) (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend property) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend property) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend property) (torch.distributed.rpc.WorkerInfo property) (torch.monitor.Aggregation property) (torch.monitor.Event property) (torch.monitor.Stat property) (torch.nn.attention.SDPBackend property) (torch.profiler.ProfilerActivity property) (torch.Tag property) name() (torch.autograd.graph.Node method) named_buffers() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_children() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_modules() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_parameters() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) NamedShape (torch.nn.Unflatten attribute) names (torch.Tensor attribute) nan_to_num() (in module torch) (torch.Tensor method) nan_to_num_() (torch.Tensor method) nanmean() (in module torch) (torch.Tensor method) nanmedian() (in module torch) (torch.Tensor method) nanquantile() (in module torch) (torch.Tensor method) nansum() (in module torch) (torch.Tensor method) narrow() (in module torch) (in module torch.nested) (torch.Tensor method) narrow_copy() (in module torch) (torch.Tensor method) nbytes (torch.Tensor attribute) nbytes() (torch.TypedStorage method) (torch.UntypedStorage method) ndim (torch.Tensor attribute) ndimension() (torch.Tensor method) ndtr() (in module torch.special) ndtri() (in module torch.special) ne() (in module torch) (torch.Tensor method) ne_() (torch.Tensor method) neg() (in module torch) (torch.Tensor method) neg_() (torch.Tensor method) negative() (in module torch) (torch.Tensor method) negative_() (torch.Tensor method) NegativeBinomial (class in torch.distributions.negative_binomial) nelement() (torch.Tensor method) nested_tensor() (in module torch.nested) nested_tensor_from_jagged() (in module torch.nested) NestedIOFunction (class in torch.autograd.function) new() (torch.UntypedStorage method) new_empty() (torch.Tensor method) new_full() (torch.Tensor method) new_group() (in module torch.distributed) new_ones() (torch.Tensor method) new_tensor() (torch.Tensor method) new_zeros() (torch.Tensor method) next (torch.fx.Node property) next_functions (torch.autograd.graph.Node property) next_rendezvous() (torch.distributed.elastic.rendezvous.RendezvousHandler method) nextafter() (in module torch) (torch.Tensor method) nextafter_() (torch.Tensor method) nll_loss() (in module torch.nn.functional) NLLLoss (class in torch.nn) no_grad (class in torch) no_sync() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.parallel.DistributedDataParallel method) Node (class in torch.fx) node_copy() (torch.fx.Graph method) nodes (torch.fx.Graph property) nonzero() (in module torch) (torch.Tensor method) noop_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks) noop_mask() (in module torch.nn.attention.flex_attention) NoopObserver (class in torch.ao.quantization.observer) norm() (in module torch) (in module torch.linalg) (torch.Tensor method) Normal (class in torch.distributions.normal) normal() (in module torch) normal_() (in module torch.nn.init) (torch.Tensor method) normalize() (in module torch.nn.functional) normalized_arguments() (torch.fx.Node method) not_equal() (in module torch) (torch.Tensor method) not_equal_() (torch.Tensor method) notify_join_context() (torch.distributed.algorithms.Join static method) NSTracer (class in torch.ao.ns._numeric_suite_fx) NullMetricHandler (class in torch.distributed.elastic.metrics.api) num_keys() (torch.distributed.Store method) num_nodes_waiting() (torch.distributed.elastic.rendezvous.RendezvousHandler method) num_worker_threads (torch.distributed.rpc.TensorPipeRpcBackendOptions property) numel() (in module torch) (torch.nn.attention.flex_attention.BlockMask method) (torch.Size method) (torch.Tensor method) NUMERIC_DEBUG_HANDLE_KEY (in module torch.ao.quantization) numpy() (torch.Tensor method) nuttall() (in module torch.signal.windows) O ObservationType (class in torch.ao.quantization.backend_config) ObserverBase (class in torch.ao.quantization.observer) OffloadPolicy (class in torch.distributed.fsdp) on_generate_code() (torch.fx.Graph method) once_differentiable() (in module torch.autograd.function) one_hot() (in module torch.nn.functional) OneCycleLR (class in torch.optim.lr_scheduler) onednn_fusion_enabled() (in module torch.jit) OneHotCategorical (class in torch.distributions.one_hot_categorical) ones() (in module torch) (in module torch.distributed.tensor) ones_() (in module torch.nn.init) ones_like() (in module torch) onnx_compatible() (torch.onnx.JitScalarType method) onnx_type() (torch.onnx.JitScalarType method) OnnxBackend (class in torch.onnx.verification) OnnxExporterError (class in torch.onnx) ONNXProgram (class in torch.onnx) OnnxTestCaseRepro (class in torch.onnx.verification) opcheck() (in module torch.library) optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) optim_state_dict_to_load() (torch.distributed.fsdp.FullyShardedDataParallel static method) optimize() (torch.onnx.ONNXProgram method) optimize_for_inference() (in module torch.jit) optimize_for_mobile() (in module torch.utils.mobile_optimizer) Optimizer (class in torch.optim) OptimStateDictConfig (class in torch.distributed.fsdp) or_masks() (in module torch.nn.attention.flex_attention) orgqr() (in module torch) (torch.Tensor method) ormqr() (in module torch) (torch.Tensor method) orthogonal() (in module torch.nn.utils.parametrizations) orthogonal_() (in module torch.nn.init) outer() (in module torch) (torch.Tensor method) OutOfMemoryError output() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) output_node() (torch.fx.Graph method) OUTPUT_SHARE_OBSERVER_WITH_INPUT (torch.ao.quantization.backend_config.ObservationType attribute) OUTPUT_USE_DIFFERENT_OBSERVER_AS_INPUT (torch.ao.quantization.backend_config.ObservationType attribute) OutputComparisonLogger (class in torch.ao.ns._numeric_suite_fx) OutputKind (class in torch.export.graph_signature) OutputLogger (class in torch.ao.ns._numeric_suite) (class in torch.ao.ns._numeric_suite_fx) OutputSpec (class in torch.export.graph_signature) owner() (torch.distributed.rpc.PyRRef method) owner_name() (torch.distributed.rpc.PyRRef method) P P2POp (class in torch.distributed) pack_padded_sequence() (in module torch.nn.utils.rnn) pack_sequence() (in module torch.nn.utils.rnn) PackageExporter (class in torch.package) PackageImporter (class in torch.package) PackagingError (class in torch.package) PackedSequence (class in torch.nn.utils.rnn) pad() (in module torch.nn.functional) pad_packed_sequence() (in module torch.nn.utils.rnn) pad_sequence() (in module torch.nn.utils.rnn) pairwise_distance() (in module torch.nn.functional) PairwiseDistance (class in torch.nn) parallel_info() (in module torch.__config__) parallelize_module() (in module torch.distributed.tensor.parallel) param_shape (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) Parameter (class in torch.nn.parameter) ParameterDict (class in torch.nn) ParameterList (class in torch.nn) parameters() (in module torch.distributed.GradBucket) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) parameters_to_vector() (in module torch.nn.utils) ParametrizationList (class in torch.nn.utils.parametrize) Pareto (class in torch.distributions.pareto) parse_nvprof_trace() (in module torch.autograd.profiler) Partial (class in torch.distributed.tensor.placement_types) path (torch.distributed.FileStore property) path_of_module() (torch.fx.Tracer method) pca_lowrank() (in module torch) PContext (class in torch.distributed.elastic.multiprocessing.api) pdist() (in module torch.nn.functional) per_channel_dynamic_qconfig (in module torch.ao.quantization.qconfig) PerAxis (class in torch.ao.quantization.observer) PerBlock (class in torch.ao.quantization.observer) PerChannelMinMaxObserver (class in torch.ao.quantization.observer) PerGroup (class in torch.ao.quantization.observer) permute() (in module torch) (torch.Tensor method) perplexity() (torch.distributions.distribution.Distribution method) PerRow (class in torch.ao.quantization.observer) PerTensor (class in torch.ao.quantization.observer) PerToken (class in torch.ao.quantization.observer) pickle_storage_type() (torch.TypedStorage method) pin_memory() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) pinv() (in module torch.linalg) pinverse() (in module torch) (torch.Tensor method) Pipe (class in torch.distributed.pipelining) pipe_split() (in module torch.distributed.pipelining) pipeline() (in module torch.distributed.pipelining) PipelineScheduleMulti (class in torch.distributed.pipelining.schedules) PipelineScheduleSingle (class in torch.distributed.pipelining.schedules) PipelineStage (class in torch.distributed.pipelining.stage) pixel_shuffle() (in module torch.nn.functional) pixel_unshuffle() (in module torch.nn.functional) PixelShuffle (class in torch.nn) PixelUnshuffle (class in torch.nn) placeholder() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) PlaceholderObserver (class in torch.ao.quantization.observer) Placement (class in torch.distributed.tensor.placement_types) placements (torch.distributed.tensor.DTensor property) Poisson (class in torch.distributions.poisson) poisson() (in module torch) poisson_nll_loss() (in module torch.nn.functional) PoissonNLLLoss (class in torch.nn) polar() (in module torch) polygamma() (in module torch) (in module torch.special) (torch.Tensor method) polygamma_() (torch.Tensor method) PolynomialLR (class in torch.optim.lr_scheduler) pool() (torch.cuda.CUDAGraph method) pop() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) popitem() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) port (torch.distributed.TCPStore property) positive() (in module torch) (torch.Tensor method) PositiveDefiniteTransform (class in torch.distributions.transforms) post_hook() (torch.distributed.algorithms.JoinHook method) PostLocalSGDOptimizer (class in torch.distributed.optim) pow() (in module torch) (torch.Tensor method) pow_() (torch.Tensor method) power_draw() (in module torch.cuda) powerSGD_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerSGDState (class in torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerTransform (class in torch.distributions.transforms) precision_matrix (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) predict() (torch.nn.AdaptiveLogSoftmaxWithLoss method) preferred_blas_library() (in module torch.backends.cuda) preferred_linalg_library() (in module torch.backends.cuda) preferred_rocm_fa_library() (in module torch.backends.cuda) PrefixStore (class in torch.distributed) PReLU (class in torch.nn) prelu() (in module torch.nn.functional) prepare (class in torch.ao.quantization) prepare_for_propagation_comparison (class in torch.ao.quantization) prepare_fx (class in torch.ao.quantization.quantize_fx) prepare_global_plan() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) prepare_local_plan() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) prepare_model_outputs() (in module torch.ao.ns._numeric_suite) prepare_model_with_stubs() (in module torch.ao.ns._numeric_suite) prepare_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) prepare_qat (class in torch.ao.quantization) prepare_qat_fx (class in torch.ao.quantization.quantize_fx) PrepareCustomConfig (class in torch.ao.quantization.fx.custom_config) PrepareModuleInput (class in torch.distributed.tensor.parallel) PrepareModuleInputOutput (class in torch.distributed.tensor.parallel) PrepareModuleOutput (class in torch.distributed.tensor.parallel) prepend() (torch.fx.Node method) preset_metadata_json() (torch.profiler._KinetoProfile method) prettify_results() (torch.fx.experimental.symbolic_shapes.DimConstraints method) prev (torch.fx.Node property) primal (torch.autograd.forward_ad.UnpackedDualTensor attribute) print() (torch.utils.benchmark.Compare method) print_comparisons_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) print_readable() (torch.fx.GraphModule method) print_tabular() (torch.fx.Graph method) probs (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.geometric.Geometric property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) process_inputs() (torch.fx.Graph method) process_outputs() (torch.fx.Graph method) ProcessFailure (class in torch.distributed.elastic.multiprocessing.errors) prod() (in module torch) (torch.Tensor method) produce_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) produce_guards_expression() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) produce_guards_verbose() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) prof() (in module torch.distributed.elastic.metrics) profile (class in torch.autograd.profiler) (class in torch.profiler) profile() (in module torch.mps.profiler) ProfilerAction (class in torch.profiler) ProfilerActivity (class in torch.profiler) promote_types() (in module torch) propagate_qconfig_ (class in torch.ao.quantization) PropagateUnbackedSymInts (class in torch.fx.experimental.symbolic_shapes) Proxy (class in torch.fx) proxy() (torch.fx.Tracer method) prune() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) PruningContainer (class in torch.nn.utils.prune) psi() (in module torch.special) put_() (torch.Tensor method) put_metric() (in module torch.distributed.elastic.metrics) PyRRef (class in torch.distributed.rpc) python_code() (torch.fx.Graph method) python_version() (torch.package.PackageImporter method) Q q_indices (torch.nn.attention.flex_attention.BlockMask attribute) q_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) q_per_channel_axis() (torch.Tensor method) q_per_channel_scales() (torch.Tensor method) q_per_channel_zero_points() (torch.Tensor method) q_scale() (torch.Tensor method) q_zero_point() (torch.Tensor method) QConfig (class in torch.ao.quantization.qconfig) QConfigMapping (class in torch.ao.quantization.qconfig_mapping) QFunctional (class in torch.ao.nn.quantized) QInt32Storage (class in torch) QInt8Storage (class in torch) qr() (in module torch) (in module torch.linalg) (torch.Tensor method) qscheme() (torch.Tensor method) quantile() (in module torch) (torch.Tensor method) quantize (class in torch.ao.quantization) quantize_dynamic (class in torch.ao.quantization) quantize_per_channel() (in module torch) quantize_per_tensor() (in module torch) quantize_qat (class in torch.ao.quantization) quantized_batch_norm() (in module torch) quantized_max_pool1d() (in module torch) quantized_max_pool2d() (in module torch) QuantStub (class in torch.ao.quantization) QuantWrapper (class in torch.ao.quantization) query() (torch.cuda.Event method) (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Event method) (torch.xpu.Stream method) queue_len() (torch.distributed.Store method) queue_pop() (torch.distributed.Store method) queue_push() (torch.distributed.Store method) QUInt2x4Storage (class in torch) QUInt4x2Storage (class in torch) QUInt8Storage (class in torch) R rad2deg() (in module torch) (torch.Tensor method) RAdam (class in torch.optim) rand() (in module torch) (in module torch.distributed.tensor) rand_like() (in module torch) randint() (in module torch) randint_like() (in module torch) randn() (in module torch) (in module torch.distributed.tensor) randn_like() (in module torch) random_() (torch.Tensor method) random_split() (in module torch.utils.data) random_structured() (in module torch.nn.utils.prune) random_unstructured() (in module torch.nn.utils.prune) RandomSampler (class in torch.utils.data) RandomStructured (class in torch.nn.utils.prune) RandomUnstructured (class in torch.nn.utils.prune) randperm() (in module torch) range() (in module torch) (in module torch.cuda.nvtx) range_pop() (in module torch.cuda.nvtx) (in module torch.profiler.itt) range_push() (in module torch.cuda.nvtx) (in module torch.profiler.itt) rate (torch.distributions.inverse_gamma.InverseGamma property) ravel() (in module torch) (torch.Tensor method) read_data() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) read_file() (in module torch.cuda.tunable) read_metadata() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) ReadItem (class in torch.distributed.checkpoint) real (torch.Tensor attribute) real() (in module torch) rebind_unbacked() (in module torch.fx.experimental.symbolic_shapes) reciprocal() (in module torch) (torch.Tensor method) reciprocal_() (torch.Tensor method) recommended_max_memory() (in module torch.mps) recompile() (torch.fx.GraphModule method) record() (in module torch.distributed.elastic.events) (in module torch.distributed.elastic.multiprocessing.errors) (torch.cuda.Event method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) record_event() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) record_function (class in torch.autograd.profiler) record_memory_history() (in module torch.mtia) record_stream() (torch.Tensor method) record_untuned_enable() (in module torch.cuda.tunable) record_untuned_is_enabled() (in module torch.cuda.tunable) RecordingObserver (class in torch.ao.quantization.observer) recursive_undo() (torch.optim.lr_scheduler.SequentialLR method) recv() (in module torch.distributed) recv_object_list() (in module torch.distributed) redistribute() (torch.distributed.tensor.DTensor method) reduce() (in module torch.distributed) reduce_add() (in module torch.cuda.comm) reduce_add_coalesced() (in module torch.cuda.comm) reduce_op (class in torch.distributed) (torch.distributed.tensor.placement_types.Partial attribute) reduce_scatter() (in module torch.distributed) reduce_scatter_tensor() (in module torch.distributed) ReduceLROnPlateau (class in torch.optim.lr_scheduler) ReduceOp (class in torch.distributed) refine_dynamic_shapes_from_suggested_fixes() (in module torch.export.dynamic_shapes) refine_names() (torch.Tensor method) ReflectionPad1d (class in torch.nn) ReflectionPad2d (class in torch.nn) ReflectionPad3d (class in torch.nn) register() (torch.distributions.constraint_registry.ConstraintRegistry method) register_autocast() (in module torch.library) register_autograd() (in module torch.library) register_backend() (torch.distributed.Backend class method) register_backward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_buffer() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_comm_hook() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.parallel.DistributedDataParallel method) register_custom_op_symbolic() (in module torch.onnx) register_dataclass() (in module torch.export) register_event_handler() (in module torch.monitor) register_extern_hook() (torch.package.PackageExporter method) register_fake() (in module torch.library) register_forward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_forward_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_fsdp_forward_method() (in module torch.distributed.fsdp) register_full_backward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_full_backward_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_handle() (torch.cuda.gds.GdsFile method) register_hook() (torch.autograd.graph.Node method) (torch.Tensor method) register_intern_hook() (torch.package.PackageExporter method) register_kernel() (in module torch.library) register_kl() (in module torch.distributions.kl) register_load_state_dict_post_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_load_state_dict_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_mock_hook() (torch.package.PackageExporter method) register_module() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_module_backward_hook() (in module torch.nn.modules.module) register_module_buffer_registration_hook() (in module torch.nn.modules.module) register_module_forward_hook() (in module torch.nn.modules.module) register_module_forward_pre_hook() (in module torch.nn.modules.module) register_module_full_backward_hook() (in module torch.nn.modules.module) register_module_full_backward_pre_hook() (in module torch.nn.modules.module) register_module_module_registration_hook() (in module torch.nn.modules.module) register_module_parameter_registration_hook() (in module torch.nn.modules.module) register_multi_grad_hook (class in torch.autograd.graph) register_package() (in module torch.serialization) register_parameter() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_parametrization() (in module torch.nn.utils.parametrize) register_post_accumulate_grad_hook() (torch.Tensor method) register_prehook() (torch.autograd.graph.Node method) register_sharding() (in module torch.distributed.tensor.experimental) register_state_dict_post_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_state_dict_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_step_post_hook() (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) register_step_pre_hook() (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) register_timers() (torch.distributed.elastic.timer.TimerServer method) register_torch_dispatch() (in module torch.library) register_vmap() (in module torch.library) reify() (torch.distributed.elastic.multiprocessing.api.DefaultLogsSpecs method) (torch.distributed.elastic.multiprocessing.api.LogsSpecs method) rekey_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) RelaxedBernoulli (class in torch.distributions.relaxed_bernoulli) RelaxedOneHotCategorical (class in torch.distributions.relaxed_categorical) RelaxedUnspecConstraint (class in torch.fx.experimental.symbolic_shapes) release() (torch.distributed.elastic.timer.TimerClient method) (torch.onnx.ONNXProgram method) ReLU (class in torch.nn) relu() (in module torch.nn.functional) ReLU6 (class in torch.ao.nn.quantized) (class in torch.nn) relu6() (in module torch.nn.functional) relu_() (in module torch.nn.functional) remainder() (in module torch) (torch.Tensor method) remainder_() (torch.Tensor method) remote() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) remote_parameters() (torch.distributed.nn.api.remote_module.RemoteModule method) RemoteModule (class in torch.distributed.nn.api.remote_module) remove() (in module torch.nn.utils.prune) (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) remove_parametrizations() (in module torch.nn.utils.parametrize) remove_spectral_norm() (in module torch.nn.utils) remove_weight_norm() (in module torch.nn.utils) rename() (torch.Tensor method) rename_() (torch.Tensor method) rename_privateuse1_backend() (in module torch.utils) render() (torch.fx.experimental.symbolic_shapes.StrictMinMaxConstraint method) RendezvousBackend (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) RendezvousClosedError (class in torch.distributed.elastic.rendezvous.api) RendezvousConnectionError (class in torch.distributed.elastic.rendezvous.api) RendezvousError (class in torch.distributed.elastic.rendezvous.api) RendezvousGracefulExitError (class in torch.distributed.elastic.rendezvous.api) RendezvousHandler (class in torch.distributed.elastic.rendezvous) RendezvousHandlerRegistry (class in torch.distributed.elastic.rendezvous) RendezvousInfo (class in torch.distributed.elastic.rendezvous) RendezvousParameters (class in torch.distributed.elastic.rendezvous) RendezvousStateError (class in torch.distributed.elastic.rendezvous.api) RendezvousStoreInfo (class in torch.distributed.elastic.rendezvous.api) RendezvousTimeout (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) RendezvousTimeoutError (class in torch.distributed.elastic.rendezvous.api) renorm() (in module torch) (torch.Tensor method) renorm_() (torch.Tensor method) repeat() (torch.Tensor method) repeat_interleave() (in module torch) (torch.Tensor method) replace() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) replace_all_batch_norm_modules_() (in module torch.func) replace_all_uses() (torch.export.graph_signature.ExportGraphSignature method) replace_all_uses_with() (torch.fx.Node method) replace_input_with() (torch.fx.Node method) replace_pattern() (in module torch.fx) replay() (torch.cuda.CUDAGraph method) Replicate (class in torch.distributed.tensor.placement_types) ReplicationPad1d (class in torch.nn) ReplicationPad2d (class in torch.nn) ReplicationPad3d (class in torch.nn) requires_grad (torch.Tensor attribute) requires_grad_() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) reset() (in module torch.compiler) (torch.cuda.CUDAGraph method) (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) (torch.quasirandom.SobolEngine method) reset_accumulated_memory_stats() (in module torch.xpu) reset_max_memory_allocated() (in module torch.cuda) reset_max_memory_cached() (in module torch.cuda) reset_min_max_vals() (torch.ao.quantization.observer.MinMaxObserver method) (torch.ao.quantization.observer.PerChannelMinMaxObserver method) reset_parameters() (torch.nn.modules.normalization.RMSNorm method) (torch.nn.RMSNorm method) reset_peak_host_memory_stats() (in module torch.cuda) reset_peak_memory_stats() (in module torch.cuda) (in module torch.xpu) reshape() (in module torch) (torch.Tensor method) reshape_as() (torch.Tensor method) ReshapeTransform (class in torch.distributions.transforms) reshard() (torch.distributed.fsdp.FSDPModule method) resizable() (torch.TypedStorage method) (torch.UntypedStorage method) resize_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) resize_as_() (torch.Tensor method) resolve_bytes() (torch.distributed.checkpoint.LoadPlanner method) resolve_conj() (in module torch) (torch.Tensor method) resolve_data() (torch.distributed.checkpoint.SavePlanner method) resolve_name() (in module torch.overrides) resolve_neg() (in module torch) (torch.Tensor method) resolve_tensor() (torch.distributed.checkpoint.LoadPlanner method) resolve_unbacked_bindings() (in module torch.fx.experimental.symbolic_shapes) result() (torch.distributed.Work method) result_type() (in module torch) retain_grad() (torch.Tensor method) retains_grad (torch.Tensor attribute) rewrite_with_congruences() (torch.fx.experimental.symbolic_shapes.DimConstraints method) rfft() (in module torch.fft) rfft2() (in module torch.fft) rfftfreq() (in module torch.fft) rfftn() (in module torch.fft) right_inverse() (torch.nn.utils.parametrize.ParametrizationList method) rms_norm() (in module torch.nn.functional) RMSNorm (class in torch.nn) (class in torch.nn.modules.normalization) RMSprop (class in torch.optim) RNN (class in torch.nn) RNNBase (class in torch.nn) RNNCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) roll() (in module torch) (torch.Tensor method) rot90() (in module torch) (torch.Tensor method) round() (in module torch) (in module torch.special) (torch.Tensor method) round_() (torch.Tensor method) row_indices() (torch.Tensor method) row_stack() (in module torch) RowwiseParallel (class in torch.distributed.tensor.parallel) rpc_async() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) rpc_sync() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) rpc_timeout (torch.distributed.rpc.RpcBackendOptions property) (torch.distributed.rpc.TensorPipeRpcBackendOptions property) RpcBackendOptions (class in torch.distributed.rpc) Rprop (class in torch.optim) RReLU (class in torch.nn) rrelu() (in module torch.nn.functional) rrelu_() (in module torch.nn.functional) rsample() (torch.distributions.beta.Beta method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.independent.Independent method) (torch.distributions.laplace.Laplace method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.normal.Normal method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.wishart.Wishart method) rsqrt() (in module torch) (torch.Tensor method) rsqrt_() (torch.Tensor method) run() (torch.distributed.elastic.agent.server.ElasticAgent method) (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) run_decompositions() (torch.export.ExportedProgram method) run_node() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) RunProcsResult (class in torch.distributed.elastic.multiprocessing.api) RunResult (class in torch.distributed.elastic.agent.server.api) S safe_globals (class in torch.serialization) sample() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.geometric.Geometric method) (torch.distributions.independent.Independent method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.poisson.Poisson method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.von_mises.VonMises method) sample_n() (torch.distributions.distribution.Distribution method) sampled_addmm() (in module torch.sparse) Sampler (class in torch.utils.data) save() (in module torch) (in module torch.distributed.checkpoint.state_dict_saver) (in module torch.export) (in module torch.jit) (torch.jit.ScriptFunction method) (torch.jit.ScriptModule method) (torch.onnx.ONNXProgram method) save_binary() (torch.package.PackageExporter method) save_for_backward() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) save_for_forward() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) save_module() (torch.package.PackageExporter method) save_on_cpu (class in torch.autograd.graph) save_pickle() (torch.package.PackageExporter method) save_source_file() (torch.package.PackageExporter method) save_source_string() (torch.package.PackageExporter method) save_state_dict() (in module torch.distributed.checkpoint.state_dict_saver) save_storage() (torch.cuda.gds.GdsFile method) save_text() (torch.package.PackageExporter method) save_to_buffer() (torch.jit.ScriptFunction method) saved_tensors (torch.autograd.function.NestedIOFunction property) saved_tensors_hooks (class in torch.autograd.graph) SavePlan (class in torch.distributed.checkpoint) SavePlanner (class in torch.distributed.checkpoint) scalar_name() (torch.onnx.JitScalarType method) scale (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.log_normal.LogNormal property) scale_fn() (torch.optim.lr_scheduler.CyclicLR method) scale_tril (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) scaled_dot_product_attention() (in module torch.nn.functional) scaled_modified_bessel_k0() (in module torch.special) scaled_modified_bessel_k1() (in module torch.special) scatter() (in module torch) (in module torch.cuda.comm) (in module torch.distributed) (torch.Tensor method) scatter_() (torch.Tensor method) scatter_add() (in module torch) (torch.Tensor method) scatter_add_() (torch.Tensor method) scatter_full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) scatter_object_list() (in module torch.distributed) scatter_reduce() (in module torch) (torch.Tensor method) scatter_reduce_() (torch.Tensor method) schedule() (in module torch.profiler) Schedule1F1B (class in torch.distributed.pipelining.schedules) ScheduleGPipe (class in torch.distributed.pipelining.schedules) ScheduleInterleaved1F1B (class in torch.distributed.pipelining.schedules) ScheduleInterleavedZeroBubble (class in torch.distributed.pipelining.schedules) ScheduleLoopedBFS (class in torch.distributed.pipelining.schedules) ScheduleZBVZeroBubble (class in torch.distributed.pipelining.schedules) script() (in module torch.jit) script_if_tracing() (in module torch.jit) ScriptFunction (class in torch.jit) ScriptModule (class in torch.jit) sdp_kernel() (in module torch.backends.cuda) sdpa_kernel() (in module torch.nn.attention) SDPAParams (class in torch.backends.cuda) SDPBackend (class in torch.nn.attention) searchsorted() (in module torch) see() (torch.autograd.profiler.EnforceUnique method) seed() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.random) (in module torch.xpu) (torch.Generator method) seed_all() (in module torch.cuda) (in module torch.xpu) select() (in module torch) (torch.Tensor method) select_model_mode_for_export() (in module torch.onnx) select_scatter() (in module torch) (torch.Tensor method) SelectiveCheckpointContext (class in torch.utils.checkpoint) self_cpu_time_total (torch.autograd.profiler.profile property) SELU (class in torch.nn) selu() (in module torch.nn.functional) send() (in module torch.distributed) send_object_list() (in module torch.distributed) seq_lengths (torch.nn.attention.flex_attention.BlockMask attribute) SequenceParallel (class in torch.distributed.tensor.parallel) Sequential (class in torch.nn) SequentialLR (class in torch.optim.lr_scheduler) SequentialSampler (class in torch.utils.data) set() (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) set_() (torch.Tensor method) set_all_reduce_hook() (torch.distributed.fsdp.FSDPModule method) set_backend_pattern_config() (torch.ao.quantization.backend_config.BackendConfig method) set_backend_pattern_configs() (torch.ao.quantization.backend_config.BackendConfig method) set_buffer() (in module torch.distributed.GradBucket) set_checkpoint_debug_enabled() (in module torch.utils.checkpoint) set_closed() (torch.distributed.elastic.rendezvous.RendezvousHandler method) set_codegen() (torch.fx.Graph method) set_crc32_options() (in module torch.serialization) set_custom_trace_id_callback() (torch.profiler.profile method) set_default_device() (in module torch) set_default_dtype() (in module torch) set_default_load_endianness() (in module torch.serialization) set_default_mmap_options() (in module torch.serialization) set_default_tensor_type() (in module torch) set_default_validate_args() (torch.distributions.distribution.Distribution static method) set_detect_anomaly (class in torch.autograd) set_deterministic_debug_mode() (in module torch) set_device() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) set_device_idx() (in module torch.accelerator) set_device_index() (in module torch.accelerator) set_device_map() (torch.distributed.rpc.TensorPipeRpcBackendOptions method) set_devices() (torch.distributed.rpc.TensorPipeRpcBackendOptions method) set_dir() (in module torch.hub) set_dtype_configs() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_exception() (torch.futures.Future method) set_extra_state() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) set_fastpath_enabled() (in module torch.backends.mha) set_filename() (in module torch.cuda.tunable) set_flags() (in module torch.backends.nnpack) set_float32_matmul_precision() (in module torch) set_float_to_observed_mapping() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_flush_denormal() (in module torch) set_fused_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_fuser_method() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_fusion_strategy() (in module torch.jit) set_global() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_grad_enabled (class in torch.autograd.grad_mode) set_input_quantized_indexes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_is_last_backward() (torch.distributed.fsdp.FSDPModule method) set_kernel_enabled() (torch._library.custom_ops.CustomOpDef method) set_logs() (in module torch._logging) set_materialize_grads() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) set_max_tuning_duration() (in module torch.cuda.tunable) set_max_tuning_iterations() (in module torch.cuda.tunable) set_model_state_dict() (in module torch.distributed.checkpoint.state_dict) set_module() (in module torch.utils) set_module_name() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_module_name_object_type_order() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_module_name_regex() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_modules_to_backward_prefetch() (torch.distributed.fsdp.FSDPModule method) set_modules_to_forward_prefetch() (torch.distributed.fsdp.FSDPModule method) set_multithreading_enabled (class in torch.autograd.grad_mode) set_name() (torch.ao.quantization.backend_config.BackendConfig method) set_non_traceable_module_classes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_non_traceable_module_names() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_num_interop_threads() (in module torch) set_num_threads() (in module torch) set_object_type() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_observation_type() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_observed_to_quantized_mapping() (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) set_optimizer_state_dict() (in module torch.distributed.checkpoint.state_dict) set_output_quantized_indexes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_overwrite_module_params_on_conversion() (in module torch.__future__) set_pattern() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_per_process_memory_fraction() (in module torch.cuda) (in module torch.mps) set_post_optim_event() (torch.distributed.fsdp.FSDPModule method) set_preserved_attributes() (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_printoptions() (in module torch) set_qat_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_reduce_scatter_divide_factor() (torch.distributed.fsdp.FSDPModule method) set_reference_quantized_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_requires_all_reduce() (torch.distributed.fsdp.FSDPModule method) set_requires_gradient_sync() (torch.distributed.fsdp.FSDPModule method) set_reshard_after_backward() (torch.distributed.fsdp.FSDPModule method) set_reshard_after_forward() (torch.distributed.fsdp.FSDPModule method) set_result() (torch.futures.Future method) set_rng_state() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.random) (in module torch.xpu) set_rng_state_all() (in module torch.cuda) (in module torch.xpu) set_root_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_rotating_buffer_size() (in module torch.cuda.tunable) set_sharing_strategy() (in module torch.multiprocessing) set_stance() (in module torch.compiler) set_standalone_module_class() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_standalone_module_name() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_state() (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend method) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend method) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend method) (torch.Generator method) set_state_dict() (in module torch.distributed.checkpoint.state_dict) set_state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) set_stream() (in module torch.accelerator) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) set_submodule() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) set_swap_module_params_on_conversion() (in module torch.__future__) set_sync_debug_mode() (in module torch.cuda) set_timeout() (torch.distributed.Store method) set_unbacked_var_to_val() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) set_unshard_in_backward() (torch.distributed.fsdp.FSDPModule method) set_up_planner() (torch.distributed.checkpoint.format_utils.DynamicMetaLoadPlanner method) (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) set_up_storage_reader() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) set_up_storage_writer() (torch.distributed.checkpoint.StorageWriter method) set_warn_always() (in module torch) setdefault() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) setup_context() (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) SGD (class in torch.optim) sgn() (in module torch) (torch.Tensor method) sgn_() (torch.Tensor method) Shadow (class in torch.ao.ns._numeric_suite) ShadowLogger (class in torch.ao.ns._numeric_suite) shape (torch.nn.attention.flex_attention.BlockMask property) (torch.Tensor attribute) ShapeEnv (class in torch.fx.experimental.symbolic_shapes) ShapeEnvSettings (class in torch.fx.experimental.symbolic_shapes) ShapesCollection (class in torch.export.dynamic_shapes) Shard (class in torch.distributed.tensor.placement_types) shard_full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) sharded_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) ShardedOptimStateDictConfig (class in torch.distributed.fsdp) ShardedStateDictConfig (class in torch.distributed.fsdp) ShardingStrategy (class in torch.distributed.fsdp) share_memory() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) share_memory_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) short() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) ShortStorage (class in torch) should_synchronize_after_execute (torch.distributed.checkpoint.staging.AsyncStager property) show() (in module torch.__config__) shutdown() (in module torch.distributed.rpc) (torch.distributed.elastic.rendezvous.RendezvousHandler method) Sigmoid (class in torch.ao.nn.quantized) (class in torch.nn) sigmoid() (in module torch) (in module torch.nn.functional) (torch.Tensor method) sigmoid_() (torch.Tensor method) SigmoidTransform (class in torch.distributions.transforms) sign (torch.distributions.transforms.Transform property) sign() (in module torch) (torch.Tensor method) sign_() (torch.Tensor method) signbit() (in module torch) (torch.Tensor method) significant_figures (torch.utils.benchmark.Measurement property) SiLU (class in torch.nn) silu() (in module torch.nn.functional) SimpleElasticAgent (class in torch.distributed.elastic.agent.server) simplify() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) sin() (in module torch) (torch.Tensor method) sin_() (torch.Tensor method) sinc() (in module torch) (in module torch.special) (torch.Tensor method) sinc_() (torch.Tensor method) sinh() (in module torch) (torch.Tensor method) sinh_() (torch.Tensor method) Size (class in torch) size (in module torch.backends.cuda.cufft_plan_cache) size() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) size_hint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) skip_data (class in torch.serialization) skip_init() (in module torch.nn.utils) slice_scatter() (in module torch) (torch.Tensor method) slogdet() (in module torch) (in module torch.linalg) (torch.Tensor method) smm() (in module torch) (torch.Tensor method) smooth_l1_loss() (in module torch.nn.functional) SmoothL1Loss (class in torch.nn) snapshot() (in module torch.mtia) (torch.cuda.MemPool method) SobolEngine (class in torch.quasirandom) soft_margin_loss() (in module torch.nn.functional) SoftMarginLoss (class in torch.nn) Softmax (class in torch.nn) softmax() (in module torch) (in module torch.nn.functional) (in module torch.sparse) (in module torch.special) (torch.Tensor method) Softmax2d (class in torch.nn) SoftmaxTransform (class in torch.distributions.transforms) Softmin (class in torch.nn) softmin() (in module torch.nn.functional) Softplus (class in torch.nn) softplus() (in module torch.nn.functional) SoftplusTransform (class in torch.distributions.transforms) Softshrink (class in torch.nn) softshrink() (in module torch.nn.functional) Softsign (class in torch.nn) softsign() (in module torch.nn.functional) solve() (in module torch.linalg) (torch.fx.experimental.symbolic_shapes.DimConstraints method) solve_ex() (in module torch.linalg) solve_triangular() (in module torch.linalg) sort() (in module torch) (torch.Tensor method) sorted_indices (torch.nn.utils.rnn.PackedSequence attribute) source_rank() (torch.distributed.Work method) sparse_() (in module torch.nn.init) sparse_bsc_tensor() (in module torch) sparse_bsr_tensor() (in module torch) sparse_compressed_tensor() (in module torch) sparse_coo_tensor() (in module torch) sparse_csc_tensor() (in module torch) sparse_csr_tensor() (in module torch) sparse_dim() (torch.Tensor method) sparse_mask() (torch.Tensor method) sparse_resize_() (torch.Tensor method) sparse_resize_and_clear_() (torch.Tensor method) SparseAdam (class in torch.optim) sparsity() (torch.nn.attention.flex_attention.BlockMask method) spawn() (in module torch.multiprocessing.spawn) SpawnContext (class in torch.multiprocessing) spdiags() (in module torch.sparse) spectral_norm() (in module torch.nn.utils) (in module torch.nn.utils.parametrizations) spherical_bessel_j0() (in module torch.special) split() (in module torch) (torch.Tensor method) split_args_kwargs_into_chunks() (in module torch.distributed.pipelining.microbatch) SplitPoint (class in torch.distributed.pipelining) spsolve() (in module torch.sparse) sqrt() (in module torch) (torch.Tensor method) sqrt_() (torch.Tensor method) square() (in module torch) (torch.Tensor method) square_() (torch.Tensor method) squeeze() (in module torch) (torch.Tensor method) squeeze_() (torch.Tensor method) sspaddmm() (in module torch) (torch.Tensor method) stack (in module torch.distributions.constraints) stack() (in module torch) stack_module_state() (in module torch.func) stack_trace (torch.fx.Node property) StackDataset (class in torch.utils.data) StackTransform (class in torch.distributions.transforms) stage() (torch.distributed.checkpoint.FileSystemWriter method) (torch.distributed.checkpoint.staging.AsyncStager method) (torch.distributed.checkpoint.staging.BlockingAsyncStager method) StandaloneModuleConfigEntry (class in torch.ao.quantization.fx.custom_config) start() (in module torch.mps.profiler) (torch.distributed.elastic.agent.server.health_check_server.HealthCheckServer method) start_processes() (in module torch.distributed.elastic.multiprocessing) Stat (class in torch.monitor) state_dict() (torch.distributed.checkpoint.stateful.Stateful method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) (torch.optim.swa_utils.SWALR method) state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) StateDictConfig (class in torch.distributed.fsdp) StateDictOptions (class in torch.distributed.checkpoint.state_dict) StateDictSettings (class in torch.distributed.fsdp) Stateful (class in torch.distributed.checkpoint.stateful) StatefulSymbolicContext (class in torch.fx.experimental.symbolic_shapes) StatelessSymbolicContext (class in torch.fx.experimental.symbolic_shapes) statically_known_true() (in module torch.fx.experimental.symbolic_shapes) stats() (torch.utils.benchmark.CallgrindStats method) std() (in module torch) (torch.Tensor method) std_mean() (in module torch) stddev (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.laplace.Laplace property) (torch.distributions.normal.Normal property) (torch.distributions.uniform.Uniform property) step() (torch.distributed.optim.DistributedOptimizer method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.distributed.pipelining.schedules.PipelineScheduleMulti method) (torch.distributed.pipelining.schedules.PipelineScheduleSingle method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.SWALR method) (torch.profiler.profile method) StepLR (class in torch.optim.lr_scheduler) stft() (in module torch) (torch.Tensor method) StickBreakingTransform (class in torch.distributions.transforms) stop() (in module torch.mps.profiler) (torch.distributed.elastic.agent.server.health_check_server.HealthCheckServer method) storage() (torch.Tensor method) storage_meta() (torch.distributed.checkpoint.StorageWriter method) storage_offset() (torch.Tensor method) storage_type() (torch.Tensor method) StorageReader (class in torch.distributed.checkpoint) StorageWriter (class in torch.distributed.checkpoint) Store (class in torch.distributed) strategy (in module torch.backends.opt_einsum) Stream (class in torch) (class in torch.cpu) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) stream() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) StreamContext (class in torch.cpu) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) strict_fusion (class in torch.jit) StrictMinMaxConstraint (class in torch.fx.experimental.symbolic_shapes) stride() (torch.Tensor method) StringTable (class in torch.autograd.profiler_util) StudentT (class in torch.distributions.studentT) sub() (in module torch) (torch.Tensor method) sub_() (torch.Tensor method) SubclassSymbolicContext (class in torch.fx.experimental.symbolic_shapes) SubprocessContext (class in torch.distributed.elastic.multiprocessing.api) SubprocessHandler (class in torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler) Subset (class in torch.utils.data) SubsetRandomSampler (class in torch.utils.data) substitute_in_graph() (in module torch.compiler) subtract() (in module torch) (torch.Tensor method) subtract_() (torch.Tensor method) sum() (in module torch) (in module torch.sparse) (torch.Tensor method) sum_to_size() (torch.Tensor method) SummaryWriter (class in torch.utils.tensorboard.writer) summon_full_params() (torch.distributed.fsdp.FullyShardedDataParallel static method) support (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.beta.Beta attribute) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric attribute) (torch.distributions.gumbel.Gumbel attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.lkj_cholesky.LKJCholesky attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.negative_binomial.NegativeBinomial attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson attribute) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.weibull.Weibull attribute) (torch.distributions.wishart.Wishart attribute) suppress_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) svd() (in module torch) (in module torch.linalg) (torch.Tensor method) svd_lowrank() (in module torch) svdvals() (in module torch.linalg) SWALR (class in torch.optim.swa_utils) swap_module (class in torch.ao.quantization) swap_tensors() (in module torch.utils) swapaxes() (in module torch) (torch.Tensor method) swapdims() (in module torch) (torch.Tensor method) SyclExtension() (in module torch.utils.cpp_extension) sym_and() (in module torch.fx.experimental.symbolic_shapes) sym_eq() (in module torch.fx.experimental.symbolic_shapes) sym_float() (in module torch) sym_fresh_size() (in module torch) sym_int() (in module torch) sym_ite() (in module torch) sym_max() (in module torch) sym_min() (in module torch) sym_not() (in module torch) sym_or() (in module torch.fx.experimental.symbolic_shapes) sym_sum() (in module torch) symbolic() (in module torch.onnx.ops) symbolic_multi_out() (in module torch.onnx.ops) symbolic_trace() (in module torch.fx) SymbolicContext (class in torch.fx.experimental.symbolic_shapes) SymBool (class in torch) SymBoolArgument (class in torch.export.graph_signature) SymFloat (class in torch) SymFloatArgument (class in torch.export.graph_signature) SymInt (class in torch) SymIntArgument (class in torch.export.graph_signature) SyncBatchNorm (class in torch.nn) synchronize() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) (torch.cuda.Event method) (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.distributed.Work method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Event method) (torch.xpu.Stream method) synchronize_staging() (torch.distributed.checkpoint.staging.AsyncStager method) (torch.distributed.checkpoint.staging.BlockingAsyncStager method) T T (torch.Tensor attribute) t() (in module torch) (torch.Tensor method) t_() (torch.Tensor method) Tag (class in torch) take() (in module torch) (torch.Tensor method) take_along_dim() (in module torch) (torch.Tensor method) tan() (in module torch) (torch.Tensor method) tan_() (torch.Tensor method) tangent (torch.autograd.forward_ad.UnpackedDualTensor attribute) Tanh (class in torch.nn) tanh() (in module torch) (in module torch.nn.functional) (torch.Tensor method) tanh_() (torch.Tensor method) Tanhshrink (class in torch.nn) tanhshrink() (in module torch.nn.functional) TanhTransform (class in torch.distributions.transforms) TCPStore (class in torch.distributed) temperature (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) temperature() (in module torch.cuda) Tensor (class in torch) tensor() (in module torch) tensor_split() (in module torch) (torch.Tensor method) tensor_storage_size() (torch.distributed.checkpoint.planner.WriteItem method) tensorboard_trace_handler() (in module torch.profiler) TensorboardEventHandler (class in torch.monitor) TensorChunkSpec (class in torch.distributed.pipelining.microbatch) TensorDataset (class in torch.utils.data) tensordot() (in module torch) tensorinv() (in module torch.linalg) TensorPipeRpcBackendOptions (class in torch.distributed.rpc) tensorsolve() (in module torch.linalg) then() (torch.futures.Future method) threshold (class in torch.ao.nn.quantized.functional) Threshold (class in torch.nn) threshold() (in module torch.nn.functional) threshold_() (in module torch.nn.functional) tile() (in module torch) (torch.Tensor method) timeit() (torch.utils.benchmark.Timer method) timeout (torch.distributed.Store property) Timer (class in torch.utils.benchmark) TimerClient (class in torch.distributed.elastic.timer) TimerRequest (class in torch.distributed.elastic.timer) TimerServer (class in torch.distributed.elastic.timer) timestamp (torch.monitor.Event property) to() (torch.jit.ScriptModule method) (torch.nn.attention.flex_attention.BlockMask method) (torch.nn.Module method) (torch.nn.utils.rnn.PackedSequence method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) to_bool() (torch.fx.Tracer method) to_dense() (torch.nn.attention.flex_attention.BlockMask method) (torch.Tensor method) to_dict() (torch.ao.quantization.backend_config.BackendConfig method) (torch.ao.quantization.backend_config.BackendPatternConfig method) (torch.ao.quantization.backend_config.DTypeConfig method) (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) (torch.ao.quantization.qconfig_mapping.QConfigMapping method) to_dlpack() (in module torch.utils.dlpack) to_empty() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) to_folder() (torch.fx.GraphModule method) to_here() (torch.distributed.rpc.PyRRef method) to_local() (torch.distributed.tensor.DTensor method) to_mkldnn() (torch.Tensor method) to_padded_tensor() (in module torch.nested) to_sparse() (torch.Tensor method) to_sparse_bsc() (torch.Tensor method) to_sparse_bsr() (torch.Tensor method) to_sparse_coo() (torch.Tensor method) to_sparse_csc() (torch.Tensor method) to_sparse_csr() (torch.Tensor method) to_string() (torch.nn.attention.flex_attention.BlockMask method) toggle_collection_dynamic() (torch.profiler._KinetoProfile method) tolist() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) topk() (in module torch) (torch.Tensor method) torch module torch.__config__ module torch.__future__ module torch._logging module torch.accelerator module torch.amp module torch.amp.autocast_mode module torch.amp.grad_scaler module torch.ao module torch.ao.nn module torch.ao.nn.intrinsic module torch.ao.nn.intrinsic.modules module torch.ao.nn.intrinsic.modules.fused module torch.ao.nn.intrinsic.qat module torch.ao.nn.intrinsic.qat.modules module torch.ao.nn.intrinsic.qat.modules.conv_fused module torch.ao.nn.intrinsic.qat.modules.linear_fused module torch.ao.nn.intrinsic.qat.modules.linear_relu module torch.ao.nn.intrinsic.quantized module torch.ao.nn.intrinsic.quantized.dynamic module torch.ao.nn.intrinsic.quantized.dynamic.modules module torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu module torch.ao.nn.intrinsic.quantized.modules module torch.ao.nn.intrinsic.quantized.modules.bn_relu module torch.ao.nn.intrinsic.quantized.modules.conv_add module torch.ao.nn.intrinsic.quantized.modules.conv_relu module torch.ao.nn.intrinsic.quantized.modules.linear_relu module torch.ao.nn.qat module torch.ao.nn.qat.dynamic module torch.ao.nn.qat.dynamic.modules module torch.ao.nn.qat.dynamic.modules.linear module torch.ao.nn.qat.modules module torch.ao.nn.qat.modules.conv module torch.ao.nn.qat.modules.embedding_ops module torch.ao.nn.qat.modules.linear module torch.ao.nn.quantizable module torch.ao.nn.quantizable.modules module torch.ao.nn.quantizable.modules.activation module torch.ao.nn.quantizable.modules.rnn module torch.ao.nn.quantized module torch.ao.nn.quantized.dynamic module torch.ao.nn.quantized.dynamic.modules module torch.ao.nn.quantized.dynamic.modules.conv module torch.ao.nn.quantized.dynamic.modules.linear module torch.ao.nn.quantized.dynamic.modules.rnn module torch.ao.nn.quantized.functional module torch.ao.nn.quantized.modules module torch.ao.nn.quantized.modules.activation module torch.ao.nn.quantized.modules.batchnorm module torch.ao.nn.quantized.modules.conv module torch.ao.nn.quantized.modules.dropout module torch.ao.nn.quantized.modules.embedding_ops module torch.ao.nn.quantized.modules.functional_modules module torch.ao.nn.quantized.modules.linear module torch.ao.nn.quantized.modules.normalization module torch.ao.nn.quantized.modules.rnn module torch.ao.nn.quantized.modules.utils module torch.ao.nn.quantized.reference module torch.ao.nn.quantized.reference.modules module torch.ao.nn.quantized.reference.modules.conv module torch.ao.nn.quantized.reference.modules.linear module torch.ao.nn.quantized.reference.modules.rnn module torch.ao.nn.quantized.reference.modules.sparse module torch.ao.nn.quantized.reference.modules.utils module torch.ao.nn.sparse module torch.ao.nn.sparse.quantized module torch.ao.nn.sparse.quantized.dynamic module torch.ao.nn.sparse.quantized.dynamic.linear module torch.ao.nn.sparse.quantized.linear module torch.ao.nn.sparse.quantized.utils module torch.ao.ns module torch.ao.ns._numeric_suite module torch.ao.ns._numeric_suite_fx module torch.ao.ns.fx module torch.ao.ns.fx.graph_matcher module torch.ao.ns.fx.graph_passes module torch.ao.ns.fx.mappings module torch.ao.ns.fx.n_shadows_utils module torch.ao.ns.fx.ns_types module torch.ao.ns.fx.pattern_utils module torch.ao.ns.fx.qconfig_multi_mapping module torch.ao.ns.fx.utils module torch.ao.ns.fx.weight_utils module torch.ao.pruning module torch.ao.pruning.scheduler module torch.ao.pruning.scheduler.base_scheduler module torch.ao.pruning.scheduler.cubic_scheduler module torch.ao.pruning.scheduler.lambda_scheduler module torch.ao.pruning.sparsifier module torch.ao.pruning.sparsifier.base_sparsifier module torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier module torch.ao.pruning.sparsifier.utils module torch.ao.pruning.sparsifier.weight_norm_sparsifier module torch.ao.quantization module torch.ao.quantization.backend_config module torch.ao.quantization.backend_config.backend_config module torch.ao.quantization.backend_config.executorch module torch.ao.quantization.backend_config.fbgemm module torch.ao.quantization.backend_config.native module torch.ao.quantization.backend_config.observation_type module torch.ao.quantization.backend_config.onednn module torch.ao.quantization.backend_config.qnnpack module torch.ao.quantization.backend_config.tensorrt module torch.ao.quantization.backend_config.utils module torch.ao.quantization.backend_config.x86 module torch.ao.quantization.fake_quantize module torch.ao.quantization.fuse_modules module torch.ao.quantization.fuser_method_mappings module torch.ao.quantization.fx module torch.ao.quantization.fx.convert module torch.ao.quantization.fx.custom_config module torch.ao.quantization.fx.fuse module torch.ao.quantization.fx.fuse_handler module torch.ao.quantization.fx.graph_module module torch.ao.quantization.fx.lower_to_fbgemm module torch.ao.quantization.fx.lower_to_qnnpack module torch.ao.quantization.fx.lstm_utils module torch.ao.quantization.fx.match_utils module torch.ao.quantization.fx.pattern_utils module torch.ao.quantization.fx.prepare module torch.ao.quantization.fx.qconfig_mapping_utils module torch.ao.quantization.fx.quantize_handler module torch.ao.quantization.fx.tracer module torch.ao.quantization.fx.utils module torch.ao.quantization.observer module torch.ao.quantization.pt2e module torch.ao.quantization.pt2e.duplicate_dq_pass module torch.ao.quantization.pt2e.export_utils module torch.ao.quantization.pt2e.graph_utils module torch.ao.quantization.pt2e.lowering module torch.ao.quantization.pt2e.port_metadata_pass module torch.ao.quantization.pt2e.prepare module torch.ao.quantization.pt2e.qat_utils module torch.ao.quantization.pt2e.representation module torch.ao.quantization.pt2e.representation.rewrite module torch.ao.quantization.pt2e.utils module torch.ao.quantization.qconfig module torch.ao.quantization.qconfig_mapping module torch.ao.quantization.quant_type module torch.ao.quantization.quantization_mappings module torch.ao.quantization.quantize_fx module torch.ao.quantization.quantize_jit module torch.ao.quantization.quantize_pt2e module torch.ao.quantization.quantizer module torch.ao.quantization.quantizer.composable_quantizer module torch.ao.quantization.quantizer.embedding_quantizer module torch.ao.quantization.quantizer.quantizer module torch.ao.quantization.quantizer.utils module torch.ao.quantization.quantizer.x86_inductor_quantizer module torch.ao.quantization.quantizer.xnnpack_quantizer module torch.ao.quantization.quantizer.xnnpack_quantizer_utils module torch.ao.quantization.quantizer.xpu_inductor_quantizer module torch.ao.quantization.stubs module torch.ao.quantization.utils module torch.autograd module torch.autograd.anomaly_mode module torch.autograd.forward_ad module torch.autograd.function module torch.autograd.functional module torch.autograd.grad_mode module torch.autograd.gradcheck module torch.autograd.graph module torch.autograd.profiler module torch.autograd.profiler_legacy module torch.autograd.profiler_util module torch.autograd.variable module torch.backends module torch.backends.cpu module torch.backends.cuda module torch.backends.cudnn module torch.backends.cudnn.rnn module torch.backends.cusparselt module torch.backends.kleidiai module torch.backends.mha module torch.backends.mkl module torch.backends.mkldnn module torch.backends.mps module torch.backends.nnpack module torch.backends.openmp module torch.backends.opt_einsum module torch.backends.quantized module torch.backends.xeon module torch.backends.xeon.run_cpu module torch.backends.xnnpack module torch.compiler module torch.compiler.config module torch.contrib module torch.cpu module torch.cpu.amp module torch.cpu.amp.autocast_mode module torch.cpu.amp.grad_scaler module torch.cuda module torch.cuda._sanitizer module torch.cuda.amp module torch.cuda.amp.autocast_mode module torch.cuda.amp.common module torch.cuda.amp.grad_scaler module torch.cuda.comm module torch.cuda.error module torch.cuda.gds module torch.cuda.graphs module torch.cuda.jiterator module torch.cuda.memory module torch.cuda.nccl module torch.cuda.nvtx module torch.cuda.profiler module torch.cuda.random module torch.cuda.sparse module torch.cuda.streams module torch.cuda.tunable module torch.distributed module torch.distributed.algorithms module torch.distributed.algorithms.ddp_comm_hooks module torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook module torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks module torch.distributed.algorithms.ddp_comm_hooks.default_hooks module torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks module torch.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks module torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook module torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks module torch.distributed.algorithms.join module torch.distributed.algorithms.model_averaging module torch.distributed.algorithms.model_averaging.averagers module torch.distributed.algorithms.model_averaging.hierarchical_model_averager module torch.distributed.algorithms.model_averaging.utils module torch.distributed.argparse_util module torch.distributed.autograd module torch.distributed.c10d_logger module torch.distributed.checkpoint module torch.distributed.checkpoint.api module torch.distributed.checkpoint.default_planner module torch.distributed.checkpoint.filesystem module torch.distributed.checkpoint.format_utils module torch.distributed.checkpoint.logger module torch.distributed.checkpoint.logging_handlers module torch.distributed.checkpoint.metadata module torch.distributed.checkpoint.optimizer module torch.distributed.checkpoint.planner module torch.distributed.checkpoint.planner_helpers module torch.distributed.checkpoint.resharding module torch.distributed.checkpoint.staging module torch.distributed.checkpoint.state_dict module torch.distributed.checkpoint.state_dict_loader module torch.distributed.checkpoint.state_dict_saver module torch.distributed.checkpoint.stateful module torch.distributed.checkpoint.storage module torch.distributed.checkpoint.utils module torch.distributed.collective_utils module torch.distributed.constants module torch.distributed.device_mesh module torch.distributed.distributed_c10d module torch.distributed.elastic module torch.distributed.elastic.agent module torch.distributed.elastic.agent.server module torch.distributed.elastic.agent.server.api module torch.distributed.elastic.agent.server.health_check_server module torch.distributed.elastic.agent.server.local_elastic_agent module torch.distributed.elastic.control_plane module torch.distributed.elastic.events module torch.distributed.elastic.events.api module torch.distributed.elastic.events.handlers module torch.distributed.elastic.metrics module torch.distributed.elastic.metrics.api module torch.distributed.elastic.multiprocessing module torch.distributed.elastic.multiprocessing.api module torch.distributed.elastic.multiprocessing.errors module torch.distributed.elastic.multiprocessing.errors.error_handler module torch.distributed.elastic.multiprocessing.errors.handlers module torch.distributed.elastic.multiprocessing.redirects module torch.distributed.elastic.multiprocessing.subprocess_handler module torch.distributed.elastic.multiprocessing.subprocess_handler.handlers module torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler module torch.distributed.elastic.multiprocessing.tail_log module torch.distributed.elastic.rendezvous module torch.distributed.elastic.rendezvous.api module torch.distributed.elastic.rendezvous.c10d_rendezvous_backend module torch.distributed.elastic.rendezvous.dynamic_rendezvous module torch.distributed.elastic.rendezvous.etcd_rendezvous module torch.distributed.elastic.rendezvous.etcd_rendezvous_backend module torch.distributed.elastic.rendezvous.etcd_server module torch.distributed.elastic.rendezvous.etcd_store module torch.distributed.elastic.rendezvous.registry module torch.distributed.elastic.rendezvous.static_tcp_rendezvous module torch.distributed.elastic.rendezvous.utils module torch.distributed.elastic.timer module torch.distributed.elastic.timer.api module torch.distributed.elastic.timer.debug_info_logging module torch.distributed.elastic.timer.file_based_local_timer module torch.distributed.elastic.timer.local_timer module torch.distributed.elastic.utils module torch.distributed.elastic.utils.api module torch.distributed.elastic.utils.data module torch.distributed.elastic.utils.data.cycling_iterator module torch.distributed.elastic.utils.data.elastic_distributed_sampler module torch.distributed.elastic.utils.distributed module torch.distributed.elastic.utils.log_level module torch.distributed.elastic.utils.logging module torch.distributed.elastic.utils.store module torch.distributed.fsdp module torch.distributed.fsdp.api module torch.distributed.fsdp.fully_sharded_data_parallel module torch.distributed.fsdp.sharded_grad_scaler module torch.distributed.fsdp.wrap module torch.distributed.launch module torch.distributed.launcher module torch.distributed.launcher.api module torch.distributed.logging_handlers module torch.distributed.nn module torch.distributed.nn.api module torch.distributed.nn.api.remote_module module torch.distributed.nn.functional module torch.distributed.nn.jit module torch.distributed.nn.jit.instantiator module torch.distributed.nn.jit.templates module torch.distributed.nn.jit.templates.remote_module_template module torch.distributed.optim module torch.distributed.optim.apply_optimizer_in_backward module torch.distributed.optim.functional_adadelta module torch.distributed.optim.functional_adagrad module torch.distributed.optim.functional_adam module torch.distributed.optim.functional_adamax module torch.distributed.optim.functional_adamw module torch.distributed.optim.functional_rmsprop module torch.distributed.optim.functional_rprop module torch.distributed.optim.functional_sgd module torch.distributed.optim.named_optimizer module torch.distributed.optim.optimizer module torch.distributed.optim.post_localSGD_optimizer module torch.distributed.optim.utils module torch.distributed.optim.zero_redundancy_optimizer module torch.distributed.pipelining module torch.distributed.pipelining.microbatch module torch.distributed.pipelining.schedules module torch.distributed.pipelining.stage module torch.distributed.remote_device module torch.distributed.rendezvous module torch.distributed.rpc module torch.distributed.rpc.api module torch.distributed.rpc.backend_registry module torch.distributed.rpc.constants module torch.distributed.rpc.functions module torch.distributed.rpc.internal module torch.distributed.rpc.options module torch.distributed.rpc.rref_proxy module torch.distributed.rpc.server_process_global_profiler module torch.distributed.run module torch.distributed.tensor module torch.distributed.tensor.debug module torch.distributed.tensor.device_mesh module torch.distributed.tensor.experimental module torch.distributed.tensor.parallel module torch.distributed.tensor.parallel.api module torch.distributed.tensor.parallel.ddp module torch.distributed.tensor.parallel.fsdp module torch.distributed.tensor.parallel.input_reshard module torch.distributed.tensor.parallel.loss module torch.distributed.tensor.parallel.style module torch.distributed.tensor.placement_types module torch.distributed.utils module torch.distributions module torch.distributions.bernoulli module torch.distributions.beta module torch.distributions.binomial module torch.distributions.categorical module torch.distributions.cauchy module torch.distributions.chi2 module torch.distributions.constraint_registry module torch.distributions.constraints module torch.distributions.continuous_bernoulli module torch.distributions.dirichlet module torch.distributions.distribution module torch.distributions.exp_family module torch.distributions.exponential module torch.distributions.fishersnedecor module torch.distributions.gamma module torch.distributions.generalized_pareto module torch.distributions.geometric module torch.distributions.gumbel module torch.distributions.half_cauchy module torch.distributions.half_normal module torch.distributions.independent module torch.distributions.inverse_gamma module torch.distributions.kl module torch.distributions.kumaraswamy module torch.distributions.laplace module torch.distributions.lkj_cholesky module torch.distributions.log_normal module torch.distributions.logistic_normal module torch.distributions.lowrank_multivariate_normal module torch.distributions.mixture_same_family module torch.distributions.multinomial module torch.distributions.multivariate_normal module torch.distributions.negative_binomial module torch.distributions.normal module torch.distributions.one_hot_categorical module torch.distributions.pareto module torch.distributions.poisson module torch.distributions.relaxed_bernoulli module torch.distributions.relaxed_categorical module torch.distributions.studentT module torch.distributions.transformed_distribution module torch.distributions.transforms module torch.distributions.uniform module torch.distributions.utils module torch.distributions.von_mises module torch.distributions.weibull module torch.distributions.wishart module torch.export module torch.export.custom_obj module torch.export.custom_ops module torch.export.decomp_utils module torch.export.dynamic_shapes module torch.export.experimental module torch.export.exported_program module torch.export.graph_signature module torch.export.passes module torch.export.unflatten module torch.fft module torch.finfo (class in torch) torch.func module torch.functional module torch.futures module torch.fx module torch.fx.annotate module torch.fx.config module torch.fx.experimental module torch.fx.experimental.accelerator_partitioner module torch.fx.experimental.const_fold module torch.fx.experimental.debug module torch.fx.experimental.graph_gradual_typechecker module torch.fx.experimental.merge_matmul module torch.fx.experimental.meta_tracer module torch.fx.experimental.migrate_gradual_types module torch.fx.experimental.migrate_gradual_types.constraint module torch.fx.experimental.migrate_gradual_types.constraint_generator module torch.fx.experimental.migrate_gradual_types.constraint_transformation module torch.fx.experimental.migrate_gradual_types.operation module torch.fx.experimental.migrate_gradual_types.transform_to_z3 module torch.fx.experimental.migrate_gradual_types.util module torch.fx.experimental.migrate_gradual_types.z3_types module torch.fx.experimental.normalize module torch.fx.experimental.optimization module torch.fx.experimental.partitioner_utils module torch.fx.experimental.proxy_tensor module torch.fx.experimental.recording module torch.fx.experimental.refinement_types module torch.fx.experimental.rewriter module torch.fx.experimental.schema_type_annotation module torch.fx.experimental.sym_node module torch.fx.experimental.symbolic_shapes module torch.fx.experimental.unification module torch.fx.experimental.unification.core module torch.fx.experimental.unification.dispatch module torch.fx.experimental.unification.match module torch.fx.experimental.unification.more module torch.fx.experimental.unification.multipledispatch module torch.fx.experimental.unification.multipledispatch.conflict module torch.fx.experimental.unification.multipledispatch.core module torch.fx.experimental.unification.multipledispatch.dispatcher module torch.fx.experimental.unification.multipledispatch.utils module torch.fx.experimental.unification.multipledispatch.variadic module torch.fx.experimental.unification.unification_tools module torch.fx.experimental.unification.utils module torch.fx.experimental.unification.variable module torch.fx.experimental.unify_refinements module torch.fx.experimental.validator module torch.fx.graph module torch.fx.graph_module module torch.fx.immutable_collections module torch.fx.interpreter module torch.fx.node module torch.fx.operator_schemas module torch.fx.passes module torch.fx.passes.annotate_getitem_nodes module torch.fx.passes.backends module torch.fx.passes.backends.cudagraphs module torch.fx.passes.dialect module torch.fx.passes.dialect.common module torch.fx.passes.dialect.common.cse_pass module torch.fx.passes.fake_tensor_prop module torch.fx.passes.graph_drawer module torch.fx.passes.graph_manipulation module torch.fx.passes.graph_transform_observer module torch.fx.passes.infra module torch.fx.passes.infra.partitioner module torch.fx.passes.infra.pass_base module torch.fx.passes.infra.pass_manager module torch.fx.passes.net_min_base module torch.fx.passes.operator_support module torch.fx.passes.param_fetch module torch.fx.passes.pass_manager module torch.fx.passes.reinplace module torch.fx.passes.runtime_assert module torch.fx.passes.shape_prop module torch.fx.passes.split_module module torch.fx.passes.split_utils module torch.fx.passes.splitter_base module torch.fx.passes.tests module torch.fx.passes.tests.test_pass_manager module torch.fx.passes.tools_common module torch.fx.passes.utils module torch.fx.passes.utils.common module torch.fx.passes.utils.fuser_utils module torch.fx.passes.utils.matcher_utils module torch.fx.passes.utils.matcher_with_name_node_map_utils module torch.fx.passes.utils.source_matcher_utils module torch.fx.proxy module torch.fx.subgraph_rewriter module torch.fx.tensor_type module torch.fx.traceback module torch.hub module torch.iinfo (class in torch) torch.jit module torch.jit.annotations module torch.jit.frontend module torch.jit.generate_bytecode module torch.jit.mobile module torch.jit.quantized module torch.jit.supported_ops module torch.jit.unsupported_tensor_ops module torch.library module torch.linalg module torch.masked module torch.masked.maskedtensor module torch.masked.maskedtensor.binary module torch.masked.maskedtensor.core module torch.masked.maskedtensor.creation module torch.masked.maskedtensor.passthrough module torch.masked.maskedtensor.reductions module torch.masked.maskedtensor.unary module torch.monitor module torch.mps module torch.mps.event module torch.mps.profiler module torch.mtia module torch.mtia.memory module torch.multiprocessing module torch.multiprocessing.pool module torch.multiprocessing.queue module torch.multiprocessing.reductions module torch.multiprocessing.spawn module torch.nested module torch.nn module torch.nn.attention module torch.nn.attention.bias module torch.nn.attention.experimental module torch.nn.attention.flex_attention module torch.nn.backends module torch.nn.backends.thnn module torch.nn.common_types module torch.nn.cpp module torch.nn.functional module torch.nn.grad module torch.nn.init module torch.nn.intrinsic module torch.nn.intrinsic.modules module torch.nn.intrinsic.modules.fused module torch.nn.intrinsic.qat module torch.nn.intrinsic.qat.modules module torch.nn.intrinsic.qat.modules.conv_fused module torch.nn.intrinsic.qat.modules.linear_fused module torch.nn.intrinsic.qat.modules.linear_relu module torch.nn.intrinsic.quantized module torch.nn.intrinsic.quantized.dynamic module torch.nn.intrinsic.quantized.dynamic.modules module torch.nn.intrinsic.quantized.dynamic.modules.linear_relu module torch.nn.intrinsic.quantized.modules module torch.nn.intrinsic.quantized.modules.bn_relu module torch.nn.intrinsic.quantized.modules.conv_relu module torch.nn.intrinsic.quantized.modules.linear_relu module torch.nn.modules module torch.nn.modules.activation module torch.nn.modules.adaptive module torch.nn.modules.batchnorm module torch.nn.modules.channelshuffle module torch.nn.modules.container module torch.nn.modules.conv module torch.nn.modules.distance module torch.nn.modules.dropout module torch.nn.modules.flatten module torch.nn.modules.fold module torch.nn.modules.instancenorm module torch.nn.modules.lazy module torch.nn.modules.linear module torch.nn.modules.loss module torch.nn.modules.module module torch.nn.modules.normalization module torch.nn.modules.padding module torch.nn.modules.pixelshuffle module torch.nn.modules.pooling module torch.nn.modules.rnn module torch.nn.modules.sparse module torch.nn.modules.transformer module torch.nn.modules.upsampling module torch.nn.modules.utils module torch.nn.parallel module torch.nn.parallel.comm module torch.nn.parallel.distributed module torch.nn.parallel.parallel_apply module torch.nn.parallel.replicate module torch.nn.parallel.scatter_gather module torch.nn.parameter module torch.nn.qat module torch.nn.qat.dynamic module torch.nn.qat.dynamic.modules module torch.nn.qat.dynamic.modules.linear module torch.nn.qat.modules module torch.nn.qat.modules.conv module torch.nn.qat.modules.embedding_ops module torch.nn.qat.modules.linear module torch.nn.quantizable module torch.nn.quantizable.modules module torch.nn.quantizable.modules.activation module torch.nn.quantizable.modules.rnn module torch.nn.quantized module torch.nn.quantized.dynamic module torch.nn.quantized.dynamic.modules module torch.nn.quantized.dynamic.modules.conv module torch.nn.quantized.dynamic.modules.linear module torch.nn.quantized.dynamic.modules.rnn module torch.nn.quantized.functional module torch.nn.quantized.modules module torch.nn.quantized.modules.activation module torch.nn.quantized.modules.batchnorm module torch.nn.quantized.modules.conv module torch.nn.quantized.modules.dropout module torch.nn.quantized.modules.embedding_ops module torch.nn.quantized.modules.functional_modules module torch.nn.quantized.modules.linear module torch.nn.quantized.modules.normalization module torch.nn.quantized.modules.rnn module torch.nn.quantized.modules.utils module torch.nn.utils module torch.nn.utils.clip_grad module torch.nn.utils.convert_parameters module torch.nn.utils.fusion module torch.nn.utils.init module torch.nn.utils.memory_format module torch.nn.utils.parametrizations module torch.nn.utils.parametrize module torch.nn.utils.prune module torch.nn.utils.rnn module torch.nn.utils.stateless module torch.onnx module torch.onnx.errors module torch.onnx.operators module torch.onnx.ops module torch.onnx.symbolic_caffe2 module torch.onnx.symbolic_helper module torch.onnx.symbolic_opset10 module torch.onnx.symbolic_opset11 module torch.onnx.symbolic_opset12 module torch.onnx.symbolic_opset13 module torch.onnx.symbolic_opset14 module torch.onnx.symbolic_opset15 module torch.onnx.symbolic_opset16 module torch.onnx.symbolic_opset17 module torch.onnx.symbolic_opset18 module torch.onnx.symbolic_opset19 module torch.onnx.symbolic_opset20 module torch.onnx.symbolic_opset7 module torch.onnx.symbolic_opset8 module torch.onnx.symbolic_opset9 module torch.onnx.utils module torch.onnx.verification module torch.optim module torch.optim.adadelta module torch.optim.adagrad module torch.optim.adam module torch.optim.adamax module torch.optim.adamw module torch.optim.asgd module torch.optim.lbfgs module torch.optim.lr_scheduler module torch.optim.nadam module torch.optim.optimizer module torch.optim.radam module torch.optim.rmsprop module torch.optim.rprop module torch.optim.sgd module torch.optim.sparse_adam module torch.optim.swa_utils module torch.overrides module torch.package module torch.package.analyze module torch.package.analyze.find_first_use_of_broken_modules module torch.package.analyze.is_from_package module torch.package.analyze.trace_dependencies module torch.package.file_structure_representation module torch.package.find_file_dependencies module torch.package.glob_group module torch.package.importer module torch.package.package_exporter module torch.package.package_importer module torch.profiler module torch.profiler.itt module torch.profiler.profiler module torch.profiler.python_tracer module torch.quantization module torch.quantization.fake_quantize module torch.quantization.fuse_modules module torch.quantization.fuser_method_mappings module torch.quantization.fx module torch.quantization.fx.convert module torch.quantization.fx.fuse module torch.quantization.fx.fusion_patterns module torch.quantization.fx.graph_module module torch.quantization.fx.match_utils module torch.quantization.fx.pattern_utils module torch.quantization.fx.prepare module torch.quantization.fx.quantization_patterns module torch.quantization.fx.quantization_types module torch.quantization.fx.utils module torch.quantization.observer module torch.quantization.qconfig module torch.quantization.quant_type module torch.quantization.quantization_mappings module torch.quantization.quantize module torch.quantization.quantize_fx module torch.quantization.quantize_jit module torch.quantization.stubs module torch.quantization.utils module torch.quasirandom module torch.random module torch.return_types module torch.serialization module torch.signal module torch.signal.windows module torch.signal.windows.windows module torch.sparse module torch.sparse.semi_structured module torch.special module torch.storage module torch.testing module torch.torch_version module torch.types module torch.utils module torch.utils.backcompat module torch.utils.backend_registration module torch.utils.benchmark module torch.utils.benchmark.examples module torch.utils.benchmark.examples.compare module torch.utils.benchmark.examples.fuzzer module torch.utils.benchmark.examples.op_benchmark module torch.utils.benchmark.examples.simple_timeit module torch.utils.benchmark.examples.spectral_ops_fuzz_test module torch.utils.benchmark.op_fuzzers module torch.utils.benchmark.op_fuzzers.binary module torch.utils.benchmark.op_fuzzers.sparse_binary module torch.utils.benchmark.op_fuzzers.sparse_unary module torch.utils.benchmark.op_fuzzers.spectral module torch.utils.benchmark.op_fuzzers.unary module torch.utils.benchmark.utils module torch.utils.benchmark.utils.common module torch.utils.benchmark.utils.compare module torch.utils.benchmark.utils.compile module torch.utils.benchmark.utils.cpp_jit module torch.utils.benchmark.utils.fuzzer module torch.utils.benchmark.utils.sparse_fuzzer module torch.utils.benchmark.utils.timer module torch.utils.benchmark.utils.valgrind_wrapper module torch.utils.benchmark.utils.valgrind_wrapper.timer_interface module torch.utils.bottleneck module torch.utils.bundled_inputs module torch.utils.checkpoint module torch.utils.collect_env module torch.utils.cpp_backtrace module torch.utils.cpp_extension module torch.utils.data module torch.utils.data.backward_compatibility module torch.utils.data.dataloader module torch.utils.data.datapipes module torch.utils.data.datapipes.dataframe module torch.utils.data.datapipes.dataframe.dataframe_wrapper module torch.utils.data.datapipes.dataframe.dataframes module torch.utils.data.datapipes.dataframe.datapipes module torch.utils.data.datapipes.dataframe.structures module torch.utils.data.datapipes.datapipe module torch.utils.data.datapipes.gen_pyi module torch.utils.data.datapipes.iter module torch.utils.data.datapipes.iter.callable module torch.utils.data.datapipes.iter.combinatorics module torch.utils.data.datapipes.iter.combining module torch.utils.data.datapipes.iter.filelister module torch.utils.data.datapipes.iter.fileopener module torch.utils.data.datapipes.iter.grouping module torch.utils.data.datapipes.iter.routeddecoder module torch.utils.data.datapipes.iter.selecting module torch.utils.data.datapipes.iter.sharding module torch.utils.data.datapipes.iter.streamreader module torch.utils.data.datapipes.iter.utils module torch.utils.data.datapipes.map module torch.utils.data.datapipes.map.callable module torch.utils.data.datapipes.map.combinatorics module torch.utils.data.datapipes.map.combining module torch.utils.data.datapipes.map.grouping module torch.utils.data.datapipes.map.utils module torch.utils.data.datapipes.utils module torch.utils.data.datapipes.utils.common module torch.utils.data.datapipes.utils.decoder module torch.utils.data.datapipes.utils.snapshot module torch.utils.data.dataset module torch.utils.data.distributed module torch.utils.data.graph module torch.utils.data.graph_settings module torch.utils.data.sampler module torch.utils.deterministic module torch.utils.dlpack module torch.utils.file_baton module torch.utils.flop_counter module torch.utils.hipify module torch.utils.hipify.constants module torch.utils.hipify.cuda_to_hip_mappings module torch.utils.hipify.hipify_python module torch.utils.hipify.version module torch.utils.hooks module torch.utils.jit module torch.utils.jit.log_extract module torch.utils.mkldnn module torch.utils.mobile_optimizer module torch.utils.model_dump module torch.utils.model_zoo module torch.utils.module_tracker module torch.utils.serialization module torch.utils.serialization.config module torch.utils.show_pickle module torch.utils.tensorboard module torch.utils.tensorboard.summary module torch.utils.tensorboard.writer module torch.utils.throughput_benchmark module torch.utils.viz module torch.utils.weak module torch.version module torch.xpu module torch.xpu.memory module torch.xpu.random module torch.xpu.streams module TORCH_COMPILE_JOB_ID torch_name() (torch.onnx.JitScalarType method) torch_save_to_dcp() (in module torch.distributed.checkpoint.format_utils) TorchAODType (class in torch.ao.quantization.observer) total_average() (torch.autograd.profiler.profile method) total_count (torch.distributions.multinomial.Multinomial attribute) trace() (in module torch) (in module torch.jit) (torch.fx.Tracer method) (torch.Tensor method) trace_module() (in module torch.jit) Tracer (class in torch.fx) train() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) Transform (class in torch.distributions.transforms) transform() (torch.fx.Transformer method) (torch.utils.benchmark.FunctionCounts method) transform_object() (torch.distributed.checkpoint.DefaultSavePlanner method) transform_tensor() (torch.distributed.checkpoint.DefaultLoadPlanner method) TransformedDistribution (class in torch.distributions.transformed_distribution) Transformer (class in torch.fx) (class in torch.nn) TransformerDecoder (class in torch.nn) TransformerDecoderLayer (class in torch.nn) TransformerEncoder (class in torch.nn) TransformerEncoderLayer (class in torch.nn) transpose() (in module torch) (torch.Tensor method) transpose_() (torch.Tensor method) trapezoid() (in module torch) trapz() (in module torch) triangular_solve() (in module torch) (torch.Tensor method) tril() (in module torch) (torch.Tensor method) tril_() (torch.Tensor method) tril_indices() (in module torch) trim_significant_figures() (torch.utils.benchmark.Compare method) triplet_margin_loss() (in module torch.nn.functional) triplet_margin_with_distance_loss() (in module torch.nn.functional) TripletMarginLoss (class in torch.nn) TripletMarginWithDistanceLoss (class in torch.nn) triton_op() (in module torch.library) triu() (in module torch) (torch.Tensor method) triu_() (torch.Tensor method) triu_indices() (in module torch) true_divide() (in module torch) (torch.Tensor method) true_divide_() (torch.Tensor method) trunc() (in module torch) (torch.Tensor method) trunc_() (torch.Tensor method) trunc_normal_() (in module torch.nn.init) tune_gemm_in_file() (in module torch.cuda.tunable) tuning_enable() (in module torch.cuda.tunable) tuning_is_enabled() (in module torch.cuda.tunable) type (torch.jit.Attribute attribute) type() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) type_as() (torch.Tensor method) TypedStorage (class in torch) U unbind() (in module torch) (torch.Tensor method) unbox() (torch.distributed.Work static method) underlying_store (torch.distributed.PrefixStore property) Unflatten (class in torch.nn) unflatten() (in module torch) (in module torch.export.unflatten) (torch.Tensor method) Unfold (class in torch.nn) unfold() (in module torch.nn.functional) (torch.Tensor method) Uniform (class in torch.distributions.uniform) uniform_() (in module torch.nn.init) (torch.Tensor method) UninitializedBuffer (class in torch.nn.parameter) UninitializedParameter (class in torch.nn.parameter) unique() (in module torch) (torch.Tensor method) unique_consecutive() (in module torch) (torch.Tensor method) unpack_dual() (in module torch.autograd.forward_ad) unpack_sequence() (in module torch.nn.utils.rnn) UnpackedDualTensor (class in torch.autograd.forward_ad) unpad_sequence() (in module torch.nn.utils.rnn) unravel_index() (in module torch) unregister_custom_op_symbolic() (in module torch.onnx) unregister_event_handler() (in module torch.monitor) unshard() (torch.distributed.fsdp.FSDPModule method) UnshardHandle (class in torch.distributed.fsdp) unsorted_indices (torch.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module torch) (torch.Tensor method) unsqueeze_() (torch.Tensor method) untyped() (torch.TypedStorage method) (torch.UntypedStorage method) untyped_storage() (torch.Tensor method) UntypedStorage (class in torch) unused() (in module torch.jit) update() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) update_arg() (torch.fx.Node method) update_bn() (in module torch.optim.swa_utils) update_bn_stats (class in torch.ao.nn.intrinsic.qat) update_kwarg() (torch.fx.Node method) update_parameters() (torch.optim.swa_utils.AveragedModel method) upsample (class in torch.ao.nn.quantized.functional) Upsample (class in torch.nn) upsample() (in module torch.nn.functional) upsample_bilinear (class in torch.ao.nn.quantized.functional) upsample_bilinear() (in module torch.nn.functional) upsample_nearest (class in torch.ao.nn.quantized.functional) upsample_nearest() (in module torch.nn.functional) UpsamplingBilinear2d (class in torch.nn) UpsamplingNearest2d (class in torch.nn) use_agent_store (torch.distributed.elastic.rendezvous.RendezvousHandler property) use_count() (torch.cuda.MemPool method) use_deterministic_algorithms() (in module torch) use_mem_pool (class in torch.cuda) utilization() (in module torch.cuda) V validate_checkpoint_id() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader class method) (torch.distributed.checkpoint.StorageReader class method) (torch.distributed.checkpoint.StorageWriter class method) value (torch.jit.Attribute attribute) value() (torch.futures.Future method) values() (torch.autograd.profiler_util.StringTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) (torch.Tensor method) vander() (in module torch) (in module torch.linalg) var() (in module torch) (torch.Tensor method) var_mean() (in module torch) variance (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) vdot() (in module torch) (torch.Tensor method) vecdot() (in module torch.linalg) vector_norm() (in module torch.linalg) vector_to_parameters() (in module torch.nn.utils) verbose (class in torch.backends.mkl) (class in torch.backends.mkldnn) VerificationInfo (class in torch.onnx.verification) VerificationOptions (class in torch.onnx.verification) verify() (in module torch.onnx.verification) (torch.export.dynamic_shapes.AdditionalInputs method) verify_aten_graph() (in module torch.onnx.verification) verify_ninja_availability() (in module torch.utils.cpp_extension) verify_onnx_program() (in module torch.onnx.verification) version() (in module torch.backends.cudnn) (in module torch.backends.cusparselt) vhp() (in module torch.autograd.functional) view() (torch.Tensor method) view_as() (torch.Tensor method) view_as_complex() (in module torch) view_as_real() (in module torch) visualize_sharding() (in module torch.distributed.tensor.debug) vjp() (in module torch.autograd.functional) (in module torch.func) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) vmap() (in module torch) (in module torch.func) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) VonMises (class in torch.distributions.von_mises) vsplit() (in module torch) (torch.Tensor method) vstack() (in module torch) W wait() (in module torch.jit) (torch.cuda.Event method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.fsdp.UnshardHandle method) (torch.distributed.Store method) (torch.distributed.Work method) (torch.Event method) (torch.futures.Future method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) wait_all() (in module torch.futures) wait_event() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) wait_stream() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) Weibull (class in torch.distributions.weibull) weight_norm() (in module torch.nn.utils) (in module torch.nn.utils.parametrizations) WeightedRandomSampler (class in torch.utils.data) where() (in module torch) (torch.Tensor method) Wishart (class in torch.distributions.wishart) with_args() (torch.ao.quantization.observer.AffineQuantizedObserverBase class method) (torch.ao.quantization.observer.ObserverBase class method) with_callable_args() (torch.ao.quantization.observer.ObserverBase class method) Work (class in torch.distributed) Worker (class in torch.distributed.elastic.agent.server) worker_main() (in module torch.distributed.elastic.control_plane) WorkerGroup (class in torch.distributed.elastic.agent.server) WorkerInfo (class in torch.distributed.rpc) WorkerSpec (class in torch.distributed.elastic.agent.server) WorkerState (class in torch.distributed.elastic.agent.server) wrap() (in module torch.fx) wrap_torch_function() (in module torch.overrides) wrap_triton() (in module torch.library) write_data() (torch.distributed.checkpoint.StorageWriter method) write_file() (in module torch.cuda.tunable) write_file_on_exit() (in module torch.cuda.tunable) WriteItem (class in torch.distributed.checkpoint.planner) X xavier_normal_() (in module torch.nn.init) xavier_uniform_() (in module torch.nn.init) xlog1py() (in module torch.special) xlogy() (in module torch) (in module torch.special) (torch.Tensor method) xlogy_() (torch.Tensor method) xpu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) Z zero_() (torch.Tensor method) zero_grad() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) ZeroPad1d (class in torch.nn) ZeroPad2d (class in torch.nn) ZeroPad3d (class in torch.nn) ZeroPointDomain (class in torch.ao.quantization.observer) ZeroRedundancyOptimizer (class in torch.distributed.optim) zeros() (in module torch) (in module torch.distributed.tensor) zeros_() (in module torch.nn.init) zeros_like() (in module torch) zeta() (in module torch.special)
Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z _ __create_chunk_list__() (torch.distributed.tensor.DTensor method) __getstate__() (torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) __init__() (torch.distributed.FileStore method) (torch.distributed.HashStore method) (torch.distributed.PrefixStore method) (torch.distributed.Store method) (torch.distributed.TCPStore method) (torch.fx.Graph method) (torch.fx.GraphModule method) (torch.monitor.Event method) (torch.monitor.Stat method) (torch.monitor.TensorboardEventHandler method) (torch.package.PackageExporter method) (torch.package.PackageImporter method) (torch.Tensor method) (torch.utils.tensorboard.writer.SummaryWriter method) __setstate__() (torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) _assert() (in module torch) _assign_worker_ranks() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _create_jit_fn() (in module torch.cuda.jiterator) _create_multi_output_jit_fn() (in module torch.cuda.jiterator) _dump_snapshot() (in module torch.cuda.memory) _exit_barrier() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _foreach_abs() (in module torch) _foreach_abs_() (in module torch) _foreach_acos() (in module torch) _foreach_acos_() (in module torch) _foreach_asin() (in module torch) _foreach_asin_() (in module torch) _foreach_atan() (in module torch) _foreach_atan_() (in module torch) _foreach_ceil() (in module torch) _foreach_ceil_() (in module torch) _foreach_cos() (in module torch) _foreach_cos_() (in module torch) _foreach_cosh() (in module torch) _foreach_cosh_() (in module torch) _foreach_erf() (in module torch) _foreach_erf_() (in module torch) _foreach_erfc() (in module torch) _foreach_erfc_() (in module torch) _foreach_exp() (in module torch) _foreach_exp_() (in module torch) _foreach_expm1() (in module torch) _foreach_expm1_() (in module torch) _foreach_floor() (in module torch) _foreach_floor_() (in module torch) _foreach_frac() (in module torch) _foreach_frac_() (in module torch) _foreach_lgamma() (in module torch) _foreach_lgamma_() (in module torch) _foreach_log() (in module torch) _foreach_log10() (in module torch) _foreach_log10_() (in module torch) _foreach_log1p() (in module torch) _foreach_log1p_() (in module torch) _foreach_log2() (in module torch) _foreach_log2_() (in module torch) _foreach_log_() (in module torch) _foreach_neg() (in module torch) _foreach_neg_() (in module torch) _foreach_reciprocal() (in module torch) _foreach_reciprocal_() (in module torch) _foreach_round() (in module torch) _foreach_round_() (in module torch) _foreach_sigmoid() (in module torch) _foreach_sigmoid_() (in module torch) _foreach_sin() (in module torch) _foreach_sin_() (in module torch) _foreach_sinh() (in module torch) _foreach_sinh_() (in module torch) _foreach_sqrt() (in module torch) _foreach_sqrt_() (in module torch) _foreach_tan() (in module torch) _foreach_tan_() (in module torch) _foreach_trunc() (in module torch) _foreach_trunc_() (in module torch) _foreach_zero_() (in module torch) _initialize_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _KinetoProfile (class in torch.profiler) _monitor_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _record_memory_history() (in module torch.cuda.memory) _rendezvous() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _restart_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _shutdown() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _snapshot() (in module torch.cuda.memory) _start_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) _stop_workers() (torch.distributed.elastic.agent.server.SimpleElasticAgent method) A abs() (in module torch) (torch.Tensor method) abs_() (torch.Tensor method) absolute() (in module torch) (torch.Tensor method) absolute_() (torch.Tensor method) AbsTransform (class in torch.distributions.transforms) acos() (in module torch) (torch.Tensor method) acos_() (torch.Tensor method) acosh() (in module torch) (torch.Tensor method) acosh_() (torch.Tensor method) acquire() (torch.distributed.elastic.timer.TimerClient method) active_pool() (torch.cuda.MemPoolContext static method) Adadelta (class in torch.optim) Adafactor (class in torch.optim) Adagrad (class in torch.optim) Adam (class in torch.optim) Adamax (class in torch.optim) AdamW (class in torch.optim) adapt() (torch.export.unflatten.FlatArgsAdapter method) adaptive_autorange() (torch.utils.benchmark.Timer method) adaptive_avg_pool1d() (in module torch.nn.functional) adaptive_avg_pool2d (class in torch.ao.nn.quantized.functional) adaptive_avg_pool2d() (in module torch.nn.functional) adaptive_avg_pool3d (class in torch.ao.nn.quantized.functional) adaptive_avg_pool3d() (in module torch.nn.functional) adaptive_max_pool1d() (in module torch.nn.functional) adaptive_max_pool2d() (in module torch.nn.functional) adaptive_max_pool3d() (in module torch.nn.functional) AdaptiveAvgPool1d (class in torch.nn) AdaptiveAvgPool2d (class in torch.nn) AdaptiveAvgPool3d (class in torch.nn) AdaptiveLogSoftmaxWithLoss (class in torch.nn) AdaptiveMaxPool1d (class in torch.nn) AdaptiveMaxPool2d (class in torch.nn) AdaptiveMaxPool3d (class in torch.nn) add() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) (torch.export.dynamic_shapes.AdditionalInputs method) (torch.fx.experimental.symbolic_shapes.DimConstraints method) (torch.monitor.Stat method) (torch.Tensor method) add_() (torch.Tensor method) add_audio() (torch.utils.tensorboard.writer.SummaryWriter method) add_custom_scalars() (torch.utils.tensorboard.writer.SummaryWriter method) add_dependency() (torch.package.PackageExporter method) add_done_callback() (torch.futures.Future method) add_dtype_config() (torch.ao.quantization.backend_config.BackendPatternConfig method) add_embedding() (torch.utils.tensorboard.writer.SummaryWriter method) add_equality() (torch.fx.experimental.symbolic_shapes.DimConstraints method) add_figure() (torch.utils.tensorboard.writer.SummaryWriter method) add_graph() (torch.utils.tensorboard.writer.SummaryWriter method) add_histogram() (torch.utils.tensorboard.writer.SummaryWriter method) add_hparams() (torch.utils.tensorboard.writer.SummaryWriter method) add_image() (torch.utils.tensorboard.writer.SummaryWriter method) add_images() (torch.utils.tensorboard.writer.SummaryWriter method) add_loggers() (in module torch.ao.ns._numeric_suite_fx) add_mesh() (torch.utils.tensorboard.writer.SummaryWriter method) add_metadata() (torch.profiler._KinetoProfile method) add_metadata_json() (torch.profiler._KinetoProfile method) add_module() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) add_param_group() (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) add_pr_curve() (torch.utils.tensorboard.writer.SummaryWriter method) add_pruning_method() (torch.nn.utils.prune.PruningContainer method) add_quant_dequant (class in torch.ao.quantization) add_relu() (torch.ao.ns._numeric_suite.Shadow method) add_safe_globals() (in module torch.serialization) add_scalar() (torch.ao.ns._numeric_suite.Shadow method) (torch.utils.tensorboard.writer.SummaryWriter method) add_scalars() (torch.utils.tensorboard.writer.SummaryWriter method) add_shadow_loggers() (in module torch.ao.ns._numeric_suite_fx) add_submodule() (torch.fx.GraphModule method) add_text() (torch.utils.tensorboard.writer.SummaryWriter method) add_var_to_val() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) add_video() (torch.utils.tensorboard.writer.SummaryWriter method) addbmm() (in module torch) (torch.Tensor method) addbmm_() (torch.Tensor method) addcdiv() (in module torch) (torch.Tensor method) addcdiv_() (torch.Tensor method) addcmul() (in module torch) (torch.Tensor method) addcmul_() (torch.Tensor method) AdditionalInputs (class in torch.export.dynamic_shapes) addmm() (in module torch) (in module torch.sparse) (torch.Tensor method) addmm_() (torch.Tensor method) addmv() (in module torch) (torch.Tensor method) addmv_() (torch.Tensor method) addr() (in module torch) (torch.Tensor method) addr_() (torch.Tensor method) adjoint() (in module torch) (torch.Tensor method) affine_grid() (in module torch.nn.functional) AffineQuantizedObserverBase (class in torch.ao.quantization.observer) AffineTransform (class in torch.distributions.transforms) Aggregation (class in torch.monitor) airy_ai() (in module torch.special) align_as() (torch.Tensor method) align_to() (torch.Tensor method) all() (in module torch) (torch.Tensor method) all_gather() (in module torch.distributed) all_gather_into_tensor() (in module torch.distributed) all_gather_object() (in module torch.distributed) all_input_nodes (torch.fx.Node property) all_paths() (torch.package.PackageExporter method) all_reduce() (in module torch.distributed) all_to_all() (in module torch.distributed) all_to_all_single() (in module torch.distributed) allclose() (in module torch) (torch.Tensor method) allocator (torch.cuda.MemPool property) allow_bf16_reduced_precision_reduction (in module torch.backends.cuda.matmul) allow_fp16_bf16_reduction_math_sdp() (in module torch.backends.cuda) allow_fp16_reduced_precision_reduction (in module torch.backends.cuda.matmul) allow_in_graph() (in module torch.compiler) allow_mutation_on_saved_tensors (class in torch.autograd.graph) allow_tf32 (in module torch.backends.cuda.matmul) (in module torch.backends.cudnn) allreduce_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) alpha_dropout() (in module torch.nn.functional) AlphaDropout (class in torch.nn) amax() (in module torch) (torch.Tensor method) amin() (in module torch) (torch.Tensor method) aminmax() (in module torch) (torch.Tensor method) and_masks() (in module torch.nn.attention.flex_attention) angle() (in module torch) (torch.Tensor method) annotate() (in module torch.jit) any() (in module torch) (torch.Tensor method) aoti_compile_and_package() (in module torch._inductor) aoti_load_package() (in module torch._inductor) append() (torch.distributed.Store method) (torch.fx.Node method) (torch.nn.ModuleList method) (torch.nn.ParameterList method) (torch.nn.Sequential method) apply() (torch.autograd.function.BackwardCFunction method) (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.nn.utils.prune.BasePruningMethod class method) (torch.nn.utils.prune.CustomFromMask class method) (torch.nn.utils.prune.Identity class method) (torch.nn.utils.prune.L1Unstructured class method) (torch.nn.utils.prune.LnStructured class method) (torch.nn.utils.prune.PruningContainer class method) (torch.nn.utils.prune.RandomStructured class method) (torch.nn.utils.prune.RandomUnstructured class method) (torch.optim.swa_utils.AveragedModel method) apply_() (torch.Tensor method) apply_jvp() (torch.autograd.function.BackwardCFunction method) apply_mask() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) apply_weights() (torch.onnx.ONNXProgram method) arange() (in module torch) arccos() (in module torch) (torch.Tensor method) arccos_() (torch.Tensor method) arccosh() (in module torch) (torch.Tensor method) arccosh_() (torch.Tensor method) arcsin() (in module torch) (torch.Tensor method) arcsin_() (torch.Tensor method) arcsinh() (in module torch) (torch.Tensor method) arcsinh_() (torch.Tensor method) arctan() (in module torch) (torch.Tensor method) arctan2() (in module torch) (torch.Tensor method) arctan2_() (torch.Tensor method) arctan_() (torch.Tensor method) arctanh() (in module torch) (torch.Tensor method) arctanh_() (torch.Tensor method) are_deterministic_algorithms_enabled() (in module torch) arg_constraints (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.beta.Beta attribute) (torch.distributions.binomial.Binomial attribute) (torch.distributions.categorical.Categorical attribute) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.chi2.Chi2 attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto attribute) (torch.distributions.geometric.Geometric attribute) (torch.distributions.gumbel.Gumbel attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent attribute) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.lkj_cholesky.LKJCholesky attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily attribute) (torch.distributions.multinomial.Multinomial attribute) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.negative_binomial.NegativeBinomial attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) (torch.distributions.pareto.Pareto attribute) (torch.distributions.poisson.Poisson attribute) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution attribute) (torch.distributions.uniform.Uniform attribute) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.weibull.Weibull attribute) (torch.distributions.wishart.Wishart attribute) argmax() (in module torch) (torch.Tensor method) argmin() (in module torch) (torch.Tensor method) args (torch.fx.Node property) argsort() (in module torch) (torch.Tensor method) argwhere() (in module torch) (torch.Tensor method) as_integer_ratio() (torch.SymFloat method) (torch.SymInt method) as_nested_tensor() (in module torch.nested) as_sparse_gradcheck() (in module torch.sparse) as_standardized() (torch.utils.benchmark.CallgrindStats method) as_strided() (in module torch) (torch.Tensor method) as_subclass() (torch.Tensor method) as_tensor() (in module torch) as_tuple() (torch.nn.attention.flex_attention.BlockMask method) asarray() (in module torch) asdict() (torch.onnx.verification.VerificationInfo method) ASGD (class in torch.optim) asin() (in module torch) (torch.Tensor method) asin_() (torch.Tensor method) asinh() (in module torch) (torch.Tensor method) asinh_() (torch.Tensor method) assert_allclose() (in module torch.testing) assert_close() (in module torch.testing) assume_constant_result() (in module torch.compiler) async_execution() (in module torch.distributed.rpc.functions) async_save() (in module torch.distributed.checkpoint.state_dict_saver) AsyncCheckpointerType (class in torch.distributed.checkpoint.state_dict_saver) AsyncStager (class in torch.distributed.checkpoint.staging) atan() (in module torch) (torch.Tensor method) atan2() (in module torch) (torch.Tensor method) atan2_() (torch.Tensor method) atan_() (torch.Tensor method) atanh() (in module torch) (torch.Tensor method) atanh_() (torch.Tensor method) atleast_1d() (in module torch) atleast_2d() (in module torch) atleast_3d() (in module torch) attach_out_of_memory_observer() (in module torch.mtia) Attribute (class in torch.jit) autocast (class in torch) (class in torch.cpu.amp) (class in torch.cuda.amp) AveragedModel (class in torch.optim.swa_utils) avg_pool1d() (in module torch.nn.functional) avg_pool2d (class in torch.ao.nn.quantized.functional) avg_pool2d() (in module torch.nn.functional) avg_pool3d (class in torch.ao.nn.quantized.functional) avg_pool3d() (in module torch.nn.functional) AvgPool1d (class in torch.nn) AvgPool2d (class in torch.nn) AvgPool3d (class in torch.nn) B Backend (class in torch.distributed) BackendConfig (class in torch.ao.quantization.backend_config) BackendPatternConfig (class in torch.ao.quantization.backend_config) BackendType (class in torch.distributed.rpc) backward() (in module torch.autograd) (in module torch.distributed.autograd) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction method) (torch.distributed.rpc.PyRRef method) (torch.Tensor method) backward_extended() (torch.autograd.function.NestedIOFunction method) BackwardCFunction (class in torch.autograd.function) BackwardPrefetch (class in torch.distributed.fsdp) baddbmm() (in module torch) (torch.Tensor method) baddbmm_() (torch.Tensor method) barrier() (in module torch.distributed) bartlett() (in module torch.signal.windows) bartlett_window() (in module torch) base_dist (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent attribute) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) BasePruningMethod (class in torch.nn.utils.prune) batch_isend_irecv() (in module torch.distributed) batch_norm() (in module torch.nn.functional) batch_shape (torch.distributions.distribution.Distribution property) batch_sizes (torch.nn.utils.rnn.PackedSequence attribute) batched_powerSGD_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) BatchNorm1d (class in torch.nn) BatchNorm2d (class in torch.ao.nn.quantized) (class in torch.nn) BatchNorm3d (class in torch.ao.nn.quantized) (class in torch.nn) BatchSampler (class in torch.utils.data) BCELoss (class in torch.nn) BCEWithLogitsLoss (class in torch.nn) benchmark (in module torch.backends.cudnn) benchmark_limit (in module torch.backends.cudnn) Bernoulli (class in torch.distributions.bernoulli) bernoulli() (in module torch) (torch.Tensor method) bernoulli_() (torch.Tensor method) bessel_j0() (in module torch.special) bessel_j1() (in module torch.special) Beta (class in torch.distributions.beta) bf16_compress_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) bf16_compress_wrapper() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) bfloat16() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BFloat16Storage (class in torch) Bilinear (class in torch.nn) bilinear() (in module torch.nn.functional) binary_cross_entropy() (in module torch.nn.functional) binary_cross_entropy_with_logits() (in module torch.nn.functional) bincount() (in module torch) (torch.Tensor method) bind_symbols() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) Binomial (class in torch.distributions.binomial) bitwise_and() (in module torch) (torch.Tensor method) bitwise_and_() (torch.Tensor method) bitwise_left_shift() (in module torch) (torch.Tensor method) bitwise_left_shift_() (torch.Tensor method) bitwise_not() (in module torch) (torch.Tensor method) bitwise_not_() (torch.Tensor method) bitwise_or() (in module torch) (torch.Tensor method) bitwise_or_() (torch.Tensor method) bitwise_right_shift() (in module torch) (torch.Tensor method) bitwise_right_shift_() (torch.Tensor method) bitwise_xor() (in module torch) (torch.Tensor method) bitwise_xor_() (torch.Tensor method) blackman() (in module torch.signal.windows) blackman_window() (in module torch) block_diag() (in module torch) BLOCK_SIZE (torch.nn.attention.flex_attention.BlockMask attribute) blocked_autorange() (torch.utils.benchmark.Timer method) BlockingAsyncStager (class in torch.distributed.checkpoint.staging) BlockMask (class in torch.nn.attention.flex_attention) bmm() (in module torch) (torch.Tensor method) BNReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) BNReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) bool() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BoolStorage (class in torch) bound_sympy() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) boxed() (torch.distributed.Work method) boxed_run() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) breakpoint() (in module torch.distributed) broadcast() (in module torch.cuda.comm) (in module torch.distributed) broadcast_coalesced() (in module torch.cuda.comm) broadcast_object_list() (in module torch.distributed) broadcast_shapes() (in module torch) broadcast_tensors() (in module torch) broadcast_to() (in module torch) (torch.Tensor method) BroadcastingTorchSaveReader (class in torch.distributed.checkpoint.format_utils) bucketize() (in module torch) Buffer (class in torch.nn.parameter) buffer() (in module torch.distributed.GradBucket) buffers() (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) build() (torch.distributed.elastic.rendezvous.api.RendezvousStoreInfo static method) build_stage() (in module torch.distributed.pipelining.stage) BuildExtension() (in module torch.utils.cpp_extension) byte() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) BytesIOWriteData (class in torch.distributed.checkpoint.planner) ByteStorage (class in torch) byteswap() (torch.UntypedStorage method) C C10dRendezvousBackend (class in torch.distributed.elastic.rendezvous.c10d_rendezvous_backend) cached() (in module torch.nn.utils.parametrize) caching_allocator_alloc() (in module torch.cuda) caching_allocator_delete() (in module torch.cuda) caching_allocator_enable() (in module torch.cuda.memory) calculate_gain() (in module torch.nn.init) calculate_qparams() (torch.ao.quantization.observer.AffineQuantizedObserverBase method) (torch.ao.quantization.observer.MinMaxObserver method) call_function() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) call_method() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) call_module() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Tracer method) (torch.fx.Transformer method) CallgrindStats (class in torch.utils.benchmark) CallMethodKey (class in torch.fx.experimental.symbolic_shapes) can_cast() (in module torch) can_device_access_peer() (in module torch.cuda) can_use_cudnn_attention() (in module torch.backends.cuda) can_use_efficient_attention() (in module torch.backends.cuda) can_use_flash_attention() (in module torch.backends.cuda) canonicalize_bool_expr() (in module torch.fx.experimental.symbolic_shapes) capture_begin() (torch.cuda.CUDAGraph method) capture_end() (torch.cuda.CUDAGraph method) cartesian_prod() (in module torch) cat (in module torch.distributions.constraints) cat() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) Categorical (class in torch.distributions.categorical) CatTransform (class in torch.distributions.transforms) Cauchy (class in torch.distributions.cauchy) cauchy_() (torch.Tensor method) causal_lower_right() (in module torch.nn.attention.bias) causal_upper_left() (in module torch.nn.attention.bias) CausalBias (class in torch.nn.attention.bias) CausalVariant (class in torch.nn.attention.bias) ccol_indices() (torch.Tensor method) cdf() (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.laplace.Laplace method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.normal.Normal method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) cdist() (in module torch) cdouble() (torch.Tensor method) ceil() (in module torch) (torch.Tensor method) ceil_() (torch.Tensor method) celu (class in torch.ao.nn.quantized.functional) CELU (class in torch.nn) celu() (in module torch.nn.functional) cfloat() (torch.Tensor method) chain_matmul() (in module torch) ChainDataset (class in torch.utils.data) ChainedScheduler (class in torch.optim.lr_scheduler) chalf() (torch.Tensor method) change_current_allocator() (in module torch.cuda) ChannelShuffle (class in torch.nn) char() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CharStorage (class in torch) check() (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) (torch.distributions.constraints.Constraint method) check_consistent() (in module torch.fx.experimental.symbolic_shapes) check_equal() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) check_export_model_diff (class in torch.onnx.verification) check_is_root() (torch.distributed.fsdp.FullyShardedDataParallel method) check_sparse_tensor_invariants (class in torch.sparse) checkpoint() (in module torch.utils.checkpoint) checkpoint_id (torch.distributed.checkpoint.FileSystemReader property) checkpoint_sequential() (in module torch.utils.checkpoint) CheckpointPolicy (class in torch.utils.checkpoint) Chi2 (class in torch.distributions.chi2) ChildFailedError (class in torch.distributed.elastic.multiprocessing.errors) children() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) cholesky() (in module torch) (in module torch.linalg) (torch.Tensor method) cholesky_ex() (in module torch.linalg) cholesky_inverse() (in module torch) (torch.Tensor method) cholesky_solve() (in module torch) (torch.Tensor method) chunk() (in module torch) (torch.Tensor method) CircularPad1d (class in torch.nn) CircularPad2d (class in torch.nn) CircularPad3d (class in torch.nn) clamp (class in torch.ao.nn.quantized.functional) clamp() (in module torch) (torch.Tensor method) clamp_() (torch.Tensor method) cleanup() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) clear() (in module torch.backends.cuda.cufft_plan_cache) (torch.autograd.profiler_util.StringTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) clear_safe_globals() (in module torch.serialization) clear_timers() (torch.distributed.elastic.timer.TimerServer method) clip() (in module torch) (torch.Tensor method) clip_() (torch.Tensor method) clip_grad_norm() (in module torch.nn.utils) clip_grad_norm_() (in module torch.nn.utils) (torch.distributed.fsdp.FullyShardedDataParallel method) clip_grad_value_() (in module torch.nn.utils) clip_grads_with_norm_() (in module torch.nn.utils) clock_rate() (in module torch.cuda) clone() (in module torch) (torch.autograd.grad_mode.inference_mode method) (torch.autograd.grad_mode.set_grad_enabled method) (torch.autograd.grad_mode.set_multithreading_enabled method) (torch.distributed.Store method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) clone_state() (torch.Generator method) close (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) close() (torch.package.PackageExporter method) (torch.utils.tensorboard.writer.SummaryWriter method) cls_to_become (torch.nn.LazyBatchNorm1d attribute) (torch.nn.LazyBatchNorm2d attribute) (torch.nn.LazyBatchNorm3d attribute) (torch.nn.LazyConv1d attribute) (torch.nn.LazyConv2d attribute) (torch.nn.LazyConv3d attribute) (torch.nn.LazyConvTranspose1d attribute) (torch.nn.LazyConvTranspose2d attribute) (torch.nn.LazyConvTranspose3d attribute) (torch.nn.LazyInstanceNorm1d attribute) (torch.nn.LazyInstanceNorm2d attribute) (torch.nn.LazyInstanceNorm3d attribute) (torch.nn.LazyLinear attribute) (torch.nn.parameter.UninitializedParameter attribute) coalesce() (torch.Tensor method) code (torch.fx.GraphModule property) (torch.jit.ScriptModule property) code_with_constants (torch.jit.ScriptModule property) col_indices() (torch.Tensor method) collate() (in module torch.utils.data._utils.collate) collect_all() (in module torch.futures) collect_callgrind() (torch.utils.benchmark.Timer method) colorize() (torch.utils.benchmark.Compare method) column_stack() (in module torch) ColwiseParallel (class in torch.distributed.tensor.parallel) combinations() (in module torch) CommDebugMode (class in torch.distributed.tensor.debug) commit_tensor() (torch.distributed.checkpoint.LoadPlanner method) Compare (class in torch.utils.benchmark) compare_model_outputs() (in module torch.ao.ns._numeric_suite) compare_model_stub() (in module torch.ao.ns._numeric_suite) compare_results (class in torch.ao.quantization) compare_set() (torch.distributed.Store method) compare_weights() (in module torch.ao.ns._numeric_suite) compile() (in module torch) (in module torch.compiler) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) compile_shader() (in module torch.mps) compiled_with_cxx11_abi() (in module torch) complex() (in module torch) complex_double() (torch.TypedStorage method) (torch.UntypedStorage method) complex_float() (torch.TypedStorage method) (torch.UntypedStorage method) ComplexDoubleStorage (class in torch) ComplexFloatStorage (class in torch) component_distribution (torch.distributions.mixture_same_family.MixtureSameFamily property) ComposeTransform (class in torch.distributions.transforms) compute_cosine_similarity() (in module torch.ao.ns.fx.utils) compute_mask() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) compute_normalized_l2_error() (in module torch.ao.ns.fx.utils) compute_sqnr() (in module torch.ao.ns.fx.utils) compute_unbacked_bindings() (in module torch.fx.experimental.symbolic_shapes) compute_values() (torch.onnx.ONNXProgram method) concat() (in module torch) ConcatDataset (class in torch.utils.data) concatenate() (in module torch) concentration (torch.distributions.inverse_gamma.InverseGamma property) concentration0 (torch.distributions.beta.Beta property) concentration1 (torch.distributions.beta.Beta property) cond() (in module torch) (in module torch._higher_order_ops.cond) (in module torch.linalg) configs (torch.ao.quantization.backend_config.BackendConfig property) configure() (in module torch.distributed.elastic.metrics) (in module torch.distributed.elastic.timer) confirmed_by_owner() (torch.distributed.rpc.PyRRef method) conj() (in module torch) (torch.Tensor method) conj_physical() (in module torch) (torch.Tensor method) conj_physical_() (torch.Tensor method) conjugate() (torch.SymFloat method) ConsoleMetricHandler (class in torch.distributed.elastic.metrics.api) consolidate_state_dict() (torch.distributed.optim.ZeroRedundancyOptimizer method) constant_() (in module torch.nn.init) ConstantLR (class in torch.optim.lr_scheduler) ConstantPad1d (class in torch.nn) ConstantPad2d (class in torch.nn) ConstantPad3d (class in torch.nn) constrain_range() (in module torch.fx.experimental.symbolic_shapes) constrain_unify() (in module torch.fx.experimental.symbolic_shapes) Constraint (class in torch.distributions.constraints) (in module torch.export) ConstraintRegistry (class in torch.distributions.constraint_registry) construct_and_record_rdzv_event() (in module torch.distributed.elastic.events) context (class in torch.distributed.autograd) context_parallel() (in module torch.distributed.tensor.experimental) contiguous() (torch.Tensor method) ContinuousBernoulli (class in torch.distributions.continuous_bernoulli) Conv1d (class in torch.ao.nn.quantized) conv1d (class in torch.ao.nn.quantized.functional) Conv1d (class in torch.nn) conv1d() (in module torch.nn.functional) Conv2d (class in torch.ao.nn.qat) (class in torch.ao.nn.quantized) conv2d (class in torch.ao.nn.quantized.functional) Conv2d (class in torch.nn) conv2d() (in module torch.nn.functional) Conv3d (class in torch.ao.nn.qat) (class in torch.ao.nn.quantized) conv3d (class in torch.ao.nn.quantized.functional) Conv3d (class in torch.nn) conv3d() (in module torch.nn.functional) conv_transpose1d() (in module torch.nn.functional) conv_transpose2d() (in module torch.nn.functional) conv_transpose3d() (in module torch.nn.functional) ConvBn1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBn2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBn3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) ConvBnReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) convert (class in torch.ao.quantization) convert() (torch.ao.quantization.observer.AffineQuantizedObserverBase method) convert_conv2d_weight_memory_format() (in module torch.nn.utils) convert_conv3d_weight_memory_format() (in module torch.nn.utils) convert_fx (class in torch.ao.quantization.quantize_fx) convert_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) convert_sync_batchnorm() (torch.nn.SyncBatchNorm class method) ConvertCustomConfig (class in torch.ao.quantization.fx.custom_config) ConvertIntKey (class in torch.fx.experimental.symbolic_shapes) ConvReLU1d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.quantized) ConvReLU2d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) ConvReLU3d (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) ConvTranspose1d (class in torch.ao.nn.quantized) (class in torch.nn) ConvTranspose2d (class in torch.ao.nn.quantized) (class in torch.nn) ConvTranspose3d (class in torch.ao.nn.quantized) (class in torch.nn) copy() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ParameterDict method) copy_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) copysign() (in module torch) (torch.Tensor method) copysign_() (torch.Tensor method) CorrCholeskyTransform (class in torch.distributions.transforms) corrcoef() (in module torch) (torch.Tensor method) cos() (in module torch) (torch.Tensor method) cos_() (torch.Tensor method) cosh() (in module torch) (torch.Tensor method) cosh_() (torch.Tensor method) cosine() (in module torch.signal.windows) cosine_embedding_loss() (in module torch.nn.functional) cosine_similarity() (in module torch.nn.functional) CosineAnnealingLR (class in torch.optim.lr_scheduler) CosineAnnealingWarmRestarts (class in torch.optim.lr_scheduler) CosineEmbeddingLoss (class in torch.nn) CosineSimilarity (class in torch.nn) count (torch.monitor.Stat property) count() (torch.autograd.forward_ad.UnpackedDualTensor method) (torch.autograd.profiler_util.Kernel method) (torch.jit.Attribute method) (torch.nn.utils.rnn.PackedSequence method) (torch.Size method) count_nonzero() (in module torch) (torch.Tensor method) counts() (torch.utils.benchmark.CallgrindStats method) cov() (in module torch) (torch.Tensor method) covariance_matrix (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) CppExtension() (in module torch.utils.cpp_extension) cpu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CPUOffload (class in torch.distributed.fsdp) CPUOffloadPolicy (class in torch.distributed.fsdp) create_arg() (torch.fx.Tracer method) create_args_for_root() (torch.fx.Tracer method) create_backend() (in module torch.distributed.elastic.rendezvous.c10d_rendezvous_backend) (in module torch.distributed.elastic.rendezvous.etcd_rendezvous_backend) create_block_mask() (in module torch.nn.attention.flex_attention) create_global_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) create_handler() (in module torch.distributed.elastic.rendezvous.dynamic_rendezvous) create_healthcheck_server() (in module torch.distributed.elastic.agent.server.health_check_server) create_local_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) create_mask() (in module torch.nn.attention.flex_attention) create_nested_block_mask() (in module torch.nn.attention.flex_attention) create_node() (torch.fx.Graph method) (torch.fx.Tracer method) create_proxy() (torch.fx.Tracer method) create_selective_checkpoint_contexts() (in module torch.utils.checkpoint) create_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symbolic_sizes_strides_storage_offset() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symboolnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symfloatnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_symintnode() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symbool() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symfloat() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unbacked_symint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unspecified_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) create_unspecified_symint_and_symbol() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) cross() (in module torch) (in module torch.linalg) (torch.Tensor method) cross_entropy() (in module torch.nn.functional) CrossEntropyLoss (class in torch.nn) crow_indices() (torch.Tensor method) ctc_loss() (in module torch.nn.functional) CTCLoss (class in torch.nn) cuda() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) CUDAExtension() (in module torch.utils.cpp_extension) CUDAGraph (class in torch.cuda) cudagraph_mark_step_begin() (in module torch.compiler) CUDAPluggableAllocator (class in torch.cuda) cudart() (in module torch.cuda) cudnn_sdp_enabled() (in module torch.backends.cuda) cufft_plan_cache (in module torch.backends.cuda) cummax() (in module torch) (torch.Tensor method) cummin() (in module torch) (torch.Tensor method) cumprod() (in module torch) (torch.Tensor method) cumprod_() (torch.Tensor method) cumsum() (in module torch) (torch.Tensor method) cumsum_() (torch.Tensor method) cumulative_trapezoid() (in module torch) CumulativeDistributionTransform (class in torch.distributions.transforms) current_accelerator() (in module torch.accelerator) current_allocated_memory() (in module torch.mps) current_blas_handle() (in module torch.cuda) current_device() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) current_device_idx() (in module torch.accelerator) current_device_index() (in module torch.accelerator) current_step() (torch.autograd.profiler.KinetoStepTracker class method) current_stream() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) custom_bwd() (in module torch.amp) (in module torch.cuda.amp) custom_from_mask() (in module torch.nn.utils.prune) custom_fwd() (in module torch.amp) (in module torch.cuda.amp) CUSTOM_KEY (in module torch.ao.quantization) custom_op() (in module torch.library) CustomDecompTable (class in torch.export.decomp_utils) CustomFromMask (class in torch.nn.utils.prune) CustomObjArgument (class in torch.export.graph_signature) CustomOpDef (class in torch._library.custom_ops) CyclicLR (class in torch.optim.lr_scheduler) D data (torch.monitor.Event property) (torch.nn.utils.rnn.PackedSequence attribute) data_parallel() (in module torch.nn.parallel) data_ptr() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) data_value_t (class in torch.monitor) DataLoader (class in torch.utils.data) DataParallel (class in torch.nn) Dataset (class in torch.utils.data) dcp_to_torch_save() (in module torch.distributed.checkpoint.format_utils) dde_suppressed() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) debug_dump() (torch.cuda.CUDAGraph method) debug_unwrap() (in module torch.func) default_activation_only_qconfig (in module torch.ao.quantization.qconfig) default_collate() (in module torch.utils.data) default_convert() (in module torch.utils.data) default_debug_observer (in module torch.ao.quantization.observer) default_debug_qconfig (in module torch.ao.quantization.qconfig) default_decompositions() (in module torch.export.exported_program) default_dynamic_qconfig (in module torch.ao.quantization.qconfig) default_dynamic_quant_observer (in module torch.ao.quantization.observer) default_eval_fn (class in torch.ao.quantization) default_factory (torch.autograd.profiler_util.StringTable attribute) default_fake_quant (in module torch.ao.quantization.fake_quantize) default_float_qparams_observer (in module torch.ao.quantization.observer) default_fused_act_fake_quant (in module torch.ao.quantization.fake_quantize) default_fused_per_channel_wt_fake_quant (in module torch.ao.quantization.fake_quantize) default_fused_wt_fake_quant (in module torch.ao.quantization.fake_quantize) default_generator (torch.torch attribute) default_histogram_fake_quant (in module torch.ao.quantization.fake_quantize) default_histogram_observer (in module torch.ao.quantization.observer) default_observer (in module torch.ao.quantization.observer) default_per_channel_qconfig (in module torch.ao.quantization.qconfig) default_per_channel_weight_fake_quant (in module torch.ao.quantization.fake_quantize) default_per_channel_weight_observer (in module torch.ao.quantization.observer) default_placeholder_observer (in module torch.ao.quantization.observer) default_qat_qconfig (in module torch.ao.quantization.qconfig) default_qat_qconfig_v2 (in module torch.ao.quantization.qconfig) default_qconfig (in module torch.ao.quantization.qconfig) default_stream() (in module torch.cuda) (in module torch.mtia) default_weight_fake_quant (in module torch.ao.quantization.fake_quantize) default_weight_observer (in module torch.ao.quantization.observer) default_weight_only_qconfig (in module torch.ao.quantization.qconfig) DefaultLoadPlanner (class in torch.distributed.checkpoint) DefaultLogsSpecs (class in torch.distributed.elastic.multiprocessing.api) DefaultSavePlanner (class in torch.distributed.checkpoint) defer_runtime_assert() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) DeferredMtiaCallError define() (in module torch.library) (torch.library.Library method) deg2rad() (in module torch) (torch.Tensor method) delete_all_unused_submodules() (torch.fx.GraphModule method) delete_key() (torch.distributed.Store method) delete_submodule() (torch.fx.GraphModule method) delta() (torch.utils.benchmark.CallgrindStats method) denied_modules() (torch.package.PackageExporter method) denoise() (torch.utils.benchmark.FunctionCounts method) dense_dim() (torch.Tensor method) deny() (torch.package.PackageExporter method) dependency_graph_string() (torch.package.PackageExporter method) dependent_property (in module torch.distributions.constraints) dequantize() (in module torch) (torch.ao.nn.quantizable.MultiheadAttention method) (torch.Tensor method) DeQuantStub (class in torch.ao.quantization) deregister_handle() (torch.cuda.gds.GdsFile method) deserialize_symexpr() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) det() (in module torch) (in module torch.linalg) (torch.Tensor method) detach() (torch.Tensor method) detach_() (torch.Tensor method) detect_anomaly (class in torch.autograd) deterministic (in module torch.backends.cudnn) device (class in torch) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) (torch.autograd.profiler_util.Kernel attribute) (torch.Generator attribute) (torch.Tensor attribute) (torch.TypedStorage property) (torch.UntypedStorage attribute) device_count() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) device_index (class in torch.accelerator) device_maps (torch.distributed.rpc.TensorPipeRpcBackendOptions property) device_memory_used() (in module torch.cuda) device_mesh (torch.distributed.tensor.DTensor property) device_of (class in torch.cuda) (class in torch.xpu) DeviceMesh (class in torch.distributed.device_mesh) devices (torch.distributed.rpc.TensorPipeRpcBackendOptions property) df (torch.distributions.chi2.Chi2 property) diag() (in module torch) (torch.Tensor method) diag_embed() (in module torch) (torch.Tensor method) diagflat() (in module torch) (torch.Tensor method) diagonal() (in module torch) (in module torch.linalg) (torch.Tensor method) diagonal_scatter() (in module torch) (torch.Tensor method) diff() (in module torch) (torch.Tensor method) digamma() (in module torch) (in module torch.special) (torch.Tensor method) digamma_() (torch.Tensor method) Dim (class in torch.export.dynamic_shapes) dim (torch.distributed.tensor.placement_types.Shard attribute) dim() (torch.Tensor method) dim_order() (torch.Tensor method) DimConstraints (class in torch.fx.experimental.symbolic_shapes) DimDynamic (class in torch.fx.experimental.symbolic_shapes) dims() (in module torch.export) dirac_() (in module torch.nn.init) Directory (class in torch.package) Dirichlet (class in torch.distributions.dirichlet) disable() (in module torch.compiler) (torch.sparse.check_sparse_tensor_invariants static method) disable_fake_quant (class in torch.ao.quantization.fake_quantize) disable_observer (class in torch.ao.quantization.fake_quantize) disable_saved_tensors_hooks (class in torch.autograd.graph) dist() (in module torch) (torch.Tensor method) DistBackendError (class in torch.distributed) DistError (class in torch.distributed) DistNetworkError (class in torch.distributed) distribute_module() (in module torch.distributed.tensor) distribute_tensor() (in module torch.distributed.tensor) DistributedDataParallel (class in torch.nn.parallel) DistributedOptimizer (class in torch.distributed.optim) DistributedSampler (class in torch.utils.data.distributed) Distribution (class in torch.distributions.distribution) DistStoreError (class in torch.distributed) div() (in module torch) (torch.Tensor method) div_() (torch.Tensor method) divide() (in module torch) (torch.Tensor method) divide_() (torch.Tensor method) DivideByKey (class in torch.fx.experimental.symbolic_shapes) done() (torch.futures.Future method) dot() (in module torch) (torch.Tensor method) double() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) DoubleStorage (class in torch) download_url_to_file() (in module torch.hub) draw() (torch.quasirandom.SobolEngine method) draw_base2() (torch.quasirandom.SobolEngine method) driver_allocated_memory() (in module torch.mps) Dropout (class in torch.nn) dropout() (in module torch.nn.functional) Dropout1d (class in torch.nn) dropout1d() (in module torch.nn.functional) Dropout2d (class in torch.nn) dropout2d() (in module torch.nn.functional) Dropout3d (class in torch.nn) dropout3d() (in module torch.nn.functional) dsplit() (in module torch) (torch.Tensor method) dstack() (in module torch) DTensor (class in torch.distributed.tensor) dtype (class in torch) (torch.BFloat16Storage attribute) (torch.BoolStorage attribute) (torch.ByteStorage attribute) (torch.CharStorage attribute) (torch.ComplexDoubleStorage attribute) (torch.ComplexFloatStorage attribute) (torch.DoubleStorage attribute) (torch.FloatStorage attribute) (torch.HalfStorage attribute) (torch.IntStorage attribute) (torch.LongStorage attribute) (torch.QInt32Storage attribute) (torch.QInt8Storage attribute) (torch.QUInt2x4Storage attribute) (torch.QUInt4x2Storage attribute) (torch.QUInt8Storage attribute) (torch.ShortStorage attribute) (torch.TypedStorage attribute) dtype() (torch.onnx.JitScalarType method) DTypeConfig (class in torch.ao.quantization.backend_config) DTypeWithConstraints (class in torch.ao.quantization.backend_config) dual_level (class in torch.autograd.forward_ad) duration (torch.autograd.profiler_util.Kernel attribute) dynamic_shapes() (torch.export.dynamic_shapes.AdditionalInputs method) (torch.export.dynamic_shapes.ShapesCollection method) DynamicMetaLoadPlanner (class in torch.distributed.checkpoint.format_utils) DynamicRendezvousHandler (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) dynamo_export() (in module torch.onnx) E eig() (in module torch.linalg) eigh() (in module torch.linalg) eigvals() (in module torch.linalg) eigvalsh() (in module torch.linalg) einsum() (in module torch) elapsed_time() (torch.cuda.Event method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) elapsed_us() (torch.autograd.profiler_util.Interval method) ElasticAgent (class in torch.distributed.elastic.agent.server) element_size() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) eliminate_dead_code() (torch.fx.Graph method) ELU (class in torch.ao.nn.quantized) elu (class in torch.ao.nn.quantized.functional) ELU (class in torch.nn) elu() (in module torch.nn.functional) elu_() (in module torch.nn.functional) Embedding (class in torch.ao.nn.quantized) (class in torch.nn) embedding() (in module torch.nn.functional) embedding_bag() (in module torch.nn.functional) EmbeddingBag (class in torch.ao.nn.quantized) (class in torch.nn) emit_itt (class in torch.autograd.profiler) emit_nvtx (class in torch.autograd.profiler) empty() (in module torch) (in module torch.distributed.tensor) empty_cache() (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) empty_like() (in module torch) empty_strided() (in module torch) EmptyMatchError (class in torch.package) enable() (in module torch.cuda.tunable) (torch.sparse.check_sparse_tensor_invariants static method) enable_cuda_sanitizer() (in module torch.cuda._sanitizer) enable_cudnn_sdp() (in module torch.backends.cuda) enable_debug_mode() (torch.cuda.CUDAGraph method) enable_fake_mode() (in module torch.onnx) enable_fake_quant (class in torch.ao.quantization.fake_quantize) enable_flash_sdp() (in module torch.backends.cuda) enable_grad (class in torch) enable_math_sdp() (in module torch.backends.cuda) enable_mem_efficient_sdp() (in module torch.backends.cuda) enable_observer (class in torch.ao.quantization.fake_quantize) enable_onednn_fusion() (in module torch.jit) enabled (in module torch.backends.cudnn) (in module torch.backends.opt_einsum) EnforceUnique (class in torch.autograd.profiler) enter_dual_level() (in module torch.autograd.forward_ad) entr() (in module torch.special) entropy() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exp_family.ExponentialFamily method) (torch.distributions.exponential.Exponential method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.inverse_gamma.InverseGamma method) (torch.distributions.kumaraswamy.Kumaraswamy method) (torch.distributions.laplace.Laplace method) (torch.distributions.log_normal.LogNormal method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.pareto.Pareto method) (torch.distributions.studentT.StudentT method) (torch.distributions.uniform.Uniform method) (torch.distributions.weibull.Weibull method) (torch.distributions.wishart.Wishart method) enumerate_support() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.distribution.Distribution method) (torch.distributions.independent.Independent method) (torch.distributions.one_hot_categorical.OneHotCategorical method) environment variable PYTORCH_JIT TORCH_COMPILE_JOB_ID eq() (in module torch) (torch.Tensor method) eq_() (torch.Tensor method) equal() (in module torch) (torch.Tensor method) EqualityConstraint (class in torch.fx.experimental.symbolic_shapes) erase_node() (torch.fx.Graph method) erase_step_count() (torch.autograd.profiler.KinetoStepTracker class method) erf() (in module torch) (in module torch.special) (torch.Tensor method) erf_() (torch.Tensor method) erfc() (in module torch) (in module torch.special) (torch.Tensor method) erfc_() (torch.Tensor method) erfcx() (in module torch.special) erfinv() (in module torch) (in module torch.special) (torch.Tensor method) erfinv_() (torch.Tensor method) ErrorHandler (class in torch.distributed.elastic.multiprocessing.errors) EtcdRendezvousBackend (class in torch.distributed.elastic.rendezvous.etcd_rendezvous_backend) EtcdRendezvousHandler (class in torch.distributed.elastic.rendezvous.etcd_rendezvous) EtcdServer (class in torch.distributed.elastic.rendezvous.etcd_server) EtcdStore (class in torch.distributed.elastic.rendezvous.etcd_store) eval() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) evaluate_guards_expression() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_guards_for_args() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_sym_node() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) evaluate_symexpr() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) Event (class in torch) (class in torch.cuda) (class in torch.distributed.elastic.events.api) (class in torch.monitor) (class in torch.mps.event) (class in torch.mtia) (class in torch.xpu) event_shape (torch.distributions.distribution.Distribution property) EventHandlerHandle (class in torch.monitor) EventMetadataValue (in module torch.distributed.elastic.events.api) events() (torch.profiler._KinetoProfile method) EventSource (class in torch.distributed.elastic.events.api) exception() (torch.distributed.Work method) exit_dual_level() (in module torch.autograd.forward_ad) exp() (in module torch) (torch.Tensor method) exp2() (in module torch) (in module torch.special) exp_() (torch.Tensor method) expand() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.chi2.Chi2 method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.inverse_gamma.InverseGamma method) (torch.distributions.kumaraswamy.Kumaraswamy method) (torch.distributions.laplace.Laplace method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.log_normal.LogNormal method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.pareto.Pareto method) (torch.distributions.poisson.Poisson method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli method) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.von_mises.VonMises method) (torch.distributions.weibull.Weibull method) (torch.distributions.wishart.Wishart method) (torch.Tensor method) expand_as() (torch.Tensor method) expires() (in module torch.distributed.elastic.timer) expit() (in module torch.special) expm1() (in module torch) (in module torch.special) (torch.Tensor method) expm1_() (torch.Tensor method) Exponential (class in torch.distributions.exponential) exponential() (in module torch.signal.windows) exponential_() (torch.Tensor method) ExponentialFamily (class in torch.distributions.exp_family) ExponentialLR (class in torch.optim.lr_scheduler) export() (in module torch.export) (in module torch.jit) (in module torch.onnx) export_chrome_trace() (torch.autograd.profiler.profile method) (torch.profiler._KinetoProfile method) export_memory_timeline() (torch.profiler._KinetoProfile method) export_stacks() (torch.profiler._KinetoProfile method) ExportBackwardSignature (class in torch.export) ExportedProgram (class in torch.export) ExportGraphSignature (class in torch.export) (class in torch.export.graph_signature) ExportOptions (class in torch.onnx) ExpTransform (class in torch.distributions.transforms) extend() (torch.nn.ModuleList method) (torch.nn.ParameterList method) (torch.nn.Sequential method) extend_logger_results_with_comparison() (in module torch.ao.ns._numeric_suite_fx) extend_results() (torch.utils.benchmark.Compare method) extern() (torch.package.PackageExporter method) ExternalStream (class in torch.cuda) externed_modules() (torch.package.PackageExporter method) extra_repr() (torch.ao.quantization.fake_quantize.FixedQParamsFakeQuantize method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.nn.modules.normalization.RMSNorm method) (torch.nn.RMSNorm method) (torch.optim.swa_utils.AveragedModel method) extract_logger_info() (in module torch.ao.ns._numeric_suite_fx) extract_results_from_loggers (class in torch.ao.quantization) extract_results_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) extract_shadow_logger_info() (in module torch.ao.ns._numeric_suite_fx) extract_weights() (in module torch.ao.ns._numeric_suite_fx) eye() (in module torch) eye_() (in module torch.nn.init) F fake_quantize_per_channel_affine() (in module torch) fake_quantize_per_tensor_affine() (in module torch) FakeQuantize (class in torch.ao.quantization.fake_quantize) FakeQuantizeBase (class in torch.ao.quantization.fake_quantize) fallback() (torch.library.Library method) fallthrough_kernel() (in module torch.library) fast_forward() (torch.quasirandom.SobolEngine method) feature_alpha_dropout() (in module torch.nn.functional) FeatureAlphaDropout (class in torch.nn) fetch_args_kwargs_from_env() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) fetch_attr() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) fft() (in module torch.fft) fft2() (in module torch.fft) fftfreq() (in module torch.fft) fftn() (in module torch.fft) fftshift() (in module torch.fft) file_structure() (torch.package.PackageImporter method) filename (torch.TypedStorage property) (torch.UntypedStorage property) FileStore (class in torch.distributed) FileSystemReader (class in torch.distributed.checkpoint) FileSystemWriter (class in torch.distributed.checkpoint) FileTimerClient (class in torch.distributed.elastic.timer) FileTimerServer (class in torch.distributed.elastic.timer) fill_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) fill_diagonal_() (torch.Tensor method) fill_uninitialized_memory (in module torch.utils.deterministic) filter() (torch.utils.benchmark.FunctionCounts method) find_mismatch() (in module torch.onnx.verification) find_nodes() (torch.fx.Graph method) finish() (torch.distributed.checkpoint.StorageWriter method) finish_plan() (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) FisherSnedecor (class in torch.distributions.fishersnedecor) fix() (in module torch) (torch.Tensor method) fix_() (torch.Tensor method) FixedQParamsFakeQuantize (class in torch.ao.quantization.fake_quantize) flags() (in module torch.backends.nnpack) flash_sdp_enabled() (in module torch.backends.cuda) FlatArgsAdapter (class in torch.export.unflatten) Flatten (class in torch.nn) flatten() (in module torch) (torch.Tensor method) flatten_parameters() (torch.nn.RNNBase method) flatten_sharded_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) flex_attention() (in module torch.nn.attention.flex_attention) flip() (in module torch) (torch.Tensor method) fliplr() (in module torch) (torch.Tensor method) flipud() (in module torch) (torch.Tensor method) float() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) float16_dynamic_qconfig (in module torch.ao.quantization.qconfig) float16_static_qconfig (in module torch.ao.quantization.qconfig) float8_e4m3fn() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e4m3fnuz() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e5m2() (torch.TypedStorage method) (torch.UntypedStorage method) float8_e5m2fnuz() (torch.TypedStorage method) (torch.UntypedStorage method) float_power() (in module torch) (torch.Tensor method) float_power_() (torch.Tensor method) float_qparams_weight_only_qconfig (in module torch.ao.quantization.qconfig) FloatFunctional (class in torch.ao.nn.quantized) FloatStorage (class in torch) floor() (in module torch) (torch.Tensor method) floor_() (torch.Tensor method) floor_divide() (in module torch) (torch.Tensor method) floor_divide_() (torch.Tensor method) flush() (torch.utils.tensorboard.writer.SummaryWriter method) fmax() (in module torch) (torch.Tensor method) fmin() (in module torch) (torch.Tensor method) fmod() (in module torch) (torch.Tensor method) fmod_() (torch.Tensor method) Fold (class in torch.nn) fold() (in module torch.nn.functional) forced_specializations() (torch.fx.experimental.symbolic_shapes.DimConstraints method) fork() (in module torch.jit) fork_rng() (in module torch.random) format_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) format_node() (torch.fx.Node method) forward() (torch.ao.nn.quantizable.MultiheadAttention method) (torch.ao.ns._numeric_suite.Logger method) (torch.ao.ns._numeric_suite.OutputLogger method) (torch.ao.ns._numeric_suite.Shadow method) (torch.ao.ns._numeric_suite.ShadowLogger method) (torch.ao.ns._numeric_suite_fx.OutputComparisonLogger method) (torch.ao.ns._numeric_suite_fx.OutputLogger method) (torch.ao.quantization.observer.AffineQuantizedObserverBase method) (torch.ao.quantization.observer.MinMaxObserver method) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction method) (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.ConvTranspose2d method) (torch.nn.EmbeddingBag method) (torch.nn.Module method) (torch.nn.modules.normalization.RMSNorm method) (torch.nn.MultiheadAttention method) (torch.nn.RMSNorm method) (torch.nn.Transformer method) (torch.nn.TransformerDecoder method) (torch.nn.TransformerDecoderLayer method) (torch.nn.TransformerEncoder method) (torch.nn.TransformerEncoderLayer method) (torch.optim.swa_utils.AveragedModel method) forward_extended() (torch.autograd.function.NestedIOFunction method) forward_shape() (torch.distributions.transforms.Transform method) fp16_bf16_reduction_math_sdp_allowed() (in module torch.backends.cuda) fp16_compress_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) fp16_compress_wrapper() (in module torch.distributed.algorithms.ddp_comm_hooks.default_hooks) frac() (in module torch) (torch.Tensor method) frac_() (torch.Tensor method) fractional_max_pool2d() (in module torch.nn.functional) fractional_max_pool3d() (in module torch.nn.functional) FractionalMaxPool2d (class in torch.nn) FractionalMaxPool3d (class in torch.nn) freeze() (in module torch.jit) (torch.fx.experimental.symbolic_shapes.ShapeEnv method) freeze_bn_stats (class in torch.ao.nn.intrinsic.qat) freeze_runtime_asserts() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) frexp() (in module torch) (torch.Tensor method) from_backend() (torch.distributed.elastic.rendezvous.dynamic_rendezvous.DynamicRendezvousHandler class method) from_buffer() (torch.TypedStorage class method) (torch.UntypedStorage static method) from_dict() (torch.ao.quantization.backend_config.BackendConfig class method) (torch.ao.quantization.backend_config.BackendPatternConfig class method) (torch.ao.quantization.backend_config.DTypeConfig class method) (torch.ao.quantization.fx.custom_config.ConvertCustomConfig class method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig class method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig class method) (torch.ao.quantization.qconfig_mapping.QConfigMapping class method) from_dlpack() (in module torch) (in module torch.utils.dlpack) from_dtype() (torch.onnx.JitScalarType class method) from_file() (in module torch) (torch.TypedStorage class method) (torch.UntypedStorage static method) from_float() (torch.ao.nn.qat.Linear class method) (torch.ao.nn.quantized.Conv1d class method) (torch.ao.nn.quantized.Conv2d class method) (torch.ao.nn.quantized.Conv3d class method) (torch.ao.nn.quantized.dynamic.Linear class method) (torch.ao.nn.quantized.Embedding class method) (torch.ao.nn.quantized.EmbeddingBag class method) (torch.ao.nn.quantized.Linear class method) from_group() (torch.distributed.device_mesh.DeviceMesh static method) from_ipc_handle() (torch.cuda.Event class method) from_kv_blocks() (torch.nn.attention.flex_attention.BlockMask class method) from_local() (torch.distributed.tensor.DTensor static method) from_numpy() (in module torch) from_onnx_type() (torch.onnx.JitScalarType class method) from_pretrained() (torch.nn.Embedding class method) (torch.nn.EmbeddingBag class method) from_reference() (torch.ao.nn.quantized.dynamic.Linear class method) (torch.ao.nn.quantized.Linear class method) from_tensors() (torch.onnx.verification.VerificationInfo class method) from_value() (torch.onnx.JitScalarType class method) frombuffer() (in module torch) fromkeys() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) fsdp_modules() (torch.distributed.fsdp.FullyShardedDataParallel static method) FSDPModule (class in torch.distributed.fsdp) full() (in module torch) (in module torch.distributed.tensor) full_kv_indices (torch.nn.attention.flex_attention.BlockMask attribute) full_kv_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) full_like() (in module torch) full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) full_q_indices (torch.nn.attention.flex_attention.BlockMask attribute) full_q_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) full_tensor() (torch.distributed.tensor.DTensor method) FullOptimStateDictConfig (class in torch.distributed.fsdp) FullStateDictConfig (class in torch.distributed.fsdp) fully_shard() (in module torch.distributed.fsdp) FullyShardedDataParallel (class in torch.distributed.fsdp) Function (class in torch.autograd) functional_call() (in module torch.func) (in module torch.nn.utils.stateless) functionalize() (in module torch.func) FunctionCounts (class in torch.utils.benchmark) fuse_conv_bn_eval() (in module torch.nn.utils) fuse_conv_bn_weights() (in module torch.nn.utils) fuse_fx (class in torch.ao.quantization.quantize_fx) fuse_linear_bn_eval() (in module torch.nn.utils) fuse_linear_bn_weights() (in module torch.nn.utils) fuse_modules (class in torch.ao.quantization.fuse_modules) FuseCustomConfig (class in torch.ao.quantization.fx.custom_config) FusedMovingAvgObsFakeQuantize (class in torch.ao.quantization.fake_quantize) Future (class in torch.futures) FXFloatFunctional (class in torch.ao.nn.quantized) G Gamma (class in torch.distributions.gamma) gammainc() (in module torch.special) gammaincc() (in module torch.special) gammaln() (in module torch.special) gather() (in module torch) (in module torch.cuda.comm) (in module torch.distributed) (torch.Tensor method) gather_object() (in module torch.distributed) gaussian() (in module torch.signal.windows) gaussian_nll_loss() (in module torch.nn.functional) GaussianNLLLoss (class in torch.nn) gcd() (in module torch) (torch.Tensor method) gcd_() (torch.Tensor method) gds_deregister_buffer() (in module torch.cuda.gds) gds_register_buffer() (in module torch.cuda.gds) GdsFile (class in torch.cuda.gds) ge() (in module torch) (torch.Tensor method) ge_() (torch.Tensor method) GELU (class in torch.nn) gelu() (in module torch.nn.functional) general_cosine() (in module torch.signal.windows) general_hamming() (in module torch.signal.windows) GeneralizedPareto (class in torch.distributions.generalized_pareto) generate_comm_debug_tracing_table() (torch.distributed.tensor.debug.CommDebugMode method) generate_json_dump() (torch.distributed.tensor.debug.CommDebugMode method) generate_methods_for_privateuse1_backend() (in module torch.utils) generate_numeric_debug_handle (class in torch.ao.quantization) generate_square_subsequent_mask() (torch.nn.Transformer static method) Generator (class in torch) Geometric (class in torch.distributions.geometric) geometric_() (torch.Tensor method) geqrf() (in module torch) (torch.Tensor method) ger() (in module torch) (torch.Tensor method) get() (torch.autograd.profiler_util.StringTable method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.elastic.rendezvous.RendezvousParameters method) (torch.distributed.Store method) (torch.fx.experimental.symbolic_shapes.CallMethodKey method) (torch.fx.experimental.symbolic_shapes.ConvertIntKey method) (torch.fx.experimental.symbolic_shapes.DivideByKey method) (torch.fx.experimental.symbolic_shapes.InnerTensorKey method) (torch.monitor.Stat method) (torch.nn.ParameterDict method) get_all_groups() (torch.distributed.device_mesh.DeviceMesh method) get_all_sharing_strategies() (in module torch.multiprocessing) get_allocator_backend() (in module torch.cuda) get_arch_list() (in module torch.cuda) (in module torch.xpu) get_as_bool() (torch.distributed.elastic.rendezvous.RendezvousParameters method) get_as_int() (torch.distributed.elastic.rendezvous.RendezvousParameters method) get_attr() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) get_axioms() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_backend() (in module torch.distributed) (torch.distributed.elastic.rendezvous.RendezvousHandler method) get_block_size (class in torch.ao.quantization.observer) get_buffer() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_comm_counts() (torch.distributed.tensor.debug.CommDebugMode method) get_compiler_abi_compatibility_and_version() (in module torch.utils.cpp_extension) get_coordinate() (torch.distributed.device_mesh.DeviceMesh method) get_cpp_backtrace() (in module torch.utils) get_cpu_capability() (in module torch.backends.cpu) get_crc32_options() (in module torch.serialization) get_ctx() (in module torch.library) get_debug_state() (torch.jit.ScriptFunction method) get_default_device() (in module torch) get_default_dtype() (in module torch) get_default_load_endianness() (in module torch.serialization) get_default_mmap_options() (in module torch.serialization) get_default_qat_qconfig_mapping (class in torch.ao.quantization.qconfig_mapping) get_default_qconfig_mapping (class in torch.ao.quantization.qconfig_mapping) get_deterministic_debug_mode() (in module torch) get_device() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) get_device_capability() (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) get_device_module() (in module torch) get_device_name() (in module torch.cuda) (in module torch.xpu) get_device_properties() (in module torch.cuda) (in module torch.xpu) get_dir() (in module torch.hub) get_ema_multi_avg_fn() (in module torch.optim.swa_utils) get_entrypoint_name() (torch.distributed.elastic.agent.server.WorkerSpec method) get_expired_timers() (torch.distributed.elastic.timer.TimerServer method) get_extra_state() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_fastpath_enabled() (in module torch.backends.mha) get_filename() (in module torch.cuda.tunable) get_float32_matmul_precision() (in module torch) get_fresh_qualname() (torch.fx.Tracer method) get_future() (torch.distributed.Work method) get_future_result() (torch.distributed.Work method) get_gencode_flags() (in module torch.cuda) (in module torch.xpu) get_global_rank() (in module torch.distributed) get_gradient_edge() (in module torch.autograd.graph) get_gradients() (in module torch.distributed.autograd) get_group() (torch.distributed.device_mesh.DeviceMesh method) get_group_rank() (in module torch.distributed) get_ignored_functions() (in module torch.overrides) get_implications() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_last_lr() (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.swa_utils.SWALR method) get_local_rank() (torch.distributed.device_mesh.DeviceMesh method) get_logger_dict() (in module torch.ao.ns._numeric_suite) get_logging_handler() (in module torch.distributed.elastic.events) get_lr() (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.swa_utils.SWALR method) get_matching_activations() (in module torch.ao.ns._numeric_suite) get_max_tuning_duration() (in module torch.cuda.tunable) get_max_tuning_iterations() (in module torch.cuda.tunable) get_model_state_dict() (in module torch.distributed.checkpoint.state_dict) get_module_rref() (torch.distributed.nn.api.remote_module.RemoteModule method) get_nontrivial_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_num_interop_threads() (in module torch) get_num_threads() (in module torch) get_observer_state_dict (class in torch.ao.quantization.observer) get_opt_einsum() (in module torch.backends.opt_einsum) get_optimizer_state_dict() (in module torch.distributed.checkpoint.state_dict) get_overridable_functions() (in module torch.overrides) get_overwrite_module_params_on_conversion() (in module torch.__future__) get_parameter() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_parameter_info() (torch.distributed.tensor.debug.CommDebugMode method) get_per_process_memory_fraction() (in module torch.cuda) get_process_group_ranks() (in module torch.distributed) get_proxy_mode() (in module torch.fx.experimental.proxy_tensor) get_pruned_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) get_rank() (in module torch.distributed) (torch.distributed.device_mesh.DeviceMesh method) get_rdeps() (torch.package.PackageExporter method) get_replace_hook() (torch.export.graph_signature.ExportGraphSignature method) get_results() (in module torch.cuda.tunable) get_rng_state() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.random) (in module torch.xpu) get_rng_state_all() (in module torch.cuda) (in module torch.xpu) get_rotating_buffer_size() (in module torch.cuda.tunable) get_run_id() (torch.distributed.elastic.rendezvous.RendezvousHandler method) get_safe_globals() (in module torch.serialization) get_sharding_info() (torch.distributed.tensor.debug.CommDebugMode method) get_sharing_strategy() (in module torch.multiprocessing) get_state() (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend method) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend method) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend method) (torch.Generator method) get_state_dict() (in module torch.distributed.checkpoint.state_dict) get_state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) get_stream_from_external() (in module torch.cuda) (in module torch.xpu) get_submodule() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) get_subprocess_handler() (in module torch.distributed.elastic.multiprocessing.subprocess_handler.handlers) get_swap_module_params_on_conversion() (in module torch.__future__) get_sync_debug_mode() (in module torch.cuda) get_testing_overrides() (in module torch.overrides) get_total_counts() (torch.distributed.tensor.debug.CommDebugMode method) get_total_norm() (in module torch.nn.utils) get_trace_id() (torch.profiler.profile method) get_unique_id() (torch.package.PackageExporter method) get_unsafe_globals_in_checkpoint() (in module torch.serialization) get_validators() (in module torch.cuda.tunable) get_worker_group() (torch.distributed.elastic.agent.server.ElasticAgent method) get_worker_info() (in module torch.distributed.rpc) (in module torch.utils.data) get_world_size() (in module torch.distributed) getattr() (torch.fx.Tracer method) global_unstructured() (in module torch.nn.utils.prune) GLU (class in torch.nn) glu() (in module torch.nn.functional) grad (torch.Tensor attribute) grad() (in module torch.autograd) (in module torch.func) grad_and_value() (in module torch.func) GradBucket (class in torch.distributed) gradcheck() (in module torch.autograd.gradcheck) GradcheckError gradgradcheck() (in module torch.autograd.gradcheck) gradient() (in module torch) GradientEdge (class in torch.autograd.graph) gradients() (in module torch.distributed.GradBucket) GradScaler (class in torch.cpu.amp) (class in torch.cuda.amp) Granularity (class in torch.ao.quantization.observer) graph (class in torch.cuda) Graph (class in torch.fx) graph (torch.fx.GraphModule property) (torch.jit.ScriptModule property) graph_copy() (torch.fx.Graph method) graph_pool_handle() (in module torch.cuda) GraphInfo (class in torch.onnx.verification) GraphInfoPrettyPrinter (class in torch.onnx.verification) GraphModule (class in torch.fx) graphsafe_get_state() (torch.Generator method) graphsafe_set_state() (torch.Generator method) greater() (in module torch) (torch.Tensor method) greater_() (torch.Tensor method) greater_equal() (in module torch) (torch.Tensor method) greater_equal_() (torch.Tensor method) greater_than (in module torch.distributions.constraints) greater_than_eq (in module torch.distributions.constraints) grid_sample() (in module torch.nn.functional) group_norm() (in module torch.nn.functional) GroupNorm (class in torch.ao.nn.quantized) (class in torch.nn) GRU (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) GRUCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) gt() (in module torch) (torch.Tensor method) gt_() (torch.Tensor method) guard_or_false() (in module torch.fx.experimental.symbolic_shapes) guard_or_true() (in module torch.fx.experimental.symbolic_shapes) guard_size_oblivious() (in module torch.fx.experimental.symbolic_shapes) Gumbel (class in torch.distributions.gumbel) gumbel_softmax() (in module torch.nn.functional) H H (torch.Tensor attribute) half() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) half_open_interval (in module torch.distributions.constraints) HalfCauchy (class in torch.distributions.half_cauchy) HalfNormal (class in torch.distributions.half_normal) HalfStorage (class in torch) hamming() (in module torch.signal.windows) hamming_window() (in module torch) handle_sym_dispatch() (in module torch.fx.experimental.proxy_tensor) handle_torch_function() (in module torch.overrides) hann() (in module torch.signal.windows) hann_window() (in module torch) Hardshrink (class in torch.nn) hardshrink() (in module torch.nn.functional) (torch.Tensor method) hardsigmoid (class in torch.ao.nn.quantized.functional) Hardsigmoid (class in torch.nn) hardsigmoid() (in module torch.nn.functional) Hardswish (class in torch.ao.nn.quantized) hardswish (class in torch.ao.nn.quantized.functional) Hardswish (class in torch.nn) hardswish() (in module torch.nn.functional) hardtanh (class in torch.ao.nn.quantized.functional) Hardtanh (class in torch.nn) hardtanh() (in module torch.nn.functional) hardtanh_() (in module torch.nn.functional) has_enumerate_support (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.binomial.Binomial attribute) (torch.distributions.categorical.Categorical attribute) (torch.distributions.independent.Independent property) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) has_extended_api() (torch.distributed.Store method) has_file() (torch.package.Directory method) has_free_symbols() (in module torch.fx.experimental.symbolic_shapes) has_free_unbacked_symbols() (in module torch.fx.experimental.symbolic_shapes) has_rsample (torch.distributions.beta.Beta attribute) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily attribute) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution property) (torch.distributions.uniform.Uniform attribute) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.wishart.Wishart attribute) has_static_value() (in module torch.fx.experimental.symbolic_shapes) has_torch_function() (in module torch.overrides) has_uninitialized_params() (torch.nn.modules.lazy.LazyModuleMixin method) HashStore (class in torch.distributed) HealthCheckServer (class in torch.distributed.elastic.agent.server.health_check_server) heartbeat (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) heaviside() (in module torch) (torch.Tensor method) help() (in module torch.hub) hessian() (in module torch.autograd.functional) (in module torch.func) hex() (torch.SymFloat method) hfft() (in module torch.fft) hfft2() (in module torch.fft) hfftn() (in module torch.fft) highlight_warnings() (torch.utils.benchmark.Compare method) hinge_embedding_loss() (in module torch.nn.functional) HingeEmbeddingLoss (class in torch.nn) hint_int() (in module torch.fx.experimental.symbolic_shapes) histc() (in module torch) (torch.Tensor method) histogram() (in module torch) (torch.Tensor method) histogramdd() (in module torch) HistogramObserver (class in torch.ao.quantization.observer) host (torch.distributed.TCPStore property) host_memory_stats() (in module torch.cuda) householder_product() (in module torch.linalg) hpu() (torch.TypedStorage method) (torch.UntypedStorage method) hsplit() (in module torch) (torch.Tensor method) hspmm() (in module torch) hstack() (in module torch) huber_loss() (in module torch.nn.functional) HuberLoss (class in torch.nn) hvp() (in module torch.autograd.functional) hypot() (in module torch) (torch.Tensor method) hypot_() (torch.Tensor method) I i0() (in module torch) (in module torch.special) (torch.Tensor method) i0_() (torch.Tensor method) i0e() (in module torch.special) i1() (in module torch.special) i1e() (in module torch.special) icdf() (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.laplace.Laplace method) (torch.distributions.normal.Normal method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) id (torch.cuda.MemPool property) (torch.distributed.rpc.WorkerInfo property) id() (torch.package.PackageImporter method) Identity (class in torch.nn) (class in torch.nn.utils.prune) identity() (in module torch.nn.utils.prune) ifft() (in module torch.fft) ifft2() (in module torch.fft) ifftn() (in module torch.fft) ifftshift() (in module torch.fft) igamma() (in module torch) (torch.Tensor method) igamma_() (torch.Tensor method) igammac() (in module torch) (torch.Tensor method) igammac_() (torch.Tensor method) ignore() (in module torch.jit) ignore_fresh_unbacked_symbols() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) ihfft() (in module torch.fft) ihfft2() (in module torch.fft) ihfftn() (in module torch.fft) imag (torch.Tensor attribute) imag() (in module torch) impl() (in module torch.library) (torch.library.Library method) impl_abstract() (in module torch.library) import_module() (torch.package.PackageImporter method) in_interval() (torch.autograd.profiler_util.MemRecordsAcc method) include_paths() (in module torch.utils.cpp_extension) increment_step() (torch.autograd.profiler.KinetoStepTracker class method) increment_version() (in module torch.autograd.graph) Independent (class in torch.distributions.independent) independent (in module torch.distributions.constraints) IndependentTransform (class in torch.distributions.transforms) index() (in module torch.distributed.GradBucket) (torch.autograd.forward_ad.UnpackedDualTensor method) (torch.autograd.profiler_util.Kernel method) (torch.jit.Attribute method) (torch.nn.utils.rnn.PackedSequence method) (torch.Size method) index_add() (in module torch) (torch.Tensor method) index_add_() (torch.Tensor method) index_copy() (in module torch) (torch.Tensor method) index_copy_() (torch.Tensor method) index_fill() (torch.Tensor method) index_fill_() (torch.Tensor method) index_put() (torch.Tensor method) index_put_() (torch.Tensor method) index_reduce() (in module torch) (torch.Tensor method) index_reduce_() (torch.Tensor method) index_select() (in module torch) (torch.Tensor method) indices() (torch.Tensor method) infer_schema() (in module torch.library) inference_mode (class in torch.autograd.grad_mode) init() (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) init_device_mesh() (in module torch.distributed.device_mesh) init_method (torch.distributed.rpc.RpcBackendOptions property) (torch.distributed.rpc.TensorPipeRpcBackendOptions property) init_process_group() (in module torch.distributed) init_rpc() (in module torch.distributed.rpc) init_step_count() (torch.autograd.profiler.KinetoStepTracker class method) initial_seed() (in module torch) (in module torch.cuda) (in module torch.random) (in module torch.xpu) (torch.Generator method) initialize_inference_session() (torch.onnx.ONNXProgram method) initialize_parameters() (torch.nn.modules.lazy.LazyModuleMixin method) inlined_graph (torch.jit.ScriptModule property) inner() (in module torch) (torch.Tensor method) InnerTensorKey (class in torch.fx.experimental.symbolic_shapes) InplaceFunction (class in torch.autograd.function) INPUT_OUTPUT_NOT_OBSERVED (torch.ao.quantization.backend_config.ObservationType attribute) InputKind (class in torch.export.graph_signature) InputSpec (class in torch.export.graph_signature) insert() (torch.nn.ModuleList method) (torch.nn.Sequential method) insert_arg() (torch.fx.Node method) inserting_after() (torch.fx.Graph method) inserting_before() (torch.fx.Graph method) instance_norm() (in module torch.nn.functional) InstanceNorm1d (class in torch.ao.nn.quantized) (class in torch.nn) InstanceNorm2d (class in torch.ao.nn.quantized) (class in torch.nn) InstanceNorm3d (class in torch.ao.nn.quantized) (class in torch.nn) int() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) int_repr() (torch.Tensor method) integer_interval (in module torch.distributions.constraints) interface() (in module torch.jit) intern() (torch.package.PackageExporter method) interned_modules() (torch.package.PackageExporter method) interpolate (class in torch.ao.nn.quantized.functional) interpolate() (in module torch.nn.functional) Interpreter (class in torch.fx) InterpreterModule (class in torch.export.unflatten) InterpreterModuleDispatcher (class in torch.export.unflatten) Interval (class in torch.autograd.profiler_util) interval (in module torch.distributions.constraints) IntStorage (class in torch) inv (torch.distributions.transforms.Transform property) inv() (in module torch.linalg) inv_ex() (in module torch.linalg) inverse() (in module torch) (torch.Tensor method) inverse_shape() (torch.distributions.transforms.Transform method) InverseGamma (class in torch.distributions.inverse_gamma) ipc_collect() (in module torch.cuda) ipc_handle() (torch.cuda.Event method) ipu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) irecv() (in module torch.distributed) irfft() (in module torch.fft) irfft2() (in module torch.fft) irfftn() (in module torch.fft) is_accessor_node() (in module torch.fx.experimental.symbolic_shapes) is_autocast_available() (in module torch.amp.autocast_mode) is_available() (in module torch.accelerator) (in module torch.backends.cudnn) (in module torch.backends.cusparselt) (in module torch.backends.mkl) (in module torch.backends.mkldnn) (in module torch.backends.mps) (in module torch.backends.nnpack) (in module torch.backends.openmp) (in module torch.backends.opt_einsum) (in module torch.cpu) (in module torch.cuda) (in module torch.distributed) (in module torch.mtia) (in module torch.profiler.itt) (in module torch.xpu) is_built() (in module torch.backends.cuda) (in module torch.backends.mps) is_capturing_metal() (in module torch.mps.profiler) is_closed() (torch.distributed.elastic.rendezvous.RendezvousHandler method) is_coalesced() (torch.Tensor method) is_compiling() (in module torch.compiler) is_completed() (torch.distributed.Work method) is_complex() (in module torch) (torch.Tensor method) is_concrete_bool() (in module torch.fx.experimental.symbolic_shapes) is_concrete_float() (in module torch.fx.experimental.symbolic_shapes) is_concrete_int() (in module torch.fx.experimental.symbolic_shapes) is_conj() (in module torch) (torch.Tensor method) is_contiguous() (torch.Tensor method) is_cuda (torch.nn.utils.rnn.PackedSequence property) (torch.Tensor attribute) (torch.TypedStorage property) (torch.UntypedStorage property) is_current_stream_capturing() (in module torch.cuda) is_dependent() (in module torch.distributions.constraints) is_deterministic_algorithms_warn_only_enabled() (in module torch) is_dynamo_compiling() (in module torch.compiler) is_enabled() (in module torch.cuda.tunable) (torch.sparse.check_sparse_tensor_invariants static method) is_exporting() (in module torch.compiler) is_flash_attention_available() (in module torch.backends.cuda) is_floating_point() (in module torch) (torch.Tensor method) is_gloo_available() (in module torch.distributed) is_grad_enabled() (in module torch) is_hpu (torch.TypedStorage property) (torch.UntypedStorage property) is_impure() (torch.fx.Node method) is_in_onnx_export() (in module torch.onnx) is_inference() (torch.Tensor method) is_inference_mode_enabled() (in module torch) is_initialized() (in module torch.cuda) (in module torch.distributed) (in module torch.mtia) (in module torch.xpu) is_integer() (torch.SymFloat method) is_last() (in module torch.distributed.GradBucket) is_leaf (torch.Tensor attribute) is_leaf_module() (torch.ao.ns._numeric_suite_fx.NSTracer method) (torch.fx.Tracer method) is_meta (torch.Tensor attribute) is_metal_capture_enabled() (in module torch.mps.profiler) is_mpi_available() (in module torch.distributed) is_nccl_available() (in module torch.distributed) is_ninja_available() (in module torch.utils.cpp_extension) is_nonzero() (in module torch) is_onnxrt_backend_supported() (in module torch.onnx) is_owner() (torch.distributed.rpc.PyRRef method) is_parametrized() (in module torch.nn.utils.parametrize) is_partial() (torch.distributed.tensor.placement_types.Placement method) is_pinned() (torch.nn.utils.rnn.PackedSequence method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) is_pruned() (in module torch.nn.utils.prune) is_quantized (torch.Tensor attribute) is_replicate() (torch.distributed.tensor.placement_types.Placement method) is_running() (torch.distributed.elastic.agent.server.WorkerState static method) is_scripting() (in module torch.jit) is_set_to() (torch.Tensor method) is_shard() (torch.distributed.tensor.placement_types.Placement method) is_shared() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) is_signed() (torch.Tensor method) is_sparse (torch.Tensor attribute) (torch.TypedStorage attribute) (torch.UntypedStorage attribute) is_sparse_csr (torch.Tensor attribute) (torch.UntypedStorage attribute) is_storage() (in module torch) is_success() (torch.distributed.Work method) is_tensor() (in module torch) is_tensor_like() (in module torch.overrides) is_tensor_method_or_property() (in module torch.overrides) is_tf32_supported() (in module torch.cuda) is_torchelastic_launched() (in module torch.distributed) is_tracing() (in module torch.jit) is_unbacked_symint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) is_warn_always_enabled() (in module torch) is_xccl_available() (in module torch.distributed.distributed_c10d) isclose() (in module torch) (torch.Tensor method) isend() (in module torch.distributed) isfinite() (in module torch) (torch.Tensor method) isin() (in module torch) isinf() (in module torch) (torch.Tensor method) isinstance() (in module torch.jit) isnan() (in module torch) (torch.Tensor method) isneginf() (in module torch) (torch.Tensor method) isposinf() (in module torch) (torch.Tensor method) isreal() (in module torch) (torch.Tensor method) istft() (in module torch) (torch.Tensor method) item() (torch.Tensor method) items() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) itemsize (torch.Tensor attribute) iter() (torch.fx.Tracer method) IterableDataset (class in torch.utils.data) J jacfwd() (in module torch.func) jacobian() (in module torch.autograd.functional) jacrev() (in module torch.func) JitScalarType (class in torch.onnx) job_id (in module torch.compiler.config) Join (class in torch.distributed.algorithms) join (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) join() (torch.multiprocessing.SpawnContext method) (torch.nn.parallel.DistributedDataParallel method) join_device (torch.distributed.algorithms.Joinable property) (torch.distributed.optim.ZeroRedundancyOptimizer property) join_hook() (torch.distributed.algorithms.Joinable method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.nn.parallel.DistributedDataParallel method) join_process_group (torch.distributed.algorithms.Joinable property) (torch.distributed.optim.ZeroRedundancyOptimizer property) Joinable (class in torch.distributed.algorithms) JoinHook (class in torch.distributed.algorithms) jvp() (in module torch.autograd.functional) (in module torch.func) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) K kaiming_normal_() (in module torch.nn.init) kaiming_uniform_() (in module torch.nn.init) kaiser() (in module torch.signal.windows) kaiser_window() (in module torch) Kernel (class in torch.autograd.profiler_util) key_averages() (torch.autograd.profiler.profile method) (torch.profiler._KinetoProfile method) keys() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.fx.Tracer method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) KinetoStepTracker (class in torch.autograd.profiler) kl_div() (in module torch.nn.functional) kl_divergence() (in module torch.distributions.kl) KLDivLoss (class in torch.nn) kron() (in module torch) kthvalue() (in module torch) (torch.Tensor method) Kumaraswamy (class in torch.distributions.kumaraswamy) kv_indices (torch.nn.attention.flex_attention.BlockMask attribute) kv_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) kwargs (torch.fx.Node property) L l1_loss() (in module torch.nn.functional) l1_unstructured() (in module torch.nn.utils.prune) L1Loss (class in torch.nn) L1Unstructured (class in torch.nn.utils.prune) LambdaLR (class in torch.optim.lr_scheduler) Laplace (class in torch.distributions.laplace) last_call (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousTimeout property) layer_norm() (in module torch.nn.functional) LayerNorm (class in torch.ao.nn.quantized) (class in torch.nn) layout (class in torch) LazyBatchNorm1d (class in torch.nn) LazyBatchNorm2d (class in torch.nn) LazyBatchNorm3d (class in torch.nn) LazyConv1d (class in torch.nn) LazyConv2d (class in torch.nn) LazyConv3d (class in torch.nn) LazyConvTranspose1d (class in torch.nn) LazyConvTranspose2d (class in torch.nn) LazyConvTranspose3d (class in torch.nn) LazyInstanceNorm1d (class in torch.nn) LazyInstanceNorm2d (class in torch.nn) LazyInstanceNorm3d (class in torch.nn) LazyLinear (class in torch.nn) LazyModuleMixin (class in torch.nn.modules.lazy) LBFGS (class in torch.optim) lcm() (in module torch) (torch.Tensor method) lcm_() (torch.Tensor method) ldexp() (in module torch) (torch.Tensor method) ldexp_() (torch.Tensor method) ldl_factor() (in module torch.linalg) ldl_factor_ex() (in module torch.linalg) ldl_solve() (in module torch.linalg) le() (in module torch) (torch.Tensor method) le_() (torch.Tensor method) leaky_relu (class in torch.ao.nn.quantized.functional) leaky_relu() (in module torch.nn.functional) leaky_relu_() (in module torch.nn.functional) LeakyReLU (class in torch.ao.nn.quantized) (class in torch.nn) lerp() (in module torch) (torch.Tensor method) lerp_() (torch.Tensor method) less() (in module torch) (torch.Tensor method) less_() (torch.Tensor method) less_equal() (in module torch) (torch.Tensor method) less_equal_() (torch.Tensor method) less_than (in module torch.distributions.constraints) lgamma() (in module torch) (torch.Tensor method) lgamma_() (torch.Tensor method) Library (class in torch.library) libuvBackend (torch.distributed.TCPStore property) Linear (class in torch.ao.nn.qat) (class in torch.ao.nn.qat.dynamic) (class in torch.ao.nn.quantized) (class in torch.ao.nn.quantized.dynamic) linear (class in torch.ao.nn.quantized.functional) Linear (class in torch.nn) linear() (in module torch.nn.functional) linearize() (in module torch.func) LinearLR (class in torch.optim.lr_scheduler) LinearReLU (class in torch.ao.nn.intrinsic) (class in torch.ao.nn.intrinsic.qat) (class in torch.ao.nn.intrinsic.quantized) (class in torch.ao.nn.intrinsic.quantized.dynamic) linspace() (in module torch) lint() (torch.fx.Graph method) list() (in module torch.hub) list_backends() (in module torch.compiler) list_gpu_processes() (in module torch.cuda) LKJCholesky (class in torch.distributions.lkj_cholesky) ln_structured() (in module torch.nn.utils.prune) LnStructured (class in torch.nn.utils.prune) load() (in module torch) (in module torch.distributed.checkpoint.state_dict_loader) (in module torch.export) (in module torch.hub) (in module torch.jit) (in module torch.utils.cpp_extension) load_binary() (torch.package.PackageImporter method) load_bytes() (torch.distributed.checkpoint.LoadPlanner method) load_inline() (in module torch.utils.cpp_extension) load_nvprof() (in module torch.autograd.profiler) load_observer_state_dict (class in torch.ao.quantization.observer) load_pickle() (torch.package.PackageImporter method) load_state_dict() (in module torch.distributed.checkpoint.state_dict_loader) (torch.distributed.checkpoint.stateful.Stateful method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) (torch.optim.swa_utils.SWALR method) load_state_dict_from_url() (in module torch.hub) load_storage() (torch.cuda.gds.GdsFile method) load_text() (torch.package.PackageImporter method) load_url() (in module torch.utils.model_zoo) LoadPlan (class in torch.distributed.checkpoint) LoadPlanner (class in torch.distributed.checkpoint) lobpcg() (in module torch) loc (torch.distributions.log_normal.LogNormal property) local_map() (in module torch.distributed.tensor.experimental) local_response_norm() (in module torch.nn.functional) local_value() (torch.distributed.rpc.PyRRef method) LocalElasticAgent (class in torch.distributed.elastic.agent.server.local_elastic_agent) LocalOptimStateDictConfig (class in torch.distributed.fsdp) LocalResponseNorm (class in torch.nn) LocalStateDictConfig (class in torch.distributed.fsdp) LocalTimerClient (class in torch.distributed.elastic.timer) LocalTimerServer (class in torch.distributed.elastic.timer) log() (in module torch) (torch.Tensor method) log10() (in module torch) (torch.Tensor method) log10_() (torch.Tensor method) log1p() (in module torch) (in module torch.special) (torch.Tensor method) log1p_() (torch.Tensor method) log2() (in module torch) (torch.Tensor method) log2_() (torch.Tensor method) log_() (torch.Tensor method) log_abs_det_jacobian() (torch.distributions.transforms.Transform method) log_cdf() (torch.distributions.generalized_pareto.GeneralizedPareto method) log_comm_debug_tracing_table_to_file() (torch.distributed.tensor.debug.CommDebugMode method) log_debug_info_for_expired_timers() (in module torch.distributed.elastic.timer.debug_info_logging) log_event() (in module torch.monitor) log_ndtr() (in module torch.special) log_normal_() (torch.Tensor method) log_prob() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.beta.Beta method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.geometric.Geometric method) (torch.distributions.gumbel.Gumbel method) (torch.distributions.half_cauchy.HalfCauchy method) (torch.distributions.half_normal.HalfNormal method) (torch.distributions.independent.Independent method) (torch.distributions.laplace.Laplace method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.poisson.Poisson method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.von_mises.VonMises method) (torch.distributions.wishart.Wishart method) (torch.nn.AdaptiveLogSoftmaxWithLoss method) log_softmax() (in module torch.nn.functional) (in module torch.sparse) (in module torch.special) log_survival_function() (torch.distributions.generalized_pareto.GeneralizedPareto method) logaddexp() (in module torch) (torch.Tensor method) logaddexp2() (in module torch) (torch.Tensor method) logcumsumexp() (in module torch) (torch.Tensor method) logdet() (in module torch) (torch.Tensor method) Logger (class in torch.ao.ns._numeric_suite) loggers_set_enabled() (in module torch.ao.ns._numeric_suite_fx) loggers_set_save_activations() (in module torch.ao.ns._numeric_suite_fx) logical_and() (in module torch) (torch.Tensor method) logical_and_() (torch.Tensor method) logical_not() (in module torch) (torch.Tensor method) logical_not_() (torch.Tensor method) logical_or() (in module torch) (torch.Tensor method) logical_or_() (torch.Tensor method) logical_xor() (in module torch) (torch.Tensor method) logical_xor_() (torch.Tensor method) logit() (in module torch) (in module torch.special) (torch.Tensor method) logit_() (torch.Tensor method) LogitRelaxedBernoulli (class in torch.distributions.relaxed_bernoulli) logits (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.geometric.Geometric property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) LogNormal (class in torch.distributions.log_normal) LogsDest (class in torch.distributed.elastic.multiprocessing.api) LogSigmoid (class in torch.nn) logsigmoid() (in module torch.nn.functional) LogSoftmax (class in torch.nn) logspace() (in module torch) LogsSpecs (class in torch.distributed.elastic.multiprocessing.api) logsumexp() (in module torch) (in module torch.special) (torch.Tensor method) long() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) LongStorage (class in torch) lookup_object() (torch.distributed.checkpoint.DefaultSavePlanner method) lookup_tensor() (torch.distributed.checkpoint.DefaultLoadPlanner method) loss_parallel() (in module torch.distributed.tensor.parallel) lower_pt2e_quantized_to_x86 (class in torch.ao.quantization.pt2e.lowering) LowerCholeskyTransform (class in torch.distributions.transforms) LowRankMultivariateNormal (class in torch.distributions.lowrank_multivariate_normal) lp_pool1d() (in module torch.nn.functional) lp_pool2d() (in module torch.nn.functional) lp_pool3d() (in module torch.nn.functional) LPPool1d (class in torch.nn) LPPool2d (class in torch.nn) LPPool3d (class in torch.nn) LRScheduler (class in torch.optim.lr_scheduler) lru_cache() (in module torch.fx.experimental.symbolic_shapes) LSTM (class in torch.ao.nn.quantizable) (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) LSTMCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) lstsq() (in module torch.linalg) lt() (in module torch) (torch.Tensor method) lt_() (torch.Tensor method) lu() (in module torch) (in module torch.linalg) (torch.Tensor method) lu_factor() (in module torch.linalg) lu_factor_ex() (in module torch.linalg) lu_solve() (in module torch) (in module torch.linalg) (torch.Tensor method) lu_unpack() (in module torch) M main_hook() (torch.distributed.algorithms.JoinHook method) make_dual() (in module torch.autograd.forward_ad) make_fx() (in module torch.fx.experimental.proxy_tensor) make_graphed_callables() (in module torch.cuda) make_tensor() (in module torch.testing) manual_seed() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.random) (in module torch.xpu) (torch.Generator method) manual_seed_all() (in module torch.cuda) (in module torch.xpu) map_() (torch.Tensor method) map_nodes_to_values() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) MappingType (class in torch.ao.quantization.observer) margin_ranking_loss() (in module torch.nn.functional) MarginRankingLoss (class in torch.nn) mark() (in module torch.cuda.nvtx) (in module torch.profiler.itt) mark_dirty() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) mark_non_differentiable() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) mask_mod (torch.nn.attention.flex_attention.BlockMask attribute) masked_fill() (torch.Tensor method) masked_fill_() (torch.Tensor method) masked_scatter() (torch.Tensor method) masked_scatter_() (torch.Tensor method) masked_select() (in module torch) (in module torch.nested) (torch.Tensor method) materialize() (torch.export.decomp_utils.CustomDecompTable method) math_sdp_enabled() (in module torch.backends.cuda) matmul() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_exp() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_norm() (in module torch.linalg) matrix_power() (in module torch) (in module torch.linalg) (torch.Tensor method) matrix_rank() (in module torch.linalg) max() (in module torch) (torch.Tensor method) max_memory_allocated() (in module torch.cuda) (in module torch.xpu) max_memory_cached() (in module torch.cuda) max_memory_reserved() (in module torch.cuda) (in module torch.xpu) max_pool1d (class in torch.ao.nn.quantized.functional) max_pool1d() (in module torch.nn.functional) max_pool2d (class in torch.ao.nn.quantized.functional) max_pool2d() (in module torch.nn.functional) max_pool3d() (in module torch.nn.functional) max_size (in module torch.backends.cuda.cufft_plan_cache) max_unpool1d() (in module torch.nn.functional) max_unpool2d() (in module torch.nn.functional) max_unpool3d() (in module torch.nn.functional) maximum() (in module torch) (torch.Tensor method) MaxPool1d (class in torch.nn) MaxPool2d (class in torch.nn) MaxPool3d (class in torch.nn) MaxUnpool1d (class in torch.nn) MaxUnpool2d (class in torch.nn) MaxUnpool3d (class in torch.nn) maybe_disable_thunkify() (in module torch.fx.experimental.proxy_tensor) maybe_enable_thunkify() (in module torch.fx.experimental.proxy_tensor) mean (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) mean() (in module torch) (torch.Tensor method) Measurement (class in torch.utils.benchmark) median() (in module torch) (torch.Tensor method) mem_efficient_sdp_enabled() (in module torch.backends.cuda) mem_get_info() (in module torch.cuda) (in module torch.xpu) memory_allocated() (in module torch.cuda) (in module torch.xpu) memory_cached() (in module torch.cuda) memory_format (class in torch) memory_reserved() (in module torch.cuda) (in module torch.xpu) memory_snapshot() (in module torch.cuda) memory_stats() (in module torch.cuda) (in module torch.mtia) (in module torch.mtia.memory) (in module torch.xpu) memory_stats_as_nested_dict() (in module torch.xpu) memory_summary() (in module torch.cuda) memory_usage() (in module torch.cuda) MemPool (class in torch.cuda) MemPoolContext (class in torch.cuda) MemRecordsAcc (class in torch.autograd.profiler_util) merge() (torch.utils.benchmark.Measurement static method) merge_chunks() (in module torch.distributed.pipelining.microbatch) merge_masks() (torch.nn.MultiheadAttention method) meshgrid() (in module torch) metadata() (torch.autograd.graph.Node method) metal_capture() (in module torch.mps.profiler) MetricHandler (class in torch.distributed.elastic.metrics.api) mgpu_tune_gemm_in_file() (in module torch.cuda.tunable) mH (torch.Tensor attribute) min() (in module torch) (torch.Tensor method) minimum() (in module torch) (torch.Tensor method) MinMaxObserver (class in torch.ao.quantization.observer) Mish (class in torch.nn) mish() (in module torch.nn.functional) MixedPrecision (class in torch.distributed.fsdp) MixedPrecisionPolicy (class in torch.distributed.fsdp) mixture_distribution (torch.distributions.mixture_same_family.MixtureSameFamily property) MixtureSameFamily (class in torch.distributions.mixture_same_family) mm() (in module torch) (in module torch.sparse) (torch.Tensor method) mock() (torch.package.PackageExporter method) mocked_modules() (torch.package.PackageExporter method) mode (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) mode() (in module torch) (torch.Tensor method) model_is_exported (class in torch.ao.quantization.pt2e.export_utils) model_proto (torch.onnx.ONNXProgram property) module torch torch.__config__ torch.__future__ torch._logging torch.accelerator torch.amp torch.amp.autocast_mode torch.amp.grad_scaler torch.ao torch.ao.nn torch.ao.nn.intrinsic torch.ao.nn.intrinsic.modules torch.ao.nn.intrinsic.modules.fused torch.ao.nn.intrinsic.qat torch.ao.nn.intrinsic.qat.modules torch.ao.nn.intrinsic.qat.modules.conv_fused torch.ao.nn.intrinsic.qat.modules.linear_fused torch.ao.nn.intrinsic.qat.modules.linear_relu torch.ao.nn.intrinsic.quantized torch.ao.nn.intrinsic.quantized.dynamic torch.ao.nn.intrinsic.quantized.dynamic.modules torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu torch.ao.nn.intrinsic.quantized.modules torch.ao.nn.intrinsic.quantized.modules.bn_relu torch.ao.nn.intrinsic.quantized.modules.conv_add torch.ao.nn.intrinsic.quantized.modules.conv_relu torch.ao.nn.intrinsic.quantized.modules.linear_relu torch.ao.nn.qat torch.ao.nn.qat.dynamic torch.ao.nn.qat.dynamic.modules torch.ao.nn.qat.dynamic.modules.linear torch.ao.nn.qat.modules torch.ao.nn.qat.modules.conv torch.ao.nn.qat.modules.embedding_ops torch.ao.nn.qat.modules.linear torch.ao.nn.quantizable torch.ao.nn.quantizable.modules torch.ao.nn.quantizable.modules.activation torch.ao.nn.quantizable.modules.rnn torch.ao.nn.quantized torch.ao.nn.quantized.dynamic torch.ao.nn.quantized.dynamic.modules torch.ao.nn.quantized.dynamic.modules.conv torch.ao.nn.quantized.dynamic.modules.linear torch.ao.nn.quantized.dynamic.modules.rnn torch.ao.nn.quantized.functional torch.ao.nn.quantized.modules torch.ao.nn.quantized.modules.activation torch.ao.nn.quantized.modules.batchnorm torch.ao.nn.quantized.modules.conv torch.ao.nn.quantized.modules.dropout torch.ao.nn.quantized.modules.embedding_ops torch.ao.nn.quantized.modules.functional_modules torch.ao.nn.quantized.modules.linear torch.ao.nn.quantized.modules.normalization torch.ao.nn.quantized.modules.rnn torch.ao.nn.quantized.modules.utils torch.ao.nn.quantized.reference torch.ao.nn.quantized.reference.modules torch.ao.nn.quantized.reference.modules.conv torch.ao.nn.quantized.reference.modules.linear torch.ao.nn.quantized.reference.modules.rnn torch.ao.nn.quantized.reference.modules.sparse torch.ao.nn.quantized.reference.modules.utils torch.ao.nn.sparse torch.ao.nn.sparse.quantized torch.ao.nn.sparse.quantized.dynamic torch.ao.nn.sparse.quantized.dynamic.linear torch.ao.nn.sparse.quantized.linear torch.ao.nn.sparse.quantized.utils torch.ao.ns torch.ao.ns._numeric_suite torch.ao.ns._numeric_suite_fx torch.ao.ns.fx torch.ao.ns.fx.graph_matcher torch.ao.ns.fx.graph_passes torch.ao.ns.fx.mappings torch.ao.ns.fx.n_shadows_utils torch.ao.ns.fx.ns_types torch.ao.ns.fx.pattern_utils torch.ao.ns.fx.qconfig_multi_mapping torch.ao.ns.fx.utils torch.ao.ns.fx.weight_utils torch.ao.pruning torch.ao.pruning.scheduler torch.ao.pruning.scheduler.base_scheduler torch.ao.pruning.scheduler.cubic_scheduler torch.ao.pruning.scheduler.lambda_scheduler torch.ao.pruning.sparsifier torch.ao.pruning.sparsifier.base_sparsifier torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier torch.ao.pruning.sparsifier.utils torch.ao.pruning.sparsifier.weight_norm_sparsifier torch.ao.quantization torch.ao.quantization.backend_config torch.ao.quantization.backend_config.backend_config torch.ao.quantization.backend_config.executorch torch.ao.quantization.backend_config.fbgemm torch.ao.quantization.backend_config.native torch.ao.quantization.backend_config.observation_type torch.ao.quantization.backend_config.onednn torch.ao.quantization.backend_config.qnnpack torch.ao.quantization.backend_config.tensorrt torch.ao.quantization.backend_config.utils torch.ao.quantization.backend_config.x86 torch.ao.quantization.fake_quantize torch.ao.quantization.fuse_modules torch.ao.quantization.fuser_method_mappings torch.ao.quantization.fx torch.ao.quantization.fx.convert torch.ao.quantization.fx.custom_config torch.ao.quantization.fx.fuse torch.ao.quantization.fx.fuse_handler torch.ao.quantization.fx.graph_module torch.ao.quantization.fx.lower_to_fbgemm torch.ao.quantization.fx.lower_to_qnnpack torch.ao.quantization.fx.lstm_utils torch.ao.quantization.fx.match_utils torch.ao.quantization.fx.pattern_utils torch.ao.quantization.fx.prepare torch.ao.quantization.fx.qconfig_mapping_utils torch.ao.quantization.fx.quantize_handler torch.ao.quantization.fx.tracer torch.ao.quantization.fx.utils torch.ao.quantization.observer torch.ao.quantization.pt2e torch.ao.quantization.pt2e.duplicate_dq_pass torch.ao.quantization.pt2e.export_utils torch.ao.quantization.pt2e.graph_utils torch.ao.quantization.pt2e.lowering torch.ao.quantization.pt2e.port_metadata_pass torch.ao.quantization.pt2e.prepare torch.ao.quantization.pt2e.qat_utils torch.ao.quantization.pt2e.representation torch.ao.quantization.pt2e.representation.rewrite torch.ao.quantization.pt2e.utils torch.ao.quantization.qconfig torch.ao.quantization.qconfig_mapping torch.ao.quantization.quant_type torch.ao.quantization.quantization_mappings torch.ao.quantization.quantize_fx torch.ao.quantization.quantize_jit torch.ao.quantization.quantize_pt2e torch.ao.quantization.quantizer torch.ao.quantization.quantizer.composable_quantizer torch.ao.quantization.quantizer.embedding_quantizer torch.ao.quantization.quantizer.quantizer torch.ao.quantization.quantizer.utils torch.ao.quantization.quantizer.x86_inductor_quantizer torch.ao.quantization.quantizer.xnnpack_quantizer torch.ao.quantization.quantizer.xnnpack_quantizer_utils torch.ao.quantization.quantizer.xpu_inductor_quantizer torch.ao.quantization.stubs torch.ao.quantization.utils torch.autograd torch.autograd.anomaly_mode torch.autograd.forward_ad torch.autograd.function torch.autograd.functional torch.autograd.grad_mode torch.autograd.gradcheck torch.autograd.graph torch.autograd.profiler torch.autograd.profiler_legacy torch.autograd.profiler_util torch.autograd.variable torch.backends torch.backends.cpu torch.backends.cuda torch.backends.cudnn torch.backends.cudnn.rnn torch.backends.cusparselt torch.backends.kleidiai torch.backends.mha torch.backends.mkl torch.backends.mkldnn torch.backends.mps torch.backends.nnpack torch.backends.openmp torch.backends.opt_einsum torch.backends.quantized torch.backends.xeon torch.backends.xeon.run_cpu torch.backends.xnnpack torch.compiler torch.compiler.config torch.contrib torch.cpu torch.cpu.amp torch.cpu.amp.autocast_mode torch.cpu.amp.grad_scaler torch.cuda torch.cuda._sanitizer torch.cuda.amp torch.cuda.amp.autocast_mode torch.cuda.amp.common torch.cuda.amp.grad_scaler torch.cuda.comm torch.cuda.error torch.cuda.gds torch.cuda.graphs torch.cuda.jiterator torch.cuda.memory torch.cuda.nccl torch.cuda.nvtx torch.cuda.profiler torch.cuda.random torch.cuda.sparse torch.cuda.streams torch.cuda.tunable torch.distributed torch.distributed.algorithms torch.distributed.algorithms.ddp_comm_hooks torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks torch.distributed.algorithms.ddp_comm_hooks.default_hooks torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks torch.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks torch.distributed.algorithms.join torch.distributed.algorithms.model_averaging torch.distributed.algorithms.model_averaging.averagers torch.distributed.algorithms.model_averaging.hierarchical_model_averager torch.distributed.algorithms.model_averaging.utils torch.distributed.argparse_util torch.distributed.autograd torch.distributed.c10d_logger torch.distributed.checkpoint torch.distributed.checkpoint.api torch.distributed.checkpoint.default_planner torch.distributed.checkpoint.filesystem torch.distributed.checkpoint.format_utils torch.distributed.checkpoint.logger torch.distributed.checkpoint.logging_handlers torch.distributed.checkpoint.metadata torch.distributed.checkpoint.optimizer torch.distributed.checkpoint.planner torch.distributed.checkpoint.planner_helpers torch.distributed.checkpoint.resharding torch.distributed.checkpoint.staging torch.distributed.checkpoint.state_dict torch.distributed.checkpoint.state_dict_loader torch.distributed.checkpoint.state_dict_saver torch.distributed.checkpoint.stateful torch.distributed.checkpoint.storage torch.distributed.checkpoint.utils torch.distributed.collective_utils torch.distributed.constants torch.distributed.device_mesh torch.distributed.distributed_c10d torch.distributed.elastic torch.distributed.elastic.agent torch.distributed.elastic.agent.server torch.distributed.elastic.agent.server.api torch.distributed.elastic.agent.server.health_check_server torch.distributed.elastic.agent.server.local_elastic_agent torch.distributed.elastic.control_plane torch.distributed.elastic.events torch.distributed.elastic.events.api torch.distributed.elastic.events.handlers torch.distributed.elastic.metrics torch.distributed.elastic.metrics.api torch.distributed.elastic.multiprocessing torch.distributed.elastic.multiprocessing.api torch.distributed.elastic.multiprocessing.errors torch.distributed.elastic.multiprocessing.errors.error_handler torch.distributed.elastic.multiprocessing.errors.handlers torch.distributed.elastic.multiprocessing.redirects torch.distributed.elastic.multiprocessing.subprocess_handler torch.distributed.elastic.multiprocessing.subprocess_handler.handlers torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler torch.distributed.elastic.multiprocessing.tail_log torch.distributed.elastic.rendezvous torch.distributed.elastic.rendezvous.api torch.distributed.elastic.rendezvous.c10d_rendezvous_backend torch.distributed.elastic.rendezvous.dynamic_rendezvous torch.distributed.elastic.rendezvous.etcd_rendezvous torch.distributed.elastic.rendezvous.etcd_rendezvous_backend torch.distributed.elastic.rendezvous.etcd_server torch.distributed.elastic.rendezvous.etcd_store torch.distributed.elastic.rendezvous.registry torch.distributed.elastic.rendezvous.static_tcp_rendezvous torch.distributed.elastic.rendezvous.utils torch.distributed.elastic.timer torch.distributed.elastic.timer.api torch.distributed.elastic.timer.debug_info_logging torch.distributed.elastic.timer.file_based_local_timer torch.distributed.elastic.timer.local_timer torch.distributed.elastic.utils torch.distributed.elastic.utils.api torch.distributed.elastic.utils.data torch.distributed.elastic.utils.data.cycling_iterator torch.distributed.elastic.utils.data.elastic_distributed_sampler torch.distributed.elastic.utils.distributed torch.distributed.elastic.utils.log_level torch.distributed.elastic.utils.logging torch.distributed.elastic.utils.store torch.distributed.fsdp torch.distributed.fsdp.api torch.distributed.fsdp.fully_sharded_data_parallel torch.distributed.fsdp.sharded_grad_scaler torch.distributed.fsdp.wrap torch.distributed.launch torch.distributed.launcher torch.distributed.launcher.api torch.distributed.logging_handlers torch.distributed.nn torch.distributed.nn.api torch.distributed.nn.api.remote_module torch.distributed.nn.functional torch.distributed.nn.jit torch.distributed.nn.jit.instantiator torch.distributed.nn.jit.templates torch.distributed.nn.jit.templates.remote_module_template torch.distributed.optim torch.distributed.optim.apply_optimizer_in_backward torch.distributed.optim.functional_adadelta torch.distributed.optim.functional_adagrad torch.distributed.optim.functional_adam torch.distributed.optim.functional_adamax torch.distributed.optim.functional_adamw torch.distributed.optim.functional_rmsprop torch.distributed.optim.functional_rprop torch.distributed.optim.functional_sgd torch.distributed.optim.named_optimizer torch.distributed.optim.optimizer torch.distributed.optim.post_localSGD_optimizer torch.distributed.optim.utils torch.distributed.optim.zero_redundancy_optimizer torch.distributed.pipelining torch.distributed.pipelining.microbatch torch.distributed.pipelining.schedules torch.distributed.pipelining.stage torch.distributed.remote_device torch.distributed.rendezvous torch.distributed.rpc torch.distributed.rpc.api torch.distributed.rpc.backend_registry torch.distributed.rpc.constants torch.distributed.rpc.functions torch.distributed.rpc.internal torch.distributed.rpc.options torch.distributed.rpc.rref_proxy torch.distributed.rpc.server_process_global_profiler torch.distributed.run torch.distributed.tensor torch.distributed.tensor.debug torch.distributed.tensor.device_mesh torch.distributed.tensor.experimental torch.distributed.tensor.parallel torch.distributed.tensor.parallel.api torch.distributed.tensor.parallel.ddp torch.distributed.tensor.parallel.fsdp torch.distributed.tensor.parallel.input_reshard torch.distributed.tensor.parallel.loss torch.distributed.tensor.parallel.style torch.distributed.tensor.placement_types torch.distributed.utils torch.distributions torch.distributions.bernoulli torch.distributions.beta torch.distributions.binomial torch.distributions.categorical torch.distributions.cauchy torch.distributions.chi2 torch.distributions.constraint_registry torch.distributions.constraints torch.distributions.continuous_bernoulli torch.distributions.dirichlet torch.distributions.distribution torch.distributions.exp_family torch.distributions.exponential torch.distributions.fishersnedecor torch.distributions.gamma torch.distributions.generalized_pareto torch.distributions.geometric torch.distributions.gumbel torch.distributions.half_cauchy torch.distributions.half_normal torch.distributions.independent torch.distributions.inverse_gamma torch.distributions.kl torch.distributions.kumaraswamy torch.distributions.laplace torch.distributions.lkj_cholesky torch.distributions.log_normal torch.distributions.logistic_normal torch.distributions.lowrank_multivariate_normal torch.distributions.mixture_same_family torch.distributions.multinomial torch.distributions.multivariate_normal torch.distributions.negative_binomial torch.distributions.normal torch.distributions.one_hot_categorical torch.distributions.pareto torch.distributions.poisson torch.distributions.relaxed_bernoulli torch.distributions.relaxed_categorical torch.distributions.studentT torch.distributions.transformed_distribution torch.distributions.transforms torch.distributions.uniform torch.distributions.utils torch.distributions.von_mises torch.distributions.weibull torch.distributions.wishart torch.export torch.export.custom_obj torch.export.custom_ops torch.export.decomp_utils torch.export.dynamic_shapes torch.export.experimental torch.export.exported_program torch.export.graph_signature torch.export.passes torch.export.unflatten torch.fft torch.func torch.functional torch.futures torch.fx torch.fx.annotate torch.fx.config torch.fx.experimental torch.fx.experimental.accelerator_partitioner torch.fx.experimental.const_fold torch.fx.experimental.debug torch.fx.experimental.graph_gradual_typechecker torch.fx.experimental.merge_matmul torch.fx.experimental.meta_tracer torch.fx.experimental.migrate_gradual_types torch.fx.experimental.migrate_gradual_types.constraint torch.fx.experimental.migrate_gradual_types.constraint_generator torch.fx.experimental.migrate_gradual_types.constraint_transformation torch.fx.experimental.migrate_gradual_types.operation torch.fx.experimental.migrate_gradual_types.transform_to_z3 torch.fx.experimental.migrate_gradual_types.util torch.fx.experimental.migrate_gradual_types.z3_types torch.fx.experimental.normalize torch.fx.experimental.optimization torch.fx.experimental.partitioner_utils torch.fx.experimental.proxy_tensor torch.fx.experimental.recording torch.fx.experimental.refinement_types torch.fx.experimental.rewriter torch.fx.experimental.schema_type_annotation torch.fx.experimental.sym_node torch.fx.experimental.symbolic_shapes torch.fx.experimental.unification torch.fx.experimental.unification.core torch.fx.experimental.unification.dispatch torch.fx.experimental.unification.match torch.fx.experimental.unification.more torch.fx.experimental.unification.multipledispatch torch.fx.experimental.unification.multipledispatch.conflict torch.fx.experimental.unification.multipledispatch.core torch.fx.experimental.unification.multipledispatch.dispatcher torch.fx.experimental.unification.multipledispatch.utils torch.fx.experimental.unification.multipledispatch.variadic torch.fx.experimental.unification.unification_tools torch.fx.experimental.unification.utils torch.fx.experimental.unification.variable torch.fx.experimental.unify_refinements torch.fx.experimental.validator torch.fx.graph torch.fx.graph_module torch.fx.immutable_collections torch.fx.interpreter torch.fx.node torch.fx.operator_schemas torch.fx.passes torch.fx.passes.annotate_getitem_nodes torch.fx.passes.backends torch.fx.passes.backends.cudagraphs torch.fx.passes.dialect torch.fx.passes.dialect.common torch.fx.passes.dialect.common.cse_pass torch.fx.passes.fake_tensor_prop torch.fx.passes.graph_drawer torch.fx.passes.graph_manipulation torch.fx.passes.graph_transform_observer torch.fx.passes.infra torch.fx.passes.infra.partitioner torch.fx.passes.infra.pass_base torch.fx.passes.infra.pass_manager torch.fx.passes.net_min_base torch.fx.passes.operator_support torch.fx.passes.param_fetch torch.fx.passes.pass_manager torch.fx.passes.reinplace torch.fx.passes.runtime_assert torch.fx.passes.shape_prop torch.fx.passes.split_module torch.fx.passes.split_utils torch.fx.passes.splitter_base torch.fx.passes.tests torch.fx.passes.tests.test_pass_manager torch.fx.passes.tools_common torch.fx.passes.utils torch.fx.passes.utils.common torch.fx.passes.utils.fuser_utils torch.fx.passes.utils.matcher_utils torch.fx.passes.utils.matcher_with_name_node_map_utils torch.fx.passes.utils.source_matcher_utils torch.fx.proxy torch.fx.subgraph_rewriter torch.fx.tensor_type torch.fx.traceback torch.hub torch.jit torch.jit.annotations torch.jit.frontend torch.jit.generate_bytecode torch.jit.mobile torch.jit.quantized torch.jit.supported_ops torch.jit.unsupported_tensor_ops torch.library torch.linalg torch.masked torch.masked.maskedtensor torch.masked.maskedtensor.binary torch.masked.maskedtensor.core torch.masked.maskedtensor.creation torch.masked.maskedtensor.passthrough torch.masked.maskedtensor.reductions torch.masked.maskedtensor.unary torch.monitor torch.mps torch.mps.event torch.mps.profiler torch.mtia torch.mtia.memory torch.multiprocessing torch.multiprocessing.pool torch.multiprocessing.queue torch.multiprocessing.reductions torch.multiprocessing.spawn torch.nested torch.nn torch.nn.attention torch.nn.attention.bias torch.nn.attention.experimental torch.nn.attention.flex_attention torch.nn.backends torch.nn.backends.thnn torch.nn.common_types torch.nn.cpp torch.nn.functional torch.nn.grad torch.nn.init torch.nn.intrinsic torch.nn.intrinsic.modules torch.nn.intrinsic.modules.fused torch.nn.intrinsic.qat torch.nn.intrinsic.qat.modules torch.nn.intrinsic.qat.modules.conv_fused torch.nn.intrinsic.qat.modules.linear_fused torch.nn.intrinsic.qat.modules.linear_relu torch.nn.intrinsic.quantized torch.nn.intrinsic.quantized.dynamic torch.nn.intrinsic.quantized.dynamic.modules torch.nn.intrinsic.quantized.dynamic.modules.linear_relu torch.nn.intrinsic.quantized.modules torch.nn.intrinsic.quantized.modules.bn_relu torch.nn.intrinsic.quantized.modules.conv_relu torch.nn.intrinsic.quantized.modules.linear_relu torch.nn.modules torch.nn.modules.activation torch.nn.modules.adaptive torch.nn.modules.batchnorm torch.nn.modules.channelshuffle torch.nn.modules.container torch.nn.modules.conv torch.nn.modules.distance torch.nn.modules.dropout torch.nn.modules.flatten torch.nn.modules.fold torch.nn.modules.instancenorm torch.nn.modules.lazy torch.nn.modules.linear torch.nn.modules.loss torch.nn.modules.module torch.nn.modules.normalization torch.nn.modules.padding torch.nn.modules.pixelshuffle torch.nn.modules.pooling torch.nn.modules.rnn torch.nn.modules.sparse torch.nn.modules.transformer torch.nn.modules.upsampling torch.nn.modules.utils torch.nn.parallel torch.nn.parallel.comm torch.nn.parallel.distributed torch.nn.parallel.parallel_apply torch.nn.parallel.replicate torch.nn.parallel.scatter_gather torch.nn.parameter torch.nn.qat torch.nn.qat.dynamic torch.nn.qat.dynamic.modules torch.nn.qat.dynamic.modules.linear torch.nn.qat.modules torch.nn.qat.modules.conv torch.nn.qat.modules.embedding_ops torch.nn.qat.modules.linear torch.nn.quantizable torch.nn.quantizable.modules torch.nn.quantizable.modules.activation torch.nn.quantizable.modules.rnn torch.nn.quantized torch.nn.quantized.dynamic torch.nn.quantized.dynamic.modules torch.nn.quantized.dynamic.modules.conv torch.nn.quantized.dynamic.modules.linear torch.nn.quantized.dynamic.modules.rnn torch.nn.quantized.functional torch.nn.quantized.modules torch.nn.quantized.modules.activation torch.nn.quantized.modules.batchnorm torch.nn.quantized.modules.conv torch.nn.quantized.modules.dropout torch.nn.quantized.modules.embedding_ops torch.nn.quantized.modules.functional_modules torch.nn.quantized.modules.linear torch.nn.quantized.modules.normalization torch.nn.quantized.modules.rnn torch.nn.quantized.modules.utils torch.nn.utils torch.nn.utils.clip_grad torch.nn.utils.convert_parameters torch.nn.utils.fusion torch.nn.utils.init torch.nn.utils.memory_format torch.nn.utils.parametrizations torch.nn.utils.parametrize torch.nn.utils.prune torch.nn.utils.rnn torch.nn.utils.stateless torch.onnx torch.onnx.errors torch.onnx.operators torch.onnx.ops torch.onnx.symbolic_caffe2 torch.onnx.symbolic_helper torch.onnx.symbolic_opset10 torch.onnx.symbolic_opset11 torch.onnx.symbolic_opset12 torch.onnx.symbolic_opset13 torch.onnx.symbolic_opset14 torch.onnx.symbolic_opset15 torch.onnx.symbolic_opset16 torch.onnx.symbolic_opset17 torch.onnx.symbolic_opset18 torch.onnx.symbolic_opset19 torch.onnx.symbolic_opset20 torch.onnx.symbolic_opset7 torch.onnx.symbolic_opset8 torch.onnx.symbolic_opset9 torch.onnx.utils torch.onnx.verification torch.optim torch.optim.adadelta torch.optim.adagrad torch.optim.adam torch.optim.adamax torch.optim.adamw torch.optim.asgd torch.optim.lbfgs torch.optim.lr_scheduler torch.optim.nadam torch.optim.optimizer torch.optim.radam torch.optim.rmsprop torch.optim.rprop torch.optim.sgd torch.optim.sparse_adam torch.optim.swa_utils torch.overrides torch.package torch.package.analyze torch.package.analyze.find_first_use_of_broken_modules torch.package.analyze.is_from_package torch.package.analyze.trace_dependencies torch.package.file_structure_representation torch.package.find_file_dependencies torch.package.glob_group torch.package.importer torch.package.package_exporter torch.package.package_importer torch.profiler torch.profiler.itt torch.profiler.profiler torch.profiler.python_tracer torch.quantization torch.quantization.fake_quantize torch.quantization.fuse_modules torch.quantization.fuser_method_mappings torch.quantization.fx torch.quantization.fx.convert torch.quantization.fx.fuse torch.quantization.fx.fusion_patterns torch.quantization.fx.graph_module torch.quantization.fx.match_utils torch.quantization.fx.pattern_utils torch.quantization.fx.prepare torch.quantization.fx.quantization_patterns torch.quantization.fx.quantization_types torch.quantization.fx.utils torch.quantization.observer torch.quantization.qconfig torch.quantization.quant_type torch.quantization.quantization_mappings torch.quantization.quantize torch.quantization.quantize_fx torch.quantization.quantize_jit torch.quantization.stubs torch.quantization.utils torch.quasirandom torch.random torch.return_types torch.serialization torch.signal torch.signal.windows torch.signal.windows.windows torch.sparse torch.sparse.semi_structured torch.special torch.storage torch.testing torch.torch_version torch.types torch.utils torch.utils.backcompat torch.utils.backend_registration torch.utils.benchmark torch.utils.benchmark.examples torch.utils.benchmark.examples.compare torch.utils.benchmark.examples.fuzzer torch.utils.benchmark.examples.op_benchmark torch.utils.benchmark.examples.simple_timeit torch.utils.benchmark.examples.spectral_ops_fuzz_test torch.utils.benchmark.op_fuzzers torch.utils.benchmark.op_fuzzers.binary torch.utils.benchmark.op_fuzzers.sparse_binary torch.utils.benchmark.op_fuzzers.sparse_unary torch.utils.benchmark.op_fuzzers.spectral torch.utils.benchmark.op_fuzzers.unary torch.utils.benchmark.utils torch.utils.benchmark.utils.common torch.utils.benchmark.utils.compare torch.utils.benchmark.utils.compile torch.utils.benchmark.utils.cpp_jit torch.utils.benchmark.utils.fuzzer torch.utils.benchmark.utils.sparse_fuzzer torch.utils.benchmark.utils.timer torch.utils.benchmark.utils.valgrind_wrapper torch.utils.benchmark.utils.valgrind_wrapper.timer_interface torch.utils.bottleneck torch.utils.bundled_inputs torch.utils.checkpoint torch.utils.collect_env torch.utils.cpp_backtrace torch.utils.cpp_extension torch.utils.data torch.utils.data.backward_compatibility torch.utils.data.dataloader torch.utils.data.datapipes torch.utils.data.datapipes.dataframe torch.utils.data.datapipes.dataframe.dataframe_wrapper torch.utils.data.datapipes.dataframe.dataframes torch.utils.data.datapipes.dataframe.datapipes torch.utils.data.datapipes.dataframe.structures torch.utils.data.datapipes.datapipe torch.utils.data.datapipes.gen_pyi torch.utils.data.datapipes.iter torch.utils.data.datapipes.iter.callable torch.utils.data.datapipes.iter.combinatorics torch.utils.data.datapipes.iter.combining torch.utils.data.datapipes.iter.filelister torch.utils.data.datapipes.iter.fileopener torch.utils.data.datapipes.iter.grouping torch.utils.data.datapipes.iter.routeddecoder torch.utils.data.datapipes.iter.selecting torch.utils.data.datapipes.iter.sharding torch.utils.data.datapipes.iter.streamreader torch.utils.data.datapipes.iter.utils torch.utils.data.datapipes.map torch.utils.data.datapipes.map.callable torch.utils.data.datapipes.map.combinatorics torch.utils.data.datapipes.map.combining torch.utils.data.datapipes.map.grouping torch.utils.data.datapipes.map.utils torch.utils.data.datapipes.utils torch.utils.data.datapipes.utils.common torch.utils.data.datapipes.utils.decoder torch.utils.data.datapipes.utils.snapshot torch.utils.data.dataset torch.utils.data.distributed torch.utils.data.graph torch.utils.data.graph_settings torch.utils.data.sampler torch.utils.deterministic torch.utils.dlpack torch.utils.file_baton torch.utils.flop_counter torch.utils.hipify torch.utils.hipify.constants torch.utils.hipify.cuda_to_hip_mappings torch.utils.hipify.hipify_python torch.utils.hipify.version torch.utils.hooks torch.utils.jit torch.utils.jit.log_extract torch.utils.mkldnn torch.utils.mobile_optimizer torch.utils.model_dump torch.utils.model_zoo torch.utils.module_tracker torch.utils.serialization torch.utils.serialization.config torch.utils.show_pickle torch.utils.tensorboard torch.utils.tensorboard.summary torch.utils.tensorboard.writer torch.utils.throughput_benchmark torch.utils.viz torch.utils.weak torch.version torch.xpu torch.xpu.memory torch.xpu.random torch.xpu.streams Module (class in torch.nn) module (torch.distributed.fsdp.FullyShardedDataParallel property) module() (torch.export.ExportedProgram method) module_load() (torch.Tensor method) ModuleCallEntry (class in torch.export) ModuleCallSignature (class in torch.export) ModuleDict (class in torch.nn) ModuleList (class in torch.nn) modules() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) ModuleTracker (class in torch.utils.module_tracker) monitored_barrier() (in module torch.distributed) move_to_device_pass() (in module torch.export.passes) moveaxis() (in module torch) (torch.Tensor method) movedim() (in module torch) (torch.Tensor method) MovingAverageMinMaxObserver (class in torch.ao.quantization.observer) MovingAveragePerChannelMinMaxObserver (class in torch.ao.quantization.observer) mps() (torch.UntypedStorage method) mse_loss() (in module torch.nn.functional) MSELoss (class in torch.nn) msort() (in module torch) (torch.Tensor method) mT (torch.Tensor attribute) mtia() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) mul() (in module torch) (torch.ao.ns._numeric_suite.Shadow method) (torch.Tensor method) mul_() (torch.Tensor method) mul_scalar() (torch.ao.ns._numeric_suite.Shadow method) multi_dot() (in module torch.linalg) multi_get() (torch.distributed.Store method) multi_margin_loss() (in module torch.nn.functional) multi_set() (torch.distributed.Store method) multigammaln() (in module torch.special) MultiheadAttention (class in torch.ao.nn.quantizable) (class in torch.nn) multilabel_margin_loss() (in module torch.nn.functional) multilabel_soft_margin_loss() (in module torch.nn.functional) MultiLabelMarginLoss (class in torch.nn) MultiLabelSoftMarginLoss (class in torch.nn) MultiMarginLoss (class in torch.nn) Multinomial (class in torch.distributions.multinomial) multinomial (in module torch.distributions.constraints) multinomial() (in module torch) (torch.Tensor method) MultiplicativeLR (class in torch.optim.lr_scheduler) multiply() (in module torch) (torch.Tensor method) multiply_() (torch.Tensor method) MultiprocessContext (class in torch.distributed.elastic.multiprocessing.api) MultiStepLR (class in torch.optim.lr_scheduler) MultivariateNormal (class in torch.distributions.multivariate_normal) mv() (in module torch) (torch.Tensor method) mvlgamma() (in module torch) (torch.Tensor method) mvlgamma_() (torch.Tensor method) N NAdam (class in torch.optim) name (torch.autograd.profiler_util.Kernel attribute) (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend property) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend property) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend property) (torch.distributed.rpc.WorkerInfo property) (torch.monitor.Aggregation property) (torch.monitor.Event property) (torch.monitor.Stat property) (torch.nn.attention.SDPBackend property) (torch.profiler.ProfilerActivity property) (torch.Tag property) name() (torch.autograd.graph.Node method) named_buffers() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_children() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_modules() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) named_parameters() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) NamedShape (torch.nn.Unflatten attribute) names (torch.Tensor attribute) nan_to_num() (in module torch) (torch.Tensor method) nan_to_num_() (torch.Tensor method) nanmean() (in module torch) (torch.Tensor method) nanmedian() (in module torch) (torch.Tensor method) nanquantile() (in module torch) (torch.Tensor method) nansum() (in module torch) (torch.Tensor method) narrow() (in module torch) (in module torch.nested) (torch.Tensor method) narrow_copy() (in module torch) (torch.Tensor method) nbytes (torch.Tensor attribute) nbytes() (torch.TypedStorage method) (torch.UntypedStorage method) ndim (torch.Tensor attribute) ndimension() (torch.Tensor method) ndtr() (in module torch.special) ndtri() (in module torch.special) ne() (in module torch) (torch.Tensor method) ne_() (torch.Tensor method) neg() (in module torch) (torch.Tensor method) neg_() (torch.Tensor method) negative() (in module torch) (torch.Tensor method) negative_() (torch.Tensor method) NegativeBinomial (class in torch.distributions.negative_binomial) nelement() (torch.Tensor method) nested_tensor() (in module torch.nested) nested_tensor_from_jagged() (in module torch.nested) NestedIOFunction (class in torch.autograd.function) new() (torch.UntypedStorage method) new_empty() (torch.Tensor method) new_full() (torch.Tensor method) new_group() (in module torch.distributed) new_ones() (torch.Tensor method) new_tensor() (torch.Tensor method) new_zeros() (torch.Tensor method) next (torch.fx.Node property) next_functions (torch.autograd.graph.Node property) next_rendezvous() (torch.distributed.elastic.rendezvous.RendezvousHandler method) nextafter() (in module torch) (torch.Tensor method) nextafter_() (torch.Tensor method) nll_loss() (in module torch.nn.functional) NLLLoss (class in torch.nn) no_grad (class in torch) no_sync() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.parallel.DistributedDataParallel method) Node (class in torch.fx) node_copy() (torch.fx.Graph method) nodes (torch.fx.Graph property) nonzero() (in module torch) (torch.Tensor method) noop_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks) noop_mask() (in module torch.nn.attention.flex_attention) NoopObserver (class in torch.ao.quantization.observer) norm() (in module torch) (in module torch.linalg) (torch.Tensor method) Normal (class in torch.distributions.normal) normal() (in module torch) normal_() (in module torch.nn.init) (torch.Tensor method) normalize() (in module torch.nn.functional) normalized_arguments() (torch.fx.Node method) not_equal() (in module torch) (torch.Tensor method) not_equal_() (torch.Tensor method) notify_join_context() (torch.distributed.algorithms.Join static method) NSTracer (class in torch.ao.ns._numeric_suite_fx) NullMetricHandler (class in torch.distributed.elastic.metrics.api) num_keys() (torch.distributed.Store method) num_nodes_waiting() (torch.distributed.elastic.rendezvous.RendezvousHandler method) num_worker_threads (torch.distributed.rpc.TensorPipeRpcBackendOptions property) numel() (in module torch) (torch.nn.attention.flex_attention.BlockMask method) (torch.Size method) (torch.Tensor method) NUMERIC_DEBUG_HANDLE_KEY (in module torch.ao.quantization) numpy() (torch.Tensor method) nuttall() (in module torch.signal.windows) O ObservationType (class in torch.ao.quantization.backend_config) ObserverBase (class in torch.ao.quantization.observer) OffloadPolicy (class in torch.distributed.fsdp) on_generate_code() (torch.fx.Graph method) once_differentiable() (in module torch.autograd.function) one_hot() (in module torch.nn.functional) OneCycleLR (class in torch.optim.lr_scheduler) onednn_fusion_enabled() (in module torch.jit) OneHotCategorical (class in torch.distributions.one_hot_categorical) ones() (in module torch) (in module torch.distributed.tensor) ones_() (in module torch.nn.init) ones_like() (in module torch) onnx_compatible() (torch.onnx.JitScalarType method) onnx_type() (torch.onnx.JitScalarType method) OnnxBackend (class in torch.onnx.verification) OnnxExporterError (class in torch.onnx) ONNXProgram (class in torch.onnx) OnnxTestCaseRepro (class in torch.onnx.verification) opcheck() (in module torch.library) optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) optim_state_dict_to_load() (torch.distributed.fsdp.FullyShardedDataParallel static method) optimize() (torch.onnx.ONNXProgram method) optimize_for_inference() (in module torch.jit) optimize_for_mobile() (in module torch.utils.mobile_optimizer) Optimizer (class in torch.optim) OptimStateDictConfig (class in torch.distributed.fsdp) or_masks() (in module torch.nn.attention.flex_attention) orgqr() (in module torch) (torch.Tensor method) ormqr() (in module torch) (torch.Tensor method) orthogonal() (in module torch.nn.utils.parametrizations) orthogonal_() (in module torch.nn.init) outer() (in module torch) (torch.Tensor method) OutOfMemoryError output() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) output_node() (torch.fx.Graph method) OUTPUT_SHARE_OBSERVER_WITH_INPUT (torch.ao.quantization.backend_config.ObservationType attribute) OUTPUT_USE_DIFFERENT_OBSERVER_AS_INPUT (torch.ao.quantization.backend_config.ObservationType attribute) OutputComparisonLogger (class in torch.ao.ns._numeric_suite_fx) OutputKind (class in torch.export.graph_signature) OutputLogger (class in torch.ao.ns._numeric_suite) (class in torch.ao.ns._numeric_suite_fx) OutputSpec (class in torch.export.graph_signature) owner() (torch.distributed.rpc.PyRRef method) owner_name() (torch.distributed.rpc.PyRRef method) P P2POp (class in torch.distributed) pack_padded_sequence() (in module torch.nn.utils.rnn) pack_sequence() (in module torch.nn.utils.rnn) PackageExporter (class in torch.package) PackageImporter (class in torch.package) PackagingError (class in torch.package) PackedSequence (class in torch.nn.utils.rnn) pad() (in module torch.nn.functional) pad_packed_sequence() (in module torch.nn.utils.rnn) pad_sequence() (in module torch.nn.utils.rnn) pairwise_distance() (in module torch.nn.functional) PairwiseDistance (class in torch.nn) parallel_info() (in module torch.__config__) parallelize_module() (in module torch.distributed.tensor.parallel) param_shape (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) Parameter (class in torch.nn.parameter) ParameterDict (class in torch.nn) ParameterList (class in torch.nn) parameters() (in module torch.distributed.GradBucket) (torch.export.ExportedProgram method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) parameters_to_vector() (in module torch.nn.utils) ParametrizationList (class in torch.nn.utils.parametrize) Pareto (class in torch.distributions.pareto) parse_nvprof_trace() (in module torch.autograd.profiler) Partial (class in torch.distributed.tensor.placement_types) path (torch.distributed.FileStore property) path_of_module() (torch.fx.Tracer method) pca_lowrank() (in module torch) PContext (class in torch.distributed.elastic.multiprocessing.api) pdist() (in module torch.nn.functional) per_channel_dynamic_qconfig (in module torch.ao.quantization.qconfig) PerAxis (class in torch.ao.quantization.observer) PerBlock (class in torch.ao.quantization.observer) PerChannelMinMaxObserver (class in torch.ao.quantization.observer) PerGroup (class in torch.ao.quantization.observer) permute() (in module torch) (torch.Tensor method) perplexity() (torch.distributions.distribution.Distribution method) PerRow (class in torch.ao.quantization.observer) PerTensor (class in torch.ao.quantization.observer) PerToken (class in torch.ao.quantization.observer) pickle_storage_type() (torch.TypedStorage method) pin_memory() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) pinv() (in module torch.linalg) pinverse() (in module torch) (torch.Tensor method) Pipe (class in torch.distributed.pipelining) pipe_split() (in module torch.distributed.pipelining) pipeline() (in module torch.distributed.pipelining) PipelineScheduleMulti (class in torch.distributed.pipelining.schedules) PipelineScheduleSingle (class in torch.distributed.pipelining.schedules) PipelineStage (class in torch.distributed.pipelining.stage) pixel_shuffle() (in module torch.nn.functional) pixel_unshuffle() (in module torch.nn.functional) PixelShuffle (class in torch.nn) PixelUnshuffle (class in torch.nn) placeholder() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Graph method) (torch.fx.Interpreter method) (torch.fx.Transformer method) PlaceholderObserver (class in torch.ao.quantization.observer) Placement (class in torch.distributed.tensor.placement_types) placements (torch.distributed.tensor.DTensor property) Poisson (class in torch.distributions.poisson) poisson() (in module torch) poisson_nll_loss() (in module torch.nn.functional) PoissonNLLLoss (class in torch.nn) polar() (in module torch) polygamma() (in module torch) (in module torch.special) (torch.Tensor method) polygamma_() (torch.Tensor method) PolynomialLR (class in torch.optim.lr_scheduler) pool() (torch.cuda.CUDAGraph method) pop() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) popitem() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) port (torch.distributed.TCPStore property) positive() (in module torch) (torch.Tensor method) PositiveDefiniteTransform (class in torch.distributions.transforms) post_hook() (torch.distributed.algorithms.JoinHook method) PostLocalSGDOptimizer (class in torch.distributed.optim) pow() (in module torch) (torch.Tensor method) pow_() (torch.Tensor method) power_draw() (in module torch.cuda) powerSGD_hook() (in module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerSGDState (class in torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerTransform (class in torch.distributions.transforms) precision_matrix (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) predict() (torch.nn.AdaptiveLogSoftmaxWithLoss method) preferred_blas_library() (in module torch.backends.cuda) preferred_linalg_library() (in module torch.backends.cuda) preferred_rocm_fa_library() (in module torch.backends.cuda) PrefixStore (class in torch.distributed) PReLU (class in torch.nn) prelu() (in module torch.nn.functional) prepare (class in torch.ao.quantization) prepare_for_propagation_comparison (class in torch.ao.quantization) prepare_fx (class in torch.ao.quantization.quantize_fx) prepare_global_plan() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) prepare_local_plan() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) prepare_model_outputs() (in module torch.ao.ns._numeric_suite) prepare_model_with_stubs() (in module torch.ao.ns._numeric_suite) prepare_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) prepare_qat (class in torch.ao.quantization) prepare_qat_fx (class in torch.ao.quantization.quantize_fx) PrepareCustomConfig (class in torch.ao.quantization.fx.custom_config) PrepareModuleInput (class in torch.distributed.tensor.parallel) PrepareModuleInputOutput (class in torch.distributed.tensor.parallel) PrepareModuleOutput (class in torch.distributed.tensor.parallel) prepend() (torch.fx.Node method) preset_metadata_json() (torch.profiler._KinetoProfile method) prettify_results() (torch.fx.experimental.symbolic_shapes.DimConstraints method) prev (torch.fx.Node property) primal (torch.autograd.forward_ad.UnpackedDualTensor attribute) print() (torch.utils.benchmark.Compare method) print_comparisons_n_shadows_model() (in module torch.ao.ns._numeric_suite_fx) print_readable() (torch.fx.GraphModule method) print_tabular() (torch.fx.Graph method) probs (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.geometric.Geometric property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli property) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) process_inputs() (torch.fx.Graph method) process_outputs() (torch.fx.Graph method) ProcessFailure (class in torch.distributed.elastic.multiprocessing.errors) prod() (in module torch) (torch.Tensor method) produce_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) produce_guards_expression() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) produce_guards_verbose() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) prof() (in module torch.distributed.elastic.metrics) profile (class in torch.autograd.profiler) (class in torch.profiler) profile() (in module torch.mps.profiler) ProfilerAction (class in torch.profiler) ProfilerActivity (class in torch.profiler) promote_types() (in module torch) propagate_qconfig_ (class in torch.ao.quantization) PropagateUnbackedSymInts (class in torch.fx.experimental.symbolic_shapes) Proxy (class in torch.fx) proxy() (torch.fx.Tracer method) prune() (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) PruningContainer (class in torch.nn.utils.prune) psi() (in module torch.special) put_() (torch.Tensor method) put_metric() (in module torch.distributed.elastic.metrics) PyRRef (class in torch.distributed.rpc) python_code() (torch.fx.Graph method) python_version() (torch.package.PackageImporter method) Q q_indices (torch.nn.attention.flex_attention.BlockMask attribute) q_num_blocks (torch.nn.attention.flex_attention.BlockMask attribute) q_per_channel_axis() (torch.Tensor method) q_per_channel_scales() (torch.Tensor method) q_per_channel_zero_points() (torch.Tensor method) q_scale() (torch.Tensor method) q_zero_point() (torch.Tensor method) QConfig (class in torch.ao.quantization.qconfig) QConfigMapping (class in torch.ao.quantization.qconfig_mapping) QFunctional (class in torch.ao.nn.quantized) QInt32Storage (class in torch) QInt8Storage (class in torch) qr() (in module torch) (in module torch.linalg) (torch.Tensor method) qscheme() (torch.Tensor method) quantile() (in module torch) (torch.Tensor method) quantize (class in torch.ao.quantization) quantize_dynamic (class in torch.ao.quantization) quantize_per_channel() (in module torch) quantize_per_tensor() (in module torch) quantize_qat (class in torch.ao.quantization) quantized_batch_norm() (in module torch) quantized_max_pool1d() (in module torch) quantized_max_pool2d() (in module torch) QuantStub (class in torch.ao.quantization) QuantWrapper (class in torch.ao.quantization) query() (torch.cuda.Event method) (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Event method) (torch.xpu.Stream method) queue_len() (torch.distributed.Store method) queue_pop() (torch.distributed.Store method) queue_push() (torch.distributed.Store method) QUInt2x4Storage (class in torch) QUInt4x2Storage (class in torch) QUInt8Storage (class in torch) R rad2deg() (in module torch) (torch.Tensor method) RAdam (class in torch.optim) rand() (in module torch) (in module torch.distributed.tensor) rand_like() (in module torch) randint() (in module torch) randint_like() (in module torch) randn() (in module torch) (in module torch.distributed.tensor) randn_like() (in module torch) random_() (torch.Tensor method) random_split() (in module torch.utils.data) random_structured() (in module torch.nn.utils.prune) random_unstructured() (in module torch.nn.utils.prune) RandomSampler (class in torch.utils.data) RandomStructured (class in torch.nn.utils.prune) RandomUnstructured (class in torch.nn.utils.prune) randperm() (in module torch) range() (in module torch) (in module torch.cuda.nvtx) range_pop() (in module torch.cuda.nvtx) (in module torch.profiler.itt) range_push() (in module torch.cuda.nvtx) (in module torch.profiler.itt) rate (torch.distributions.inverse_gamma.InverseGamma property) ravel() (in module torch) (torch.Tensor method) read_data() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) read_file() (in module torch.cuda.tunable) read_metadata() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) ReadItem (class in torch.distributed.checkpoint) real (torch.Tensor attribute) real() (in module torch) rebind_unbacked() (in module torch.fx.experimental.symbolic_shapes) reciprocal() (in module torch) (torch.Tensor method) reciprocal_() (torch.Tensor method) recommended_max_memory() (in module torch.mps) recompile() (torch.fx.GraphModule method) record() (in module torch.distributed.elastic.events) (in module torch.distributed.elastic.multiprocessing.errors) (torch.cuda.Event method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) record_event() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) record_function (class in torch.autograd.profiler) record_memory_history() (in module torch.mtia) record_stream() (torch.Tensor method) record_untuned_enable() (in module torch.cuda.tunable) record_untuned_is_enabled() (in module torch.cuda.tunable) RecordingObserver (class in torch.ao.quantization.observer) recursive_undo() (torch.optim.lr_scheduler.SequentialLR method) recv() (in module torch.distributed) recv_object_list() (in module torch.distributed) redistribute() (torch.distributed.tensor.DTensor method) reduce() (in module torch.distributed) reduce_add() (in module torch.cuda.comm) reduce_add_coalesced() (in module torch.cuda.comm) reduce_op (class in torch.distributed) (torch.distributed.tensor.placement_types.Partial attribute) reduce_scatter() (in module torch.distributed) reduce_scatter_tensor() (in module torch.distributed) ReduceLROnPlateau (class in torch.optim.lr_scheduler) ReduceOp (class in torch.distributed) refine_dynamic_shapes_from_suggested_fixes() (in module torch.export.dynamic_shapes) refine_names() (torch.Tensor method) ReflectionPad1d (class in torch.nn) ReflectionPad2d (class in torch.nn) ReflectionPad3d (class in torch.nn) register() (torch.distributions.constraint_registry.ConstraintRegistry method) register_autocast() (in module torch.library) register_autograd() (in module torch.library) register_backend() (torch.distributed.Backend class method) register_backward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_buffer() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_comm_hook() (torch.distributed.fsdp.FullyShardedDataParallel method) (torch.nn.parallel.DistributedDataParallel method) register_custom_op_symbolic() (in module torch.onnx) register_dataclass() (in module torch.export) register_event_handler() (in module torch.monitor) register_extern_hook() (torch.package.PackageExporter method) register_fake() (in module torch.library) register_forward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_forward_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_fsdp_forward_method() (in module torch.distributed.fsdp) register_full_backward_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_full_backward_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_handle() (torch.cuda.gds.GdsFile method) register_hook() (torch.autograd.graph.Node method) (torch.Tensor method) register_intern_hook() (torch.package.PackageExporter method) register_kernel() (in module torch.library) register_kl() (in module torch.distributions.kl) register_load_state_dict_post_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_load_state_dict_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_mock_hook() (torch.package.PackageExporter method) register_module() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_module_backward_hook() (in module torch.nn.modules.module) register_module_buffer_registration_hook() (in module torch.nn.modules.module) register_module_forward_hook() (in module torch.nn.modules.module) register_module_forward_pre_hook() (in module torch.nn.modules.module) register_module_full_backward_hook() (in module torch.nn.modules.module) register_module_full_backward_pre_hook() (in module torch.nn.modules.module) register_module_module_registration_hook() (in module torch.nn.modules.module) register_module_parameter_registration_hook() (in module torch.nn.modules.module) register_multi_grad_hook (class in torch.autograd.graph) register_package() (in module torch.serialization) register_parameter() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) register_parametrization() (in module torch.nn.utils.parametrize) register_post_accumulate_grad_hook() (torch.Tensor method) register_prehook() (torch.autograd.graph.Node method) register_sharding() (in module torch.distributed.tensor.experimental) register_state_dict_post_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_state_dict_pre_hook() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) register_step_post_hook() (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) register_step_pre_hook() (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) register_timers() (torch.distributed.elastic.timer.TimerServer method) register_torch_dispatch() (in module torch.library) register_vmap() (in module torch.library) reify() (torch.distributed.elastic.multiprocessing.api.DefaultLogsSpecs method) (torch.distributed.elastic.multiprocessing.api.LogsSpecs method) rekey_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) RelaxedBernoulli (class in torch.distributions.relaxed_bernoulli) RelaxedOneHotCategorical (class in torch.distributions.relaxed_categorical) RelaxedUnspecConstraint (class in torch.fx.experimental.symbolic_shapes) release() (torch.distributed.elastic.timer.TimerClient method) (torch.onnx.ONNXProgram method) ReLU (class in torch.nn) relu() (in module torch.nn.functional) ReLU6 (class in torch.ao.nn.quantized) (class in torch.nn) relu6() (in module torch.nn.functional) relu_() (in module torch.nn.functional) remainder() (in module torch) (torch.Tensor method) remainder_() (torch.Tensor method) remote() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) remote_parameters() (torch.distributed.nn.api.remote_module.RemoteModule method) RemoteModule (class in torch.distributed.nn.api.remote_module) remove() (in module torch.nn.utils.prune) (torch.nn.utils.prune.BasePruningMethod method) (torch.nn.utils.prune.CustomFromMask method) (torch.nn.utils.prune.Identity method) (torch.nn.utils.prune.L1Unstructured method) (torch.nn.utils.prune.LnStructured method) (torch.nn.utils.prune.PruningContainer method) (torch.nn.utils.prune.RandomStructured method) (torch.nn.utils.prune.RandomUnstructured method) remove_parametrizations() (in module torch.nn.utils.parametrize) remove_spectral_norm() (in module torch.nn.utils) remove_weight_norm() (in module torch.nn.utils) rename() (torch.Tensor method) rename_() (torch.Tensor method) rename_privateuse1_backend() (in module torch.utils) render() (torch.fx.experimental.symbolic_shapes.StrictMinMaxConstraint method) RendezvousBackend (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) RendezvousClosedError (class in torch.distributed.elastic.rendezvous.api) RendezvousConnectionError (class in torch.distributed.elastic.rendezvous.api) RendezvousError (class in torch.distributed.elastic.rendezvous.api) RendezvousGracefulExitError (class in torch.distributed.elastic.rendezvous.api) RendezvousHandler (class in torch.distributed.elastic.rendezvous) RendezvousHandlerRegistry (class in torch.distributed.elastic.rendezvous) RendezvousInfo (class in torch.distributed.elastic.rendezvous) RendezvousParameters (class in torch.distributed.elastic.rendezvous) RendezvousStateError (class in torch.distributed.elastic.rendezvous.api) RendezvousStoreInfo (class in torch.distributed.elastic.rendezvous.api) RendezvousTimeout (class in torch.distributed.elastic.rendezvous.dynamic_rendezvous) RendezvousTimeoutError (class in torch.distributed.elastic.rendezvous.api) renorm() (in module torch) (torch.Tensor method) renorm_() (torch.Tensor method) repeat() (torch.Tensor method) repeat_interleave() (in module torch) (torch.Tensor method) replace() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) replace_all_batch_norm_modules_() (in module torch.func) replace_all_uses() (torch.export.graph_signature.ExportGraphSignature method) replace_all_uses_with() (torch.fx.Node method) replace_input_with() (torch.fx.Node method) replace_pattern() (in module torch.fx) replay() (torch.cuda.CUDAGraph method) Replicate (class in torch.distributed.tensor.placement_types) ReplicationPad1d (class in torch.nn) ReplicationPad2d (class in torch.nn) ReplicationPad3d (class in torch.nn) requires_grad (torch.Tensor attribute) requires_grad_() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) reset() (in module torch.compiler) (torch.cuda.CUDAGraph method) (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) (torch.distributed.checkpoint.StorageWriter method) (torch.quasirandom.SobolEngine method) reset_accumulated_memory_stats() (in module torch.xpu) reset_max_memory_allocated() (in module torch.cuda) reset_max_memory_cached() (in module torch.cuda) reset_min_max_vals() (torch.ao.quantization.observer.MinMaxObserver method) (torch.ao.quantization.observer.PerChannelMinMaxObserver method) reset_parameters() (torch.nn.modules.normalization.RMSNorm method) (torch.nn.RMSNorm method) reset_peak_host_memory_stats() (in module torch.cuda) reset_peak_memory_stats() (in module torch.cuda) (in module torch.xpu) reshape() (in module torch) (torch.Tensor method) reshape_as() (torch.Tensor method) ReshapeTransform (class in torch.distributions.transforms) reshard() (torch.distributed.fsdp.FSDPModule method) resizable() (torch.TypedStorage method) (torch.UntypedStorage method) resize_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) resize_as_() (torch.Tensor method) resolve_bytes() (torch.distributed.checkpoint.LoadPlanner method) resolve_conj() (in module torch) (torch.Tensor method) resolve_data() (torch.distributed.checkpoint.SavePlanner method) resolve_name() (in module torch.overrides) resolve_neg() (in module torch) (torch.Tensor method) resolve_tensor() (torch.distributed.checkpoint.LoadPlanner method) resolve_unbacked_bindings() (in module torch.fx.experimental.symbolic_shapes) result() (torch.distributed.Work method) result_type() (in module torch) retain_grad() (torch.Tensor method) retains_grad (torch.Tensor attribute) rewrite_with_congruences() (torch.fx.experimental.symbolic_shapes.DimConstraints method) rfft() (in module torch.fft) rfft2() (in module torch.fft) rfftfreq() (in module torch.fft) rfftn() (in module torch.fft) right_inverse() (torch.nn.utils.parametrize.ParametrizationList method) rms_norm() (in module torch.nn.functional) RMSNorm (class in torch.nn) (class in torch.nn.modules.normalization) RMSprop (class in torch.optim) RNN (class in torch.nn) RNNBase (class in torch.nn) RNNCell (class in torch.ao.nn.quantized.dynamic) (class in torch.nn) roll() (in module torch) (torch.Tensor method) rot90() (in module torch) (torch.Tensor method) round() (in module torch) (in module torch.special) (torch.Tensor method) round_() (torch.Tensor method) row_indices() (torch.Tensor method) row_stack() (in module torch) RowwiseParallel (class in torch.distributed.tensor.parallel) rpc_async() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) rpc_sync() (in module torch.distributed.rpc) (torch.distributed.rpc.PyRRef method) rpc_timeout (torch.distributed.rpc.RpcBackendOptions property) (torch.distributed.rpc.TensorPipeRpcBackendOptions property) RpcBackendOptions (class in torch.distributed.rpc) Rprop (class in torch.optim) RReLU (class in torch.nn) rrelu() (in module torch.nn.functional) rrelu_() (in module torch.nn.functional) rsample() (torch.distributions.beta.Beta method) (torch.distributions.cauchy.Cauchy method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.dirichlet.Dirichlet method) (torch.distributions.distribution.Distribution method) (torch.distributions.exponential.Exponential method) (torch.distributions.fishersnedecor.FisherSnedecor method) (torch.distributions.gamma.Gamma method) (torch.distributions.generalized_pareto.GeneralizedPareto method) (torch.distributions.independent.Independent method) (torch.distributions.laplace.Laplace method) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal method) (torch.distributions.multivariate_normal.MultivariateNormal method) (torch.distributions.normal.Normal method) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli method) (torch.distributions.studentT.StudentT method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.uniform.Uniform method) (torch.distributions.wishart.Wishart method) rsqrt() (in module torch) (torch.Tensor method) rsqrt_() (torch.Tensor method) run() (torch.distributed.elastic.agent.server.ElasticAgent method) (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) run_decompositions() (torch.export.ExportedProgram method) run_node() (torch.fx.experimental.symbolic_shapes.PropagateUnbackedSymInts method) (torch.fx.Interpreter method) RunProcsResult (class in torch.distributed.elastic.multiprocessing.api) RunResult (class in torch.distributed.elastic.agent.server.api) S safe_globals (class in torch.serialization) sample() (torch.distributions.bernoulli.Bernoulli method) (torch.distributions.binomial.Binomial method) (torch.distributions.categorical.Categorical method) (torch.distributions.continuous_bernoulli.ContinuousBernoulli method) (torch.distributions.distribution.Distribution method) (torch.distributions.geometric.Geometric method) (torch.distributions.independent.Independent method) (torch.distributions.lkj_cholesky.LKJCholesky method) (torch.distributions.mixture_same_family.MixtureSameFamily method) (torch.distributions.multinomial.Multinomial method) (torch.distributions.negative_binomial.NegativeBinomial method) (torch.distributions.normal.Normal method) (torch.distributions.one_hot_categorical.OneHotCategorical method) (torch.distributions.poisson.Poisson method) (torch.distributions.transformed_distribution.TransformedDistribution method) (torch.distributions.von_mises.VonMises method) sample_n() (torch.distributions.distribution.Distribution method) sampled_addmm() (in module torch.sparse) Sampler (class in torch.utils.data) save() (in module torch) (in module torch.distributed.checkpoint.state_dict_saver) (in module torch.export) (in module torch.jit) (torch.jit.ScriptFunction method) (torch.jit.ScriptModule method) (torch.onnx.ONNXProgram method) save_binary() (torch.package.PackageExporter method) save_for_backward() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) save_for_forward() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) save_module() (torch.package.PackageExporter method) save_on_cpu (class in torch.autograd.graph) save_pickle() (torch.package.PackageExporter method) save_source_file() (torch.package.PackageExporter method) save_source_string() (torch.package.PackageExporter method) save_state_dict() (in module torch.distributed.checkpoint.state_dict_saver) save_storage() (torch.cuda.gds.GdsFile method) save_text() (torch.package.PackageExporter method) save_to_buffer() (torch.jit.ScriptFunction method) saved_tensors (torch.autograd.function.NestedIOFunction property) saved_tensors_hooks (class in torch.autograd.graph) SavePlan (class in torch.distributed.checkpoint) SavePlanner (class in torch.distributed.checkpoint) scalar_name() (torch.onnx.JitScalarType method) scale (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.log_normal.LogNormal property) scale_fn() (torch.optim.lr_scheduler.CyclicLR method) scale_tril (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.wishart.Wishart property) scaled_dot_product_attention() (in module torch.nn.functional) scaled_modified_bessel_k0() (in module torch.special) scaled_modified_bessel_k1() (in module torch.special) scatter() (in module torch) (in module torch.cuda.comm) (in module torch.distributed) (torch.Tensor method) scatter_() (torch.Tensor method) scatter_add() (in module torch) (torch.Tensor method) scatter_add_() (torch.Tensor method) scatter_full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) scatter_object_list() (in module torch.distributed) scatter_reduce() (in module torch) (torch.Tensor method) scatter_reduce_() (torch.Tensor method) schedule() (in module torch.profiler) Schedule1F1B (class in torch.distributed.pipelining.schedules) ScheduleGPipe (class in torch.distributed.pipelining.schedules) ScheduleInterleaved1F1B (class in torch.distributed.pipelining.schedules) ScheduleInterleavedZeroBubble (class in torch.distributed.pipelining.schedules) ScheduleLoopedBFS (class in torch.distributed.pipelining.schedules) ScheduleZBVZeroBubble (class in torch.distributed.pipelining.schedules) script() (in module torch.jit) script_if_tracing() (in module torch.jit) ScriptFunction (class in torch.jit) ScriptModule (class in torch.jit) sdp_kernel() (in module torch.backends.cuda) sdpa_kernel() (in module torch.nn.attention) SDPAParams (class in torch.backends.cuda) SDPBackend (class in torch.nn.attention) searchsorted() (in module torch) see() (torch.autograd.profiler.EnforceUnique method) seed() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.random) (in module torch.xpu) (torch.Generator method) seed_all() (in module torch.cuda) (in module torch.xpu) select() (in module torch) (torch.Tensor method) select_model_mode_for_export() (in module torch.onnx) select_scatter() (in module torch) (torch.Tensor method) SelectiveCheckpointContext (class in torch.utils.checkpoint) self_cpu_time_total (torch.autograd.profiler.profile property) SELU (class in torch.nn) selu() (in module torch.nn.functional) send() (in module torch.distributed) send_object_list() (in module torch.distributed) seq_lengths (torch.nn.attention.flex_attention.BlockMask attribute) SequenceParallel (class in torch.distributed.tensor.parallel) Sequential (class in torch.nn) SequentialLR (class in torch.optim.lr_scheduler) SequentialSampler (class in torch.utils.data) set() (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.Store method) set_() (torch.Tensor method) set_all_reduce_hook() (torch.distributed.fsdp.FSDPModule method) set_backend_pattern_config() (torch.ao.quantization.backend_config.BackendConfig method) set_backend_pattern_configs() (torch.ao.quantization.backend_config.BackendConfig method) set_buffer() (in module torch.distributed.GradBucket) set_checkpoint_debug_enabled() (in module torch.utils.checkpoint) set_closed() (torch.distributed.elastic.rendezvous.RendezvousHandler method) set_codegen() (torch.fx.Graph method) set_crc32_options() (in module torch.serialization) set_custom_trace_id_callback() (torch.profiler.profile method) set_default_device() (in module torch) set_default_dtype() (in module torch) set_default_load_endianness() (in module torch.serialization) set_default_mmap_options() (in module torch.serialization) set_default_tensor_type() (in module torch) set_default_validate_args() (torch.distributions.distribution.Distribution static method) set_detect_anomaly (class in torch.autograd) set_deterministic_debug_mode() (in module torch) set_device() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) set_device_idx() (in module torch.accelerator) set_device_index() (in module torch.accelerator) set_device_map() (torch.distributed.rpc.TensorPipeRpcBackendOptions method) set_devices() (torch.distributed.rpc.TensorPipeRpcBackendOptions method) set_dir() (in module torch.hub) set_dtype_configs() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_exception() (torch.futures.Future method) set_extra_state() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) set_fastpath_enabled() (in module torch.backends.mha) set_filename() (in module torch.cuda.tunable) set_flags() (in module torch.backends.nnpack) set_float32_matmul_precision() (in module torch) set_float_to_observed_mapping() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_flush_denormal() (in module torch) set_fused_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_fuser_method() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_fusion_strategy() (in module torch.jit) set_global() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_grad_enabled (class in torch.autograd.grad_mode) set_input_quantized_indexes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_is_last_backward() (torch.distributed.fsdp.FSDPModule method) set_kernel_enabled() (torch._library.custom_ops.CustomOpDef method) set_logs() (in module torch._logging) set_materialize_grads() (torch.autograd.function.BackwardCFunction method) (torch.autograd.function.FunctionCtx method) (torch.autograd.function.InplaceFunction method) (torch.autograd.function.NestedIOFunction method) set_max_tuning_duration() (in module torch.cuda.tunable) set_max_tuning_iterations() (in module torch.cuda.tunable) set_model_state_dict() (in module torch.distributed.checkpoint.state_dict) set_module() (in module torch.utils) set_module_name() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_module_name_object_type_order() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_module_name_regex() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_modules_to_backward_prefetch() (torch.distributed.fsdp.FSDPModule method) set_modules_to_forward_prefetch() (torch.distributed.fsdp.FSDPModule method) set_multithreading_enabled (class in torch.autograd.grad_mode) set_name() (torch.ao.quantization.backend_config.BackendConfig method) set_non_traceable_module_classes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_non_traceable_module_names() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_num_interop_threads() (in module torch) set_num_threads() (in module torch) set_object_type() (torch.ao.quantization.qconfig_mapping.QConfigMapping method) set_observation_type() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_observed_to_quantized_mapping() (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) set_optimizer_state_dict() (in module torch.distributed.checkpoint.state_dict) set_output_quantized_indexes() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_overwrite_module_params_on_conversion() (in module torch.__future__) set_pattern() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_per_process_memory_fraction() (in module torch.cuda) (in module torch.mps) set_post_optim_event() (torch.distributed.fsdp.FSDPModule method) set_preserved_attributes() (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_printoptions() (in module torch) set_qat_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_reduce_scatter_divide_factor() (torch.distributed.fsdp.FSDPModule method) set_reference_quantized_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_requires_all_reduce() (torch.distributed.fsdp.FSDPModule method) set_requires_gradient_sync() (torch.distributed.fsdp.FSDPModule method) set_reshard_after_backward() (torch.distributed.fsdp.FSDPModule method) set_reshard_after_forward() (torch.distributed.fsdp.FSDPModule method) set_result() (torch.futures.Future method) set_rng_state() (in module torch) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.random) (in module torch.xpu) set_rng_state_all() (in module torch.cuda) (in module torch.xpu) set_root_module() (torch.ao.quantization.backend_config.BackendPatternConfig method) set_rotating_buffer_size() (in module torch.cuda.tunable) set_sharing_strategy() (in module torch.multiprocessing) set_stance() (in module torch.compiler) set_standalone_module_class() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_standalone_module_name() (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) set_state() (torch.distributed.elastic.rendezvous.c10d_rendezvous_backend.C10dRendezvousBackend method) (torch.distributed.elastic.rendezvous.dynamic_rendezvous.RendezvousBackend method) (torch.distributed.elastic.rendezvous.etcd_rendezvous_backend.EtcdRendezvousBackend method) (torch.Generator method) set_state_dict() (in module torch.distributed.checkpoint.state_dict) set_state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) set_stream() (in module torch.accelerator) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) set_submodule() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) set_swap_module_params_on_conversion() (in module torch.__future__) set_sync_debug_mode() (in module torch.cuda) set_timeout() (torch.distributed.Store method) set_unbacked_var_to_val() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) set_unshard_in_backward() (torch.distributed.fsdp.FSDPModule method) set_up_planner() (torch.distributed.checkpoint.format_utils.DynamicMetaLoadPlanner method) (torch.distributed.checkpoint.LoadPlanner method) (torch.distributed.checkpoint.SavePlanner method) set_up_storage_reader() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader method) (torch.distributed.checkpoint.StorageReader method) set_up_storage_writer() (torch.distributed.checkpoint.StorageWriter method) set_warn_always() (in module torch) setdefault() (torch.autograd.profiler_util.StringTable method) (torch.nn.ParameterDict method) setup_context() (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) SGD (class in torch.optim) sgn() (in module torch) (torch.Tensor method) sgn_() (torch.Tensor method) Shadow (class in torch.ao.ns._numeric_suite) ShadowLogger (class in torch.ao.ns._numeric_suite) shape (torch.nn.attention.flex_attention.BlockMask property) (torch.Tensor attribute) ShapeEnv (class in torch.fx.experimental.symbolic_shapes) ShapeEnvSettings (class in torch.fx.experimental.symbolic_shapes) ShapesCollection (class in torch.export.dynamic_shapes) Shard (class in torch.distributed.tensor.placement_types) shard_full_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) sharded_optim_state_dict() (torch.distributed.fsdp.FullyShardedDataParallel static method) ShardedOptimStateDictConfig (class in torch.distributed.fsdp) ShardedStateDictConfig (class in torch.distributed.fsdp) ShardingStrategy (class in torch.distributed.fsdp) share_memory() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) share_memory_() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) short() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) ShortStorage (class in torch) should_synchronize_after_execute (torch.distributed.checkpoint.staging.AsyncStager property) show() (in module torch.__config__) shutdown() (in module torch.distributed.rpc) (torch.distributed.elastic.rendezvous.RendezvousHandler method) Sigmoid (class in torch.ao.nn.quantized) (class in torch.nn) sigmoid() (in module torch) (in module torch.nn.functional) (torch.Tensor method) sigmoid_() (torch.Tensor method) SigmoidTransform (class in torch.distributions.transforms) sign (torch.distributions.transforms.Transform property) sign() (in module torch) (torch.Tensor method) sign_() (torch.Tensor method) signbit() (in module torch) (torch.Tensor method) significant_figures (torch.utils.benchmark.Measurement property) SiLU (class in torch.nn) silu() (in module torch.nn.functional) SimpleElasticAgent (class in torch.distributed.elastic.agent.server) simplify() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) sin() (in module torch) (torch.Tensor method) sin_() (torch.Tensor method) sinc() (in module torch) (in module torch.special) (torch.Tensor method) sinc_() (torch.Tensor method) sinh() (in module torch) (torch.Tensor method) sinh_() (torch.Tensor method) Size (class in torch) size (in module torch.backends.cuda.cufft_plan_cache) size() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) size_hint() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) skip_data (class in torch.serialization) skip_init() (in module torch.nn.utils) slice_scatter() (in module torch) (torch.Tensor method) slogdet() (in module torch) (in module torch.linalg) (torch.Tensor method) smm() (in module torch) (torch.Tensor method) smooth_l1_loss() (in module torch.nn.functional) SmoothL1Loss (class in torch.nn) snapshot() (in module torch.mtia) (torch.cuda.MemPool method) SobolEngine (class in torch.quasirandom) soft_margin_loss() (in module torch.nn.functional) SoftMarginLoss (class in torch.nn) Softmax (class in torch.nn) softmax() (in module torch) (in module torch.nn.functional) (in module torch.sparse) (in module torch.special) (torch.Tensor method) Softmax2d (class in torch.nn) SoftmaxTransform (class in torch.distributions.transforms) Softmin (class in torch.nn) softmin() (in module torch.nn.functional) Softplus (class in torch.nn) softplus() (in module torch.nn.functional) SoftplusTransform (class in torch.distributions.transforms) Softshrink (class in torch.nn) softshrink() (in module torch.nn.functional) Softsign (class in torch.nn) softsign() (in module torch.nn.functional) solve() (in module torch.linalg) (torch.fx.experimental.symbolic_shapes.DimConstraints method) solve_ex() (in module torch.linalg) solve_triangular() (in module torch.linalg) sort() (in module torch) (torch.Tensor method) sorted_indices (torch.nn.utils.rnn.PackedSequence attribute) source_rank() (torch.distributed.Work method) sparse_() (in module torch.nn.init) sparse_bsc_tensor() (in module torch) sparse_bsr_tensor() (in module torch) sparse_compressed_tensor() (in module torch) sparse_coo_tensor() (in module torch) sparse_csc_tensor() (in module torch) sparse_csr_tensor() (in module torch) sparse_dim() (torch.Tensor method) sparse_mask() (torch.Tensor method) sparse_resize_() (torch.Tensor method) sparse_resize_and_clear_() (torch.Tensor method) SparseAdam (class in torch.optim) sparsity() (torch.nn.attention.flex_attention.BlockMask method) spawn() (in module torch.multiprocessing.spawn) SpawnContext (class in torch.multiprocessing) spdiags() (in module torch.sparse) spectral_norm() (in module torch.nn.utils) (in module torch.nn.utils.parametrizations) spherical_bessel_j0() (in module torch.special) split() (in module torch) (torch.Tensor method) split_args_kwargs_into_chunks() (in module torch.distributed.pipelining.microbatch) SplitPoint (class in torch.distributed.pipelining) spsolve() (in module torch.sparse) sqrt() (in module torch) (torch.Tensor method) sqrt_() (torch.Tensor method) square() (in module torch) (torch.Tensor method) square_() (torch.Tensor method) squeeze() (in module torch) (torch.Tensor method) squeeze_() (torch.Tensor method) sspaddmm() (in module torch) (torch.Tensor method) stack (in module torch.distributions.constraints) stack() (in module torch) stack_module_state() (in module torch.func) stack_trace (torch.fx.Node property) StackDataset (class in torch.utils.data) StackTransform (class in torch.distributions.transforms) stage() (torch.distributed.checkpoint.FileSystemWriter method) (torch.distributed.checkpoint.staging.AsyncStager method) (torch.distributed.checkpoint.staging.BlockingAsyncStager method) StandaloneModuleConfigEntry (class in torch.ao.quantization.fx.custom_config) start() (in module torch.mps.profiler) (torch.distributed.elastic.agent.server.health_check_server.HealthCheckServer method) start_processes() (in module torch.distributed.elastic.multiprocessing) Stat (class in torch.monitor) state_dict() (torch.distributed.checkpoint.stateful.Stateful method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) (torch.optim.swa_utils.SWALR method) state_dict_type() (torch.distributed.fsdp.FullyShardedDataParallel static method) StateDictConfig (class in torch.distributed.fsdp) StateDictOptions (class in torch.distributed.checkpoint.state_dict) StateDictSettings (class in torch.distributed.fsdp) Stateful (class in torch.distributed.checkpoint.stateful) StatefulSymbolicContext (class in torch.fx.experimental.symbolic_shapes) StatelessSymbolicContext (class in torch.fx.experimental.symbolic_shapes) statically_known_true() (in module torch.fx.experimental.symbolic_shapes) stats() (torch.utils.benchmark.CallgrindStats method) std() (in module torch) (torch.Tensor method) std_mean() (in module torch) stddev (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.laplace.Laplace property) (torch.distributions.normal.Normal property) (torch.distributions.uniform.Uniform property) step() (torch.distributed.optim.DistributedOptimizer method) (torch.distributed.optim.PostLocalSGDOptimizer method) (torch.distributed.optim.ZeroRedundancyOptimizer method) (torch.distributed.pipelining.schedules.PipelineScheduleMulti method) (torch.distributed.pipelining.schedules.PipelineScheduleSingle method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.lr_scheduler.ChainedScheduler method) (torch.optim.lr_scheduler.ConstantLR method) (torch.optim.lr_scheduler.CosineAnnealingLR method) (torch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (torch.optim.lr_scheduler.CyclicLR method) (torch.optim.lr_scheduler.ExponentialLR method) (torch.optim.lr_scheduler.LambdaLR method) (torch.optim.lr_scheduler.LinearLR method) (torch.optim.lr_scheduler.LRScheduler method) (torch.optim.lr_scheduler.MultiplicativeLR method) (torch.optim.lr_scheduler.MultiStepLR method) (torch.optim.lr_scheduler.OneCycleLR method) (torch.optim.lr_scheduler.PolynomialLR method) (torch.optim.lr_scheduler.ReduceLROnPlateau method) (torch.optim.lr_scheduler.SequentialLR method) (torch.optim.lr_scheduler.StepLR method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.SWALR method) (torch.profiler.profile method) StepLR (class in torch.optim.lr_scheduler) stft() (in module torch) (torch.Tensor method) StickBreakingTransform (class in torch.distributions.transforms) stop() (in module torch.mps.profiler) (torch.distributed.elastic.agent.server.health_check_server.HealthCheckServer method) storage() (torch.Tensor method) storage_meta() (torch.distributed.checkpoint.StorageWriter method) storage_offset() (torch.Tensor method) storage_type() (torch.Tensor method) StorageReader (class in torch.distributed.checkpoint) StorageWriter (class in torch.distributed.checkpoint) Store (class in torch.distributed) strategy (in module torch.backends.opt_einsum) Stream (class in torch) (class in torch.cpu) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) stream() (in module torch.cpu) (in module torch.cuda) (in module torch.mtia) (in module torch.xpu) StreamContext (class in torch.cpu) (class in torch.cuda) (class in torch.mtia) (class in torch.xpu) strict_fusion (class in torch.jit) StrictMinMaxConstraint (class in torch.fx.experimental.symbolic_shapes) stride() (torch.Tensor method) StringTable (class in torch.autograd.profiler_util) StudentT (class in torch.distributions.studentT) sub() (in module torch) (torch.Tensor method) sub_() (torch.Tensor method) SubclassSymbolicContext (class in torch.fx.experimental.symbolic_shapes) SubprocessContext (class in torch.distributed.elastic.multiprocessing.api) SubprocessHandler (class in torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler) Subset (class in torch.utils.data) SubsetRandomSampler (class in torch.utils.data) substitute_in_graph() (in module torch.compiler) subtract() (in module torch) (torch.Tensor method) subtract_() (torch.Tensor method) sum() (in module torch) (in module torch.sparse) (torch.Tensor method) sum_to_size() (torch.Tensor method) SummaryWriter (class in torch.utils.tensorboard.writer) summon_full_params() (torch.distributed.fsdp.FullyShardedDataParallel static method) support (torch.distributions.bernoulli.Bernoulli attribute) (torch.distributions.beta.Beta attribute) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy attribute) (torch.distributions.continuous_bernoulli.ContinuousBernoulli attribute) (torch.distributions.dirichlet.Dirichlet attribute) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential attribute) (torch.distributions.fishersnedecor.FisherSnedecor attribute) (torch.distributions.gamma.Gamma attribute) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric attribute) (torch.distributions.gumbel.Gumbel attribute) (torch.distributions.half_cauchy.HalfCauchy attribute) (torch.distributions.half_normal.HalfNormal attribute) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma attribute) (torch.distributions.kumaraswamy.Kumaraswamy attribute) (torch.distributions.laplace.Laplace attribute) (torch.distributions.lkj_cholesky.LKJCholesky attribute) (torch.distributions.log_normal.LogNormal attribute) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal attribute) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal attribute) (torch.distributions.negative_binomial.NegativeBinomial attribute) (torch.distributions.normal.Normal attribute) (torch.distributions.one_hot_categorical.OneHotCategorical attribute) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson attribute) (torch.distributions.relaxed_bernoulli.LogitRelaxedBernoulli attribute) (torch.distributions.relaxed_bernoulli.RelaxedBernoulli attribute) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical attribute) (torch.distributions.studentT.StudentT attribute) (torch.distributions.transformed_distribution.TransformedDistribution property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises attribute) (torch.distributions.weibull.Weibull attribute) (torch.distributions.wishart.Wishart attribute) suppress_guards() (torch.fx.experimental.symbolic_shapes.ShapeEnv method) svd() (in module torch) (in module torch.linalg) (torch.Tensor method) svd_lowrank() (in module torch) svdvals() (in module torch.linalg) SWALR (class in torch.optim.swa_utils) swap_module (class in torch.ao.quantization) swap_tensors() (in module torch.utils) swapaxes() (in module torch) (torch.Tensor method) swapdims() (in module torch) (torch.Tensor method) SyclExtension() (in module torch.utils.cpp_extension) sym_and() (in module torch.fx.experimental.symbolic_shapes) sym_eq() (in module torch.fx.experimental.symbolic_shapes) sym_float() (in module torch) sym_fresh_size() (in module torch) sym_int() (in module torch) sym_ite() (in module torch) sym_max() (in module torch) sym_min() (in module torch) sym_not() (in module torch) sym_or() (in module torch.fx.experimental.symbolic_shapes) sym_sum() (in module torch) symbolic() (in module torch.onnx.ops) symbolic_multi_out() (in module torch.onnx.ops) symbolic_trace() (in module torch.fx) SymbolicContext (class in torch.fx.experimental.symbolic_shapes) SymBool (class in torch) SymBoolArgument (class in torch.export.graph_signature) SymFloat (class in torch) SymFloatArgument (class in torch.export.graph_signature) SymInt (class in torch) SymIntArgument (class in torch.export.graph_signature) SyncBatchNorm (class in torch.nn) synchronize() (in module torch.accelerator) (in module torch.cpu) (in module torch.cuda) (in module torch.mps) (in module torch.mtia) (in module torch.xpu) (torch.cuda.Event method) (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.distributed.Work method) (torch.Event method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Event method) (torch.xpu.Stream method) synchronize_staging() (torch.distributed.checkpoint.staging.AsyncStager method) (torch.distributed.checkpoint.staging.BlockingAsyncStager method) T T (torch.Tensor attribute) t() (in module torch) (torch.Tensor method) t_() (torch.Tensor method) Tag (class in torch) take() (in module torch) (torch.Tensor method) take_along_dim() (in module torch) (torch.Tensor method) tan() (in module torch) (torch.Tensor method) tan_() (torch.Tensor method) tangent (torch.autograd.forward_ad.UnpackedDualTensor attribute) Tanh (class in torch.nn) tanh() (in module torch) (in module torch.nn.functional) (torch.Tensor method) tanh_() (torch.Tensor method) Tanhshrink (class in torch.nn) tanhshrink() (in module torch.nn.functional) TanhTransform (class in torch.distributions.transforms) TCPStore (class in torch.distributed) temperature (torch.distributions.relaxed_bernoulli.RelaxedBernoulli property) (torch.distributions.relaxed_categorical.RelaxedOneHotCategorical property) temperature() (in module torch.cuda) Tensor (class in torch) tensor() (in module torch) tensor_split() (in module torch) (torch.Tensor method) tensor_storage_size() (torch.distributed.checkpoint.planner.WriteItem method) tensorboard_trace_handler() (in module torch.profiler) TensorboardEventHandler (class in torch.monitor) TensorChunkSpec (class in torch.distributed.pipelining.microbatch) TensorDataset (class in torch.utils.data) tensordot() (in module torch) tensorinv() (in module torch.linalg) TensorPipeRpcBackendOptions (class in torch.distributed.rpc) tensorsolve() (in module torch.linalg) then() (torch.futures.Future method) threshold (class in torch.ao.nn.quantized.functional) Threshold (class in torch.nn) threshold() (in module torch.nn.functional) threshold_() (in module torch.nn.functional) tile() (in module torch) (torch.Tensor method) timeit() (torch.utils.benchmark.Timer method) timeout (torch.distributed.Store property) Timer (class in torch.utils.benchmark) TimerClient (class in torch.distributed.elastic.timer) TimerRequest (class in torch.distributed.elastic.timer) TimerServer (class in torch.distributed.elastic.timer) timestamp (torch.monitor.Event property) to() (torch.jit.ScriptModule method) (torch.nn.attention.flex_attention.BlockMask method) (torch.nn.Module method) (torch.nn.utils.rnn.PackedSequence method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) to_bool() (torch.fx.Tracer method) to_dense() (torch.nn.attention.flex_attention.BlockMask method) (torch.Tensor method) to_dict() (torch.ao.quantization.backend_config.BackendConfig method) (torch.ao.quantization.backend_config.BackendPatternConfig method) (torch.ao.quantization.backend_config.DTypeConfig method) (torch.ao.quantization.fx.custom_config.ConvertCustomConfig method) (torch.ao.quantization.fx.custom_config.FuseCustomConfig method) (torch.ao.quantization.fx.custom_config.PrepareCustomConfig method) (torch.ao.quantization.qconfig_mapping.QConfigMapping method) to_dlpack() (in module torch.utils.dlpack) to_empty() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) to_folder() (torch.fx.GraphModule method) to_here() (torch.distributed.rpc.PyRRef method) to_local() (torch.distributed.tensor.DTensor method) to_mkldnn() (torch.Tensor method) to_padded_tensor() (in module torch.nested) to_sparse() (torch.Tensor method) to_sparse_bsc() (torch.Tensor method) to_sparse_bsr() (torch.Tensor method) to_sparse_coo() (torch.Tensor method) to_sparse_csc() (torch.Tensor method) to_sparse_csr() (torch.Tensor method) to_string() (torch.nn.attention.flex_attention.BlockMask method) toggle_collection_dynamic() (torch.profiler._KinetoProfile method) tolist() (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) topk() (in module torch) (torch.Tensor method) torch module torch.__config__ module torch.__future__ module torch._logging module torch.accelerator module torch.amp module torch.amp.autocast_mode module torch.amp.grad_scaler module torch.ao module torch.ao.nn module torch.ao.nn.intrinsic module torch.ao.nn.intrinsic.modules module torch.ao.nn.intrinsic.modules.fused module torch.ao.nn.intrinsic.qat module torch.ao.nn.intrinsic.qat.modules module torch.ao.nn.intrinsic.qat.modules.conv_fused module torch.ao.nn.intrinsic.qat.modules.linear_fused module torch.ao.nn.intrinsic.qat.modules.linear_relu module torch.ao.nn.intrinsic.quantized module torch.ao.nn.intrinsic.quantized.dynamic module torch.ao.nn.intrinsic.quantized.dynamic.modules module torch.ao.nn.intrinsic.quantized.dynamic.modules.linear_relu module torch.ao.nn.intrinsic.quantized.modules module torch.ao.nn.intrinsic.quantized.modules.bn_relu module torch.ao.nn.intrinsic.quantized.modules.conv_add module torch.ao.nn.intrinsic.quantized.modules.conv_relu module torch.ao.nn.intrinsic.quantized.modules.linear_relu module torch.ao.nn.qat module torch.ao.nn.qat.dynamic module torch.ao.nn.qat.dynamic.modules module torch.ao.nn.qat.dynamic.modules.linear module torch.ao.nn.qat.modules module torch.ao.nn.qat.modules.conv module torch.ao.nn.qat.modules.embedding_ops module torch.ao.nn.qat.modules.linear module torch.ao.nn.quantizable module torch.ao.nn.quantizable.modules module torch.ao.nn.quantizable.modules.activation module torch.ao.nn.quantizable.modules.rnn module torch.ao.nn.quantized module torch.ao.nn.quantized.dynamic module torch.ao.nn.quantized.dynamic.modules module torch.ao.nn.quantized.dynamic.modules.conv module torch.ao.nn.quantized.dynamic.modules.linear module torch.ao.nn.quantized.dynamic.modules.rnn module torch.ao.nn.quantized.functional module torch.ao.nn.quantized.modules module torch.ao.nn.quantized.modules.activation module torch.ao.nn.quantized.modules.batchnorm module torch.ao.nn.quantized.modules.conv module torch.ao.nn.quantized.modules.dropout module torch.ao.nn.quantized.modules.embedding_ops module torch.ao.nn.quantized.modules.functional_modules module torch.ao.nn.quantized.modules.linear module torch.ao.nn.quantized.modules.normalization module torch.ao.nn.quantized.modules.rnn module torch.ao.nn.quantized.modules.utils module torch.ao.nn.quantized.reference module torch.ao.nn.quantized.reference.modules module torch.ao.nn.quantized.reference.modules.conv module torch.ao.nn.quantized.reference.modules.linear module torch.ao.nn.quantized.reference.modules.rnn module torch.ao.nn.quantized.reference.modules.sparse module torch.ao.nn.quantized.reference.modules.utils module torch.ao.nn.sparse module torch.ao.nn.sparse.quantized module torch.ao.nn.sparse.quantized.dynamic module torch.ao.nn.sparse.quantized.dynamic.linear module torch.ao.nn.sparse.quantized.linear module torch.ao.nn.sparse.quantized.utils module torch.ao.ns module torch.ao.ns._numeric_suite module torch.ao.ns._numeric_suite_fx module torch.ao.ns.fx module torch.ao.ns.fx.graph_matcher module torch.ao.ns.fx.graph_passes module torch.ao.ns.fx.mappings module torch.ao.ns.fx.n_shadows_utils module torch.ao.ns.fx.ns_types module torch.ao.ns.fx.pattern_utils module torch.ao.ns.fx.qconfig_multi_mapping module torch.ao.ns.fx.utils module torch.ao.ns.fx.weight_utils module torch.ao.pruning module torch.ao.pruning.scheduler module torch.ao.pruning.scheduler.base_scheduler module torch.ao.pruning.scheduler.cubic_scheduler module torch.ao.pruning.scheduler.lambda_scheduler module torch.ao.pruning.sparsifier module torch.ao.pruning.sparsifier.base_sparsifier module torch.ao.pruning.sparsifier.nearly_diagonal_sparsifier module torch.ao.pruning.sparsifier.utils module torch.ao.pruning.sparsifier.weight_norm_sparsifier module torch.ao.quantization module torch.ao.quantization.backend_config module torch.ao.quantization.backend_config.backend_config module torch.ao.quantization.backend_config.executorch module torch.ao.quantization.backend_config.fbgemm module torch.ao.quantization.backend_config.native module torch.ao.quantization.backend_config.observation_type module torch.ao.quantization.backend_config.onednn module torch.ao.quantization.backend_config.qnnpack module torch.ao.quantization.backend_config.tensorrt module torch.ao.quantization.backend_config.utils module torch.ao.quantization.backend_config.x86 module torch.ao.quantization.fake_quantize module torch.ao.quantization.fuse_modules module torch.ao.quantization.fuser_method_mappings module torch.ao.quantization.fx module torch.ao.quantization.fx.convert module torch.ao.quantization.fx.custom_config module torch.ao.quantization.fx.fuse module torch.ao.quantization.fx.fuse_handler module torch.ao.quantization.fx.graph_module module torch.ao.quantization.fx.lower_to_fbgemm module torch.ao.quantization.fx.lower_to_qnnpack module torch.ao.quantization.fx.lstm_utils module torch.ao.quantization.fx.match_utils module torch.ao.quantization.fx.pattern_utils module torch.ao.quantization.fx.prepare module torch.ao.quantization.fx.qconfig_mapping_utils module torch.ao.quantization.fx.quantize_handler module torch.ao.quantization.fx.tracer module torch.ao.quantization.fx.utils module torch.ao.quantization.observer module torch.ao.quantization.pt2e module torch.ao.quantization.pt2e.duplicate_dq_pass module torch.ao.quantization.pt2e.export_utils module torch.ao.quantization.pt2e.graph_utils module torch.ao.quantization.pt2e.lowering module torch.ao.quantization.pt2e.port_metadata_pass module torch.ao.quantization.pt2e.prepare module torch.ao.quantization.pt2e.qat_utils module torch.ao.quantization.pt2e.representation module torch.ao.quantization.pt2e.representation.rewrite module torch.ao.quantization.pt2e.utils module torch.ao.quantization.qconfig module torch.ao.quantization.qconfig_mapping module torch.ao.quantization.quant_type module torch.ao.quantization.quantization_mappings module torch.ao.quantization.quantize_fx module torch.ao.quantization.quantize_jit module torch.ao.quantization.quantize_pt2e module torch.ao.quantization.quantizer module torch.ao.quantization.quantizer.composable_quantizer module torch.ao.quantization.quantizer.embedding_quantizer module torch.ao.quantization.quantizer.quantizer module torch.ao.quantization.quantizer.utils module torch.ao.quantization.quantizer.x86_inductor_quantizer module torch.ao.quantization.quantizer.xnnpack_quantizer module torch.ao.quantization.quantizer.xnnpack_quantizer_utils module torch.ao.quantization.quantizer.xpu_inductor_quantizer module torch.ao.quantization.stubs module torch.ao.quantization.utils module torch.autograd module torch.autograd.anomaly_mode module torch.autograd.forward_ad module torch.autograd.function module torch.autograd.functional module torch.autograd.grad_mode module torch.autograd.gradcheck module torch.autograd.graph module torch.autograd.profiler module torch.autograd.profiler_legacy module torch.autograd.profiler_util module torch.autograd.variable module torch.backends module torch.backends.cpu module torch.backends.cuda module torch.backends.cudnn module torch.backends.cudnn.rnn module torch.backends.cusparselt module torch.backends.kleidiai module torch.backends.mha module torch.backends.mkl module torch.backends.mkldnn module torch.backends.mps module torch.backends.nnpack module torch.backends.openmp module torch.backends.opt_einsum module torch.backends.quantized module torch.backends.xeon module torch.backends.xeon.run_cpu module torch.backends.xnnpack module torch.compiler module torch.compiler.config module torch.contrib module torch.cpu module torch.cpu.amp module torch.cpu.amp.autocast_mode module torch.cpu.amp.grad_scaler module torch.cuda module torch.cuda._sanitizer module torch.cuda.amp module torch.cuda.amp.autocast_mode module torch.cuda.amp.common module torch.cuda.amp.grad_scaler module torch.cuda.comm module torch.cuda.error module torch.cuda.gds module torch.cuda.graphs module torch.cuda.jiterator module torch.cuda.memory module torch.cuda.nccl module torch.cuda.nvtx module torch.cuda.profiler module torch.cuda.random module torch.cuda.sparse module torch.cuda.streams module torch.cuda.tunable module torch.distributed module torch.distributed.algorithms module torch.distributed.algorithms.ddp_comm_hooks module torch.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook module torch.distributed.algorithms.ddp_comm_hooks.debugging_hooks module torch.distributed.algorithms.ddp_comm_hooks.default_hooks module torch.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks module torch.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks module torch.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook module torch.distributed.algorithms.ddp_comm_hooks.powerSGD_hook module torch.distributed.algorithms.ddp_comm_hooks.quantization_hooks module torch.distributed.algorithms.join module torch.distributed.algorithms.model_averaging module torch.distributed.algorithms.model_averaging.averagers module torch.distributed.algorithms.model_averaging.hierarchical_model_averager module torch.distributed.algorithms.model_averaging.utils module torch.distributed.argparse_util module torch.distributed.autograd module torch.distributed.c10d_logger module torch.distributed.checkpoint module torch.distributed.checkpoint.api module torch.distributed.checkpoint.default_planner module torch.distributed.checkpoint.filesystem module torch.distributed.checkpoint.format_utils module torch.distributed.checkpoint.logger module torch.distributed.checkpoint.logging_handlers module torch.distributed.checkpoint.metadata module torch.distributed.checkpoint.optimizer module torch.distributed.checkpoint.planner module torch.distributed.checkpoint.planner_helpers module torch.distributed.checkpoint.resharding module torch.distributed.checkpoint.staging module torch.distributed.checkpoint.state_dict module torch.distributed.checkpoint.state_dict_loader module torch.distributed.checkpoint.state_dict_saver module torch.distributed.checkpoint.stateful module torch.distributed.checkpoint.storage module torch.distributed.checkpoint.utils module torch.distributed.collective_utils module torch.distributed.constants module torch.distributed.device_mesh module torch.distributed.distributed_c10d module torch.distributed.elastic module torch.distributed.elastic.agent module torch.distributed.elastic.agent.server module torch.distributed.elastic.agent.server.api module torch.distributed.elastic.agent.server.health_check_server module torch.distributed.elastic.agent.server.local_elastic_agent module torch.distributed.elastic.control_plane module torch.distributed.elastic.events module torch.distributed.elastic.events.api module torch.distributed.elastic.events.handlers module torch.distributed.elastic.metrics module torch.distributed.elastic.metrics.api module torch.distributed.elastic.multiprocessing module torch.distributed.elastic.multiprocessing.api module torch.distributed.elastic.multiprocessing.errors module torch.distributed.elastic.multiprocessing.errors.error_handler module torch.distributed.elastic.multiprocessing.errors.handlers module torch.distributed.elastic.multiprocessing.redirects module torch.distributed.elastic.multiprocessing.subprocess_handler module torch.distributed.elastic.multiprocessing.subprocess_handler.handlers module torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler module torch.distributed.elastic.multiprocessing.tail_log module torch.distributed.elastic.rendezvous module torch.distributed.elastic.rendezvous.api module torch.distributed.elastic.rendezvous.c10d_rendezvous_backend module torch.distributed.elastic.rendezvous.dynamic_rendezvous module torch.distributed.elastic.rendezvous.etcd_rendezvous module torch.distributed.elastic.rendezvous.etcd_rendezvous_backend module torch.distributed.elastic.rendezvous.etcd_server module torch.distributed.elastic.rendezvous.etcd_store module torch.distributed.elastic.rendezvous.registry module torch.distributed.elastic.rendezvous.static_tcp_rendezvous module torch.distributed.elastic.rendezvous.utils module torch.distributed.elastic.timer module torch.distributed.elastic.timer.api module torch.distributed.elastic.timer.debug_info_logging module torch.distributed.elastic.timer.file_based_local_timer module torch.distributed.elastic.timer.local_timer module torch.distributed.elastic.utils module torch.distributed.elastic.utils.api module torch.distributed.elastic.utils.data module torch.distributed.elastic.utils.data.cycling_iterator module torch.distributed.elastic.utils.data.elastic_distributed_sampler module torch.distributed.elastic.utils.distributed module torch.distributed.elastic.utils.log_level module torch.distributed.elastic.utils.logging module torch.distributed.elastic.utils.store module torch.distributed.fsdp module torch.distributed.fsdp.api module torch.distributed.fsdp.fully_sharded_data_parallel module torch.distributed.fsdp.sharded_grad_scaler module torch.distributed.fsdp.wrap module torch.distributed.launch module torch.distributed.launcher module torch.distributed.launcher.api module torch.distributed.logging_handlers module torch.distributed.nn module torch.distributed.nn.api module torch.distributed.nn.api.remote_module module torch.distributed.nn.functional module torch.distributed.nn.jit module torch.distributed.nn.jit.instantiator module torch.distributed.nn.jit.templates module torch.distributed.nn.jit.templates.remote_module_template module torch.distributed.optim module torch.distributed.optim.apply_optimizer_in_backward module torch.distributed.optim.functional_adadelta module torch.distributed.optim.functional_adagrad module torch.distributed.optim.functional_adam module torch.distributed.optim.functional_adamax module torch.distributed.optim.functional_adamw module torch.distributed.optim.functional_rmsprop module torch.distributed.optim.functional_rprop module torch.distributed.optim.functional_sgd module torch.distributed.optim.named_optimizer module torch.distributed.optim.optimizer module torch.distributed.optim.post_localSGD_optimizer module torch.distributed.optim.utils module torch.distributed.optim.zero_redundancy_optimizer module torch.distributed.pipelining module torch.distributed.pipelining.microbatch module torch.distributed.pipelining.schedules module torch.distributed.pipelining.stage module torch.distributed.remote_device module torch.distributed.rendezvous module torch.distributed.rpc module torch.distributed.rpc.api module torch.distributed.rpc.backend_registry module torch.distributed.rpc.constants module torch.distributed.rpc.functions module torch.distributed.rpc.internal module torch.distributed.rpc.options module torch.distributed.rpc.rref_proxy module torch.distributed.rpc.server_process_global_profiler module torch.distributed.run module torch.distributed.tensor module torch.distributed.tensor.debug module torch.distributed.tensor.device_mesh module torch.distributed.tensor.experimental module torch.distributed.tensor.parallel module torch.distributed.tensor.parallel.api module torch.distributed.tensor.parallel.ddp module torch.distributed.tensor.parallel.fsdp module torch.distributed.tensor.parallel.input_reshard module torch.distributed.tensor.parallel.loss module torch.distributed.tensor.parallel.style module torch.distributed.tensor.placement_types module torch.distributed.utils module torch.distributions module torch.distributions.bernoulli module torch.distributions.beta module torch.distributions.binomial module torch.distributions.categorical module torch.distributions.cauchy module torch.distributions.chi2 module torch.distributions.constraint_registry module torch.distributions.constraints module torch.distributions.continuous_bernoulli module torch.distributions.dirichlet module torch.distributions.distribution module torch.distributions.exp_family module torch.distributions.exponential module torch.distributions.fishersnedecor module torch.distributions.gamma module torch.distributions.generalized_pareto module torch.distributions.geometric module torch.distributions.gumbel module torch.distributions.half_cauchy module torch.distributions.half_normal module torch.distributions.independent module torch.distributions.inverse_gamma module torch.distributions.kl module torch.distributions.kumaraswamy module torch.distributions.laplace module torch.distributions.lkj_cholesky module torch.distributions.log_normal module torch.distributions.logistic_normal module torch.distributions.lowrank_multivariate_normal module torch.distributions.mixture_same_family module torch.distributions.multinomial module torch.distributions.multivariate_normal module torch.distributions.negative_binomial module torch.distributions.normal module torch.distributions.one_hot_categorical module torch.distributions.pareto module torch.distributions.poisson module torch.distributions.relaxed_bernoulli module torch.distributions.relaxed_categorical module torch.distributions.studentT module torch.distributions.transformed_distribution module torch.distributions.transforms module torch.distributions.uniform module torch.distributions.utils module torch.distributions.von_mises module torch.distributions.weibull module torch.distributions.wishart module torch.export module torch.export.custom_obj module torch.export.custom_ops module torch.export.decomp_utils module torch.export.dynamic_shapes module torch.export.experimental module torch.export.exported_program module torch.export.graph_signature module torch.export.passes module torch.export.unflatten module torch.fft module torch.finfo (class in torch) torch.func module torch.functional module torch.futures module torch.fx module torch.fx.annotate module torch.fx.config module torch.fx.experimental module torch.fx.experimental.accelerator_partitioner module torch.fx.experimental.const_fold module torch.fx.experimental.debug module torch.fx.experimental.graph_gradual_typechecker module torch.fx.experimental.merge_matmul module torch.fx.experimental.meta_tracer module torch.fx.experimental.migrate_gradual_types module torch.fx.experimental.migrate_gradual_types.constraint module torch.fx.experimental.migrate_gradual_types.constraint_generator module torch.fx.experimental.migrate_gradual_types.constraint_transformation module torch.fx.experimental.migrate_gradual_types.operation module torch.fx.experimental.migrate_gradual_types.transform_to_z3 module torch.fx.experimental.migrate_gradual_types.util module torch.fx.experimental.migrate_gradual_types.z3_types module torch.fx.experimental.normalize module torch.fx.experimental.optimization module torch.fx.experimental.partitioner_utils module torch.fx.experimental.proxy_tensor module torch.fx.experimental.recording module torch.fx.experimental.refinement_types module torch.fx.experimental.rewriter module torch.fx.experimental.schema_type_annotation module torch.fx.experimental.sym_node module torch.fx.experimental.symbolic_shapes module torch.fx.experimental.unification module torch.fx.experimental.unification.core module torch.fx.experimental.unification.dispatch module torch.fx.experimental.unification.match module torch.fx.experimental.unification.more module torch.fx.experimental.unification.multipledispatch module torch.fx.experimental.unification.multipledispatch.conflict module torch.fx.experimental.unification.multipledispatch.core module torch.fx.experimental.unification.multipledispatch.dispatcher module torch.fx.experimental.unification.multipledispatch.utils module torch.fx.experimental.unification.multipledispatch.variadic module torch.fx.experimental.unification.unification_tools module torch.fx.experimental.unification.utils module torch.fx.experimental.unification.variable module torch.fx.experimental.unify_refinements module torch.fx.experimental.validator module torch.fx.graph module torch.fx.graph_module module torch.fx.immutable_collections module torch.fx.interpreter module torch.fx.node module torch.fx.operator_schemas module torch.fx.passes module torch.fx.passes.annotate_getitem_nodes module torch.fx.passes.backends module torch.fx.passes.backends.cudagraphs module torch.fx.passes.dialect module torch.fx.passes.dialect.common module torch.fx.passes.dialect.common.cse_pass module torch.fx.passes.fake_tensor_prop module torch.fx.passes.graph_drawer module torch.fx.passes.graph_manipulation module torch.fx.passes.graph_transform_observer module torch.fx.passes.infra module torch.fx.passes.infra.partitioner module torch.fx.passes.infra.pass_base module torch.fx.passes.infra.pass_manager module torch.fx.passes.net_min_base module torch.fx.passes.operator_support module torch.fx.passes.param_fetch module torch.fx.passes.pass_manager module torch.fx.passes.reinplace module torch.fx.passes.runtime_assert module torch.fx.passes.shape_prop module torch.fx.passes.split_module module torch.fx.passes.split_utils module torch.fx.passes.splitter_base module torch.fx.passes.tests module torch.fx.passes.tests.test_pass_manager module torch.fx.passes.tools_common module torch.fx.passes.utils module torch.fx.passes.utils.common module torch.fx.passes.utils.fuser_utils module torch.fx.passes.utils.matcher_utils module torch.fx.passes.utils.matcher_with_name_node_map_utils module torch.fx.passes.utils.source_matcher_utils module torch.fx.proxy module torch.fx.subgraph_rewriter module torch.fx.tensor_type module torch.fx.traceback module torch.hub module torch.iinfo (class in torch) torch.jit module torch.jit.annotations module torch.jit.frontend module torch.jit.generate_bytecode module torch.jit.mobile module torch.jit.quantized module torch.jit.supported_ops module torch.jit.unsupported_tensor_ops module torch.library module torch.linalg module torch.masked module torch.masked.maskedtensor module torch.masked.maskedtensor.binary module torch.masked.maskedtensor.core module torch.masked.maskedtensor.creation module torch.masked.maskedtensor.passthrough module torch.masked.maskedtensor.reductions module torch.masked.maskedtensor.unary module torch.monitor module torch.mps module torch.mps.event module torch.mps.profiler module torch.mtia module torch.mtia.memory module torch.multiprocessing module torch.multiprocessing.pool module torch.multiprocessing.queue module torch.multiprocessing.reductions module torch.multiprocessing.spawn module torch.nested module torch.nn module torch.nn.attention module torch.nn.attention.bias module torch.nn.attention.experimental module torch.nn.attention.flex_attention module torch.nn.backends module torch.nn.backends.thnn module torch.nn.common_types module torch.nn.cpp module torch.nn.functional module torch.nn.grad module torch.nn.init module torch.nn.intrinsic module torch.nn.intrinsic.modules module torch.nn.intrinsic.modules.fused module torch.nn.intrinsic.qat module torch.nn.intrinsic.qat.modules module torch.nn.intrinsic.qat.modules.conv_fused module torch.nn.intrinsic.qat.modules.linear_fused module torch.nn.intrinsic.qat.modules.linear_relu module torch.nn.intrinsic.quantized module torch.nn.intrinsic.quantized.dynamic module torch.nn.intrinsic.quantized.dynamic.modules module torch.nn.intrinsic.quantized.dynamic.modules.linear_relu module torch.nn.intrinsic.quantized.modules module torch.nn.intrinsic.quantized.modules.bn_relu module torch.nn.intrinsic.quantized.modules.conv_relu module torch.nn.intrinsic.quantized.modules.linear_relu module torch.nn.modules module torch.nn.modules.activation module torch.nn.modules.adaptive module torch.nn.modules.batchnorm module torch.nn.modules.channelshuffle module torch.nn.modules.container module torch.nn.modules.conv module torch.nn.modules.distance module torch.nn.modules.dropout module torch.nn.modules.flatten module torch.nn.modules.fold module torch.nn.modules.instancenorm module torch.nn.modules.lazy module torch.nn.modules.linear module torch.nn.modules.loss module torch.nn.modules.module module torch.nn.modules.normalization module torch.nn.modules.padding module torch.nn.modules.pixelshuffle module torch.nn.modules.pooling module torch.nn.modules.rnn module torch.nn.modules.sparse module torch.nn.modules.transformer module torch.nn.modules.upsampling module torch.nn.modules.utils module torch.nn.parallel module torch.nn.parallel.comm module torch.nn.parallel.distributed module torch.nn.parallel.parallel_apply module torch.nn.parallel.replicate module torch.nn.parallel.scatter_gather module torch.nn.parameter module torch.nn.qat module torch.nn.qat.dynamic module torch.nn.qat.dynamic.modules module torch.nn.qat.dynamic.modules.linear module torch.nn.qat.modules module torch.nn.qat.modules.conv module torch.nn.qat.modules.embedding_ops module torch.nn.qat.modules.linear module torch.nn.quantizable module torch.nn.quantizable.modules module torch.nn.quantizable.modules.activation module torch.nn.quantizable.modules.rnn module torch.nn.quantized module torch.nn.quantized.dynamic module torch.nn.quantized.dynamic.modules module torch.nn.quantized.dynamic.modules.conv module torch.nn.quantized.dynamic.modules.linear module torch.nn.quantized.dynamic.modules.rnn module torch.nn.quantized.functional module torch.nn.quantized.modules module torch.nn.quantized.modules.activation module torch.nn.quantized.modules.batchnorm module torch.nn.quantized.modules.conv module torch.nn.quantized.modules.dropout module torch.nn.quantized.modules.embedding_ops module torch.nn.quantized.modules.functional_modules module torch.nn.quantized.modules.linear module torch.nn.quantized.modules.normalization module torch.nn.quantized.modules.rnn module torch.nn.quantized.modules.utils module torch.nn.utils module torch.nn.utils.clip_grad module torch.nn.utils.convert_parameters module torch.nn.utils.fusion module torch.nn.utils.init module torch.nn.utils.memory_format module torch.nn.utils.parametrizations module torch.nn.utils.parametrize module torch.nn.utils.prune module torch.nn.utils.rnn module torch.nn.utils.stateless module torch.onnx module torch.onnx.errors module torch.onnx.operators module torch.onnx.ops module torch.onnx.symbolic_caffe2 module torch.onnx.symbolic_helper module torch.onnx.symbolic_opset10 module torch.onnx.symbolic_opset11 module torch.onnx.symbolic_opset12 module torch.onnx.symbolic_opset13 module torch.onnx.symbolic_opset14 module torch.onnx.symbolic_opset15 module torch.onnx.symbolic_opset16 module torch.onnx.symbolic_opset17 module torch.onnx.symbolic_opset18 module torch.onnx.symbolic_opset19 module torch.onnx.symbolic_opset20 module torch.onnx.symbolic_opset7 module torch.onnx.symbolic_opset8 module torch.onnx.symbolic_opset9 module torch.onnx.utils module torch.onnx.verification module torch.optim module torch.optim.adadelta module torch.optim.adagrad module torch.optim.adam module torch.optim.adamax module torch.optim.adamw module torch.optim.asgd module torch.optim.lbfgs module torch.optim.lr_scheduler module torch.optim.nadam module torch.optim.optimizer module torch.optim.radam module torch.optim.rmsprop module torch.optim.rprop module torch.optim.sgd module torch.optim.sparse_adam module torch.optim.swa_utils module torch.overrides module torch.package module torch.package.analyze module torch.package.analyze.find_first_use_of_broken_modules module torch.package.analyze.is_from_package module torch.package.analyze.trace_dependencies module torch.package.file_structure_representation module torch.package.find_file_dependencies module torch.package.glob_group module torch.package.importer module torch.package.package_exporter module torch.package.package_importer module torch.profiler module torch.profiler.itt module torch.profiler.profiler module torch.profiler.python_tracer module torch.quantization module torch.quantization.fake_quantize module torch.quantization.fuse_modules module torch.quantization.fuser_method_mappings module torch.quantization.fx module torch.quantization.fx.convert module torch.quantization.fx.fuse module torch.quantization.fx.fusion_patterns module torch.quantization.fx.graph_module module torch.quantization.fx.match_utils module torch.quantization.fx.pattern_utils module torch.quantization.fx.prepare module torch.quantization.fx.quantization_patterns module torch.quantization.fx.quantization_types module torch.quantization.fx.utils module torch.quantization.observer module torch.quantization.qconfig module torch.quantization.quant_type module torch.quantization.quantization_mappings module torch.quantization.quantize module torch.quantization.quantize_fx module torch.quantization.quantize_jit module torch.quantization.stubs module torch.quantization.utils module torch.quasirandom module torch.random module torch.return_types module torch.serialization module torch.signal module torch.signal.windows module torch.signal.windows.windows module torch.sparse module torch.sparse.semi_structured module torch.special module torch.storage module torch.testing module torch.torch_version module torch.types module torch.utils module torch.utils.backcompat module torch.utils.backend_registration module torch.utils.benchmark module torch.utils.benchmark.examples module torch.utils.benchmark.examples.compare module torch.utils.benchmark.examples.fuzzer module torch.utils.benchmark.examples.op_benchmark module torch.utils.benchmark.examples.simple_timeit module torch.utils.benchmark.examples.spectral_ops_fuzz_test module torch.utils.benchmark.op_fuzzers module torch.utils.benchmark.op_fuzzers.binary module torch.utils.benchmark.op_fuzzers.sparse_binary module torch.utils.benchmark.op_fuzzers.sparse_unary module torch.utils.benchmark.op_fuzzers.spectral module torch.utils.benchmark.op_fuzzers.unary module torch.utils.benchmark.utils module torch.utils.benchmark.utils.common module torch.utils.benchmark.utils.compare module torch.utils.benchmark.utils.compile module torch.utils.benchmark.utils.cpp_jit module torch.utils.benchmark.utils.fuzzer module torch.utils.benchmark.utils.sparse_fuzzer module torch.utils.benchmark.utils.timer module torch.utils.benchmark.utils.valgrind_wrapper module torch.utils.benchmark.utils.valgrind_wrapper.timer_interface module torch.utils.bottleneck module torch.utils.bundled_inputs module torch.utils.checkpoint module torch.utils.collect_env module torch.utils.cpp_backtrace module torch.utils.cpp_extension module torch.utils.data module torch.utils.data.backward_compatibility module torch.utils.data.dataloader module torch.utils.data.datapipes module torch.utils.data.datapipes.dataframe module torch.utils.data.datapipes.dataframe.dataframe_wrapper module torch.utils.data.datapipes.dataframe.dataframes module torch.utils.data.datapipes.dataframe.datapipes module torch.utils.data.datapipes.dataframe.structures module torch.utils.data.datapipes.datapipe module torch.utils.data.datapipes.gen_pyi module torch.utils.data.datapipes.iter module torch.utils.data.datapipes.iter.callable module torch.utils.data.datapipes.iter.combinatorics module torch.utils.data.datapipes.iter.combining module torch.utils.data.datapipes.iter.filelister module torch.utils.data.datapipes.iter.fileopener module torch.utils.data.datapipes.iter.grouping module torch.utils.data.datapipes.iter.routeddecoder module torch.utils.data.datapipes.iter.selecting module torch.utils.data.datapipes.iter.sharding module torch.utils.data.datapipes.iter.streamreader module torch.utils.data.datapipes.iter.utils module torch.utils.data.datapipes.map module torch.utils.data.datapipes.map.callable module torch.utils.data.datapipes.map.combinatorics module torch.utils.data.datapipes.map.combining module torch.utils.data.datapipes.map.grouping module torch.utils.data.datapipes.map.utils module torch.utils.data.datapipes.utils module torch.utils.data.datapipes.utils.common module torch.utils.data.datapipes.utils.decoder module torch.utils.data.datapipes.utils.snapshot module torch.utils.data.dataset module torch.utils.data.distributed module torch.utils.data.graph module torch.utils.data.graph_settings module torch.utils.data.sampler module torch.utils.deterministic module torch.utils.dlpack module torch.utils.file_baton module torch.utils.flop_counter module torch.utils.hipify module torch.utils.hipify.constants module torch.utils.hipify.cuda_to_hip_mappings module torch.utils.hipify.hipify_python module torch.utils.hipify.version module torch.utils.hooks module torch.utils.jit module torch.utils.jit.log_extract module torch.utils.mkldnn module torch.utils.mobile_optimizer module torch.utils.model_dump module torch.utils.model_zoo module torch.utils.module_tracker module torch.utils.serialization module torch.utils.serialization.config module torch.utils.show_pickle module torch.utils.tensorboard module torch.utils.tensorboard.summary module torch.utils.tensorboard.writer module torch.utils.throughput_benchmark module torch.utils.viz module torch.utils.weak module torch.version module torch.xpu module torch.xpu.memory module torch.xpu.random module torch.xpu.streams module TORCH_COMPILE_JOB_ID torch_name() (torch.onnx.JitScalarType method) torch_save_to_dcp() (in module torch.distributed.checkpoint.format_utils) TorchAODType (class in torch.ao.quantization.observer) total_average() (torch.autograd.profiler.profile method) total_count (torch.distributions.multinomial.Multinomial attribute) trace() (in module torch) (in module torch.jit) (torch.fx.Tracer method) (torch.Tensor method) trace_module() (in module torch.jit) Tracer (class in torch.fx) train() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) Transform (class in torch.distributions.transforms) transform() (torch.fx.Transformer method) (torch.utils.benchmark.FunctionCounts method) transform_object() (torch.distributed.checkpoint.DefaultSavePlanner method) transform_tensor() (torch.distributed.checkpoint.DefaultLoadPlanner method) TransformedDistribution (class in torch.distributions.transformed_distribution) Transformer (class in torch.fx) (class in torch.nn) TransformerDecoder (class in torch.nn) TransformerDecoderLayer (class in torch.nn) TransformerEncoder (class in torch.nn) TransformerEncoderLayer (class in torch.nn) transpose() (in module torch) (torch.Tensor method) transpose_() (torch.Tensor method) trapezoid() (in module torch) trapz() (in module torch) triangular_solve() (in module torch) (torch.Tensor method) tril() (in module torch) (torch.Tensor method) tril_() (torch.Tensor method) tril_indices() (in module torch) trim_significant_figures() (torch.utils.benchmark.Compare method) triplet_margin_loss() (in module torch.nn.functional) triplet_margin_with_distance_loss() (in module torch.nn.functional) TripletMarginLoss (class in torch.nn) TripletMarginWithDistanceLoss (class in torch.nn) triton_op() (in module torch.library) triu() (in module torch) (torch.Tensor method) triu_() (torch.Tensor method) triu_indices() (in module torch) true_divide() (in module torch) (torch.Tensor method) true_divide_() (torch.Tensor method) trunc() (in module torch) (torch.Tensor method) trunc_() (torch.Tensor method) trunc_normal_() (in module torch.nn.init) tune_gemm_in_file() (in module torch.cuda.tunable) tuning_enable() (in module torch.cuda.tunable) tuning_is_enabled() (in module torch.cuda.tunable) type (torch.jit.Attribute attribute) type() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) (torch.TypedStorage method) (torch.UntypedStorage method) type_as() (torch.Tensor method) TypedStorage (class in torch) U unbind() (in module torch) (torch.Tensor method) unbox() (torch.distributed.Work static method) underlying_store (torch.distributed.PrefixStore property) Unflatten (class in torch.nn) unflatten() (in module torch) (in module torch.export.unflatten) (torch.Tensor method) Unfold (class in torch.nn) unfold() (in module torch.nn.functional) (torch.Tensor method) Uniform (class in torch.distributions.uniform) uniform_() (in module torch.nn.init) (torch.Tensor method) UninitializedBuffer (class in torch.nn.parameter) UninitializedParameter (class in torch.nn.parameter) unique() (in module torch) (torch.Tensor method) unique_consecutive() (in module torch) (torch.Tensor method) unpack_dual() (in module torch.autograd.forward_ad) unpack_sequence() (in module torch.nn.utils.rnn) UnpackedDualTensor (class in torch.autograd.forward_ad) unpad_sequence() (in module torch.nn.utils.rnn) unravel_index() (in module torch) unregister_custom_op_symbolic() (in module torch.onnx) unregister_event_handler() (in module torch.monitor) unshard() (torch.distributed.fsdp.FSDPModule method) UnshardHandle (class in torch.distributed.fsdp) unsorted_indices (torch.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module torch) (torch.Tensor method) unsqueeze_() (torch.Tensor method) untyped() (torch.TypedStorage method) (torch.UntypedStorage method) untyped_storage() (torch.Tensor method) UntypedStorage (class in torch) unused() (in module torch.jit) update() (torch.autograd.profiler_util.StringTable method) (torch.export.decomp_utils.CustomDecompTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) update_arg() (torch.fx.Node method) update_bn() (in module torch.optim.swa_utils) update_bn_stats (class in torch.ao.nn.intrinsic.qat) update_kwarg() (torch.fx.Node method) update_parameters() (torch.optim.swa_utils.AveragedModel method) upsample (class in torch.ao.nn.quantized.functional) Upsample (class in torch.nn) upsample() (in module torch.nn.functional) upsample_bilinear (class in torch.ao.nn.quantized.functional) upsample_bilinear() (in module torch.nn.functional) upsample_nearest (class in torch.ao.nn.quantized.functional) upsample_nearest() (in module torch.nn.functional) UpsamplingBilinear2d (class in torch.nn) UpsamplingNearest2d (class in torch.nn) use_agent_store (torch.distributed.elastic.rendezvous.RendezvousHandler property) use_count() (torch.cuda.MemPool method) use_deterministic_algorithms() (in module torch) use_mem_pool (class in torch.cuda) utilization() (in module torch.cuda) V validate_checkpoint_id() (torch.distributed.checkpoint.format_utils.BroadcastingTorchSaveReader class method) (torch.distributed.checkpoint.StorageReader class method) (torch.distributed.checkpoint.StorageWriter class method) value (torch.jit.Attribute attribute) value() (torch.futures.Future method) values() (torch.autograd.profiler_util.StringTable method) (torch.nn.ModuleDict method) (torch.nn.ParameterDict method) (torch.Tensor method) vander() (in module torch) (in module torch.linalg) var() (in module torch) (torch.Tensor method) var_mean() (in module torch) variance (torch.distributions.bernoulli.Bernoulli property) (torch.distributions.beta.Beta property) (torch.distributions.binomial.Binomial property) (torch.distributions.categorical.Categorical property) (torch.distributions.cauchy.Cauchy property) (torch.distributions.continuous_bernoulli.ContinuousBernoulli property) (torch.distributions.dirichlet.Dirichlet property) (torch.distributions.distribution.Distribution property) (torch.distributions.exponential.Exponential property) (torch.distributions.fishersnedecor.FisherSnedecor property) (torch.distributions.gamma.Gamma property) (torch.distributions.generalized_pareto.GeneralizedPareto property) (torch.distributions.geometric.Geometric property) (torch.distributions.gumbel.Gumbel property) (torch.distributions.half_cauchy.HalfCauchy property) (torch.distributions.half_normal.HalfNormal property) (torch.distributions.independent.Independent property) (torch.distributions.inverse_gamma.InverseGamma property) (torch.distributions.kumaraswamy.Kumaraswamy property) (torch.distributions.laplace.Laplace property) (torch.distributions.log_normal.LogNormal property) (torch.distributions.lowrank_multivariate_normal.LowRankMultivariateNormal property) (torch.distributions.mixture_same_family.MixtureSameFamily property) (torch.distributions.multinomial.Multinomial property) (torch.distributions.multivariate_normal.MultivariateNormal property) (torch.distributions.negative_binomial.NegativeBinomial property) (torch.distributions.normal.Normal property) (torch.distributions.one_hot_categorical.OneHotCategorical property) (torch.distributions.pareto.Pareto property) (torch.distributions.poisson.Poisson property) (torch.distributions.studentT.StudentT property) (torch.distributions.uniform.Uniform property) (torch.distributions.von_mises.VonMises property) (torch.distributions.weibull.Weibull property) (torch.distributions.wishart.Wishart property) vdot() (in module torch) (torch.Tensor method) vecdot() (in module torch.linalg) vector_norm() (in module torch.linalg) vector_to_parameters() (in module torch.nn.utils) verbose (class in torch.backends.mkl) (class in torch.backends.mkldnn) VerificationInfo (class in torch.onnx.verification) VerificationOptions (class in torch.onnx.verification) verify() (in module torch.onnx.verification) (torch.export.dynamic_shapes.AdditionalInputs method) verify_aten_graph() (in module torch.onnx.verification) verify_ninja_availability() (in module torch.utils.cpp_extension) verify_onnx_program() (in module torch.onnx.verification) version() (in module torch.backends.cudnn) (in module torch.backends.cusparselt) vhp() (in module torch.autograd.functional) view() (torch.Tensor method) view_as() (torch.Tensor method) view_as_complex() (in module torch) view_as_real() (in module torch) visualize_sharding() (in module torch.distributed.tensor.debug) vjp() (in module torch.autograd.functional) (in module torch.func) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) vmap() (in module torch) (in module torch.func) (torch.autograd.Function static method) (torch.autograd.function.InplaceFunction static method) (torch.autograd.function.NestedIOFunction static method) VonMises (class in torch.distributions.von_mises) vsplit() (in module torch) (torch.Tensor method) vstack() (in module torch) W wait() (in module torch.jit) (torch.cuda.Event method) (torch.distributed.elastic.rendezvous.etcd_store.EtcdStore method) (torch.distributed.fsdp.UnshardHandle method) (torch.distributed.Store method) (torch.distributed.Work method) (torch.Event method) (torch.futures.Future method) (torch.mps.event.Event method) (torch.mtia.Event method) (torch.xpu.Event method) wait_all() (in module torch.futures) wait_event() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) wait_stream() (torch.cuda.ExternalStream method) (torch.cuda.Stream method) (torch.mtia.Stream method) (torch.Stream method) (torch.xpu.Stream method) Weibull (class in torch.distributions.weibull) weight_norm() (in module torch.nn.utils) (in module torch.nn.utils.parametrizations) WeightedRandomSampler (class in torch.utils.data) where() (in module torch) (torch.Tensor method) Wishart (class in torch.distributions.wishart) with_args() (torch.ao.quantization.observer.AffineQuantizedObserverBase class method) (torch.ao.quantization.observer.ObserverBase class method) with_callable_args() (torch.ao.quantization.observer.ObserverBase class method) Work (class in torch.distributed) Worker (class in torch.distributed.elastic.agent.server) worker_main() (in module torch.distributed.elastic.control_plane) WorkerGroup (class in torch.distributed.elastic.agent.server) WorkerInfo (class in torch.distributed.rpc) WorkerSpec (class in torch.distributed.elastic.agent.server) WorkerState (class in torch.distributed.elastic.agent.server) wrap() (in module torch.fx) wrap_torch_function() (in module torch.overrides) wrap_triton() (in module torch.library) write_data() (torch.distributed.checkpoint.StorageWriter method) write_file() (in module torch.cuda.tunable) write_file_on_exit() (in module torch.cuda.tunable) WriteItem (class in torch.distributed.checkpoint.planner) X xavier_normal_() (in module torch.nn.init) xavier_uniform_() (in module torch.nn.init) xlog1py() (in module torch.special) xlogy() (in module torch) (in module torch.special) (torch.Tensor method) xlogy_() (torch.Tensor method) xpu() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.swa_utils.AveragedModel method) (torch.Tensor method) Z zero_() (torch.Tensor method) zero_grad() (torch.jit.ScriptModule method) (torch.nn.Module method) (torch.optim.Adadelta method) (torch.optim.Adafactor method) (torch.optim.Adagrad method) (torch.optim.Adam method) (torch.optim.Adamax method) (torch.optim.AdamW method) (torch.optim.ASGD method) (torch.optim.LBFGS method) (torch.optim.NAdam method) (torch.optim.Optimizer method) (torch.optim.RAdam method) (torch.optim.RMSprop method) (torch.optim.Rprop method) (torch.optim.SGD method) (torch.optim.SparseAdam method) (torch.optim.swa_utils.AveragedModel method) ZeroPad1d (class in torch.nn) ZeroPad2d (class in torch.nn) ZeroPad3d (class in torch.nn) ZeroPointDomain (class in torch.ao.quantization.observer) ZeroRedundancyOptimizer (class in torch.distributed.optim) zeros() (in module torch) (in module torch.distributed.tensor) zeros_() (in module torch.nn.init) zeros_like() (in module torch) zeta() (in module torch.special)