# Index

Source: https://www.tensorplay.cn/docs/genindex.html

[A](#A)
 | [B](#B)
 | [C](#C)
 | [D](#D)
 | [E](#E)
 | [F](#F)
 | [G](#G)
 | [H](#H)
 | [I](#I)
 | [J](#J)
 | [K](#K)
 | [L](#L)
 | [M](#M)
 | [N](#N)
 | [O](#O)
 | [P](#P)
 | [Q](#Q)
 | [R](#R)
 | [S](#S)
 | [T](#T)
 | [U](#U)
 | [V](#V)
 | [W](#W)
 | [X](#X)
 | [Z](#Z)
 

## A

| abs() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) abs_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) absolute() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) absolute_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) AbsTransform (class in tensorplay.distributions) AcceleratorError acos() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) acos_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) acosh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) acosh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) activate_flash_attention_impl() (in module tensorplay.nn.attention) activation (tensorplay.ao.quantization.QConfig attribute) adaptive_avg_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) adaptive_avg_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_avg_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) adaptive_max_pool1d_with_indices() (in module tensorplay.nn.functional) adaptive_max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool2d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) adaptive_max_pool3d_with_indices() (in module tensorplay.nn.functional) add() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.monitor.Stat method) (tensorplay.nn.attention.bias.CausalBias method) add_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) add_dependency() (tensorplay.package.PackageExporter method) add_done_callback() (tensorplay.Future method) (tensorplay.futures.Future method) add_module() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) add_param_group() (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) add_post_hook() (tensorplay.Node method) add_pre_hook() (tensorplay.Node method) add_pruning_method() (tensorplay.ao.pruning.PruningContainer method) add_relu() (in module tensorplay) add_safe_globals() (in module tensorplay.serialization) addbmm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addbmm_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) addcdiv() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addcdiv_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) addcmul() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addcmul_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) AdditionalInputs (class in tensorplay.export) addmm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addmm_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) addmv() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addmv_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addr (tensorplay.distributed.elastic.rendezvous.RendezvousStoreInfo property) addr() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) addr_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) adjoint() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) affine_grid() (in module tensorplay.nn.functional) affine_grid_generator() (in module tensorplay.functional) AffineTransform (class in tensorplay.distributions) Aggregation (class in tensorplay.monitor) airy_ai() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) alias() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) alias_copy() (in module tensorplay.functional) all() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) all_gather() (in module tensorplay.distributed.nn.functional) all_gather_single() (in module tensorplay.distributed.nn.functional) all_paths() (tensorplay.package.PackageExporter method) all_reduce() (in module tensorplay.distributed.nn.functional) all_to_all() (in module tensorplay.distributed.nn.functional) all_to_all_single() (in module tensorplay.distributed.nn.functional) allclose() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) AllGatherStates (class in tensorplay.distributed.rpc) allow_fp16_bf16_reduction_math_sdp() (in module tensorplay.backends.cuda) allow_in_graph() (in module tensorplay.compiler) allreduce_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks) alpha_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) alpha_dropout_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) amax() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) amin() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) aminmax() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) and_masks() (in module tensorplay.nn.attention) angle() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) any() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) append() (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) append_cpu_child() (tensorplay.autograd.profiler_util.FunctionEvent method) apply() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.pruning.BasePruningMethod class method) (tensorplay.ao.pruning.CustomFromMask class method) (tensorplay.ao.pruning.Identity class method) (tensorplay.ao.pruning.L1Unstructured class method) (tensorplay.ao.pruning.LnStructured class method) (tensorplay.ao.pruning.PruningContainer class method) (tensorplay.ao.pruning.RandomStructured class method) (tensorplay.ao.pruning.RandomUnstructured class method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.autograd.function.Function class method) (tensorplay.autograd.function.InplaceFunction class method) (tensorplay.autograd.function.NestedIOFunction class method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |apply_() (tensorplay.nn.attention.bias.CausalBias method) apply_mask() (tensorplay.ao.pruning.BasePruningMethod method) (tensorplay.ao.pruning.CustomFromMask method) (tensorplay.ao.pruning.Identity method) (tensorplay.ao.pruning.L1Unstructured method) (tensorplay.ao.pruning.LnStructured method) (tensorplay.ao.pruning.PruningContainer method) (tensorplay.ao.pruning.RandomStructured method) (tensorplay.ao.pruning.RandomUnstructured method) arange() (in module tensorplay.functional) arccos() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arccos_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arccosh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arccosh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arcsin() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arcsin_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arcsinh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arcsinh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctan() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctan2() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctan2_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctan_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctanh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) arctanh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) are_deterministic_algorithms_enabled() (in module tensorplay) arg_constraints (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) argmax() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) argmin() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) argsort() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) argwhere() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) as_functional_optim() (in module tensorplay.distributed.optim) as_integer_ratio() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) (tensorplay.SymFloat method) (tensorplay.SymInt method) as_masked_tensor() (in module tensorplay.masked) as_nested_tensor() (in module tensorplay.nested) as_strided() (in module tensorplay.functional) as_strided_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) as_strided_copy() (in module tensorplay.functional) as_strided_scatter() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) as_subclass() (tensorplay.nn.attention.bias.CausalBias method) as_tensor() (in module tensorplay) as_tuple() (tensorplay.nn.attention.BlockMask method) asarray() (in module tensorplay) asin() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) asin_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) asinh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) asinh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) assert_allclose() (in module tensorplay.testing) assert_close() (in module tensorplay.testing) assertion_dep_token (tensorplay.export.ExportGraphSignature property) assume_constant_result() (in module tensorplay.compiler) async_save() (in module tensorplay.distributed.checkpoint) AsyncCheckpointerType (class in tensorplay.distributed.checkpoint) AsyncSaveResponse (class in tensorplay.distributed.checkpoint) atan() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) atan2() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) atan2_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) atan_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) atanh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) atanh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) atleast_1d() (in module tensorplay.functional) atleast_2d() (in module tensorplay.functional) atleast_3d() (in module tensorplay.functional) autocast (class in tensorplay) autocast_decrement_nesting() (in module tensorplay) autocast_increment_nesting() (in module tensorplay) average_parameters() (tensorplay.distributed.algorithms.model_averaging.HierarchicalModelAverager method) (tensorplay.distributed.algorithms.model_averaging.PeriodicModelAverager method) avg_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) avg_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) avg_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) |
| --- | --- |

## B

| backend_registered() (in module tensorplay.distributed.rpc) BackendType (class in tensorplay.distributed.rpc) BackendValue (class in tensorplay.distributed.rpc) backward() (in module tensorplay.autograd) (in module tensorplay.distributed.autograd) (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.nn.attention.bias.CausalBias method) BackwardPrefetch (class in tensorplay.distributed.fsdp) baddbmm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) baddbmm_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) bartlett() (in module tensorplay.signal.windows) bartlett_window() (in module tensorplay.functional) BasePruningMethod (class in tensorplay.ao.pruning) batch_norm() (in module tensorplay) (in module tensorplay.nn.functional) batch_norm_backward_elemt() (in module tensorplay.functional) batch_norm_backward_reduce() (in module tensorplay.functional) batch_norm_elemt() (in module tensorplay.functional) batch_norm_gather_stats() (in module tensorplay.functional) batch_norm_gather_stats_with_counts() (in module tensorplay.functional) batch_norm_stats() (in module tensorplay.functional) batch_norm_update_stats() (in module tensorplay.functional) batch_norm_without_running_stats() (in module tensorplay.func) batch_shape (tensorplay.distributions.Bernoulli property) (tensorplay.distributions.Beta property) (tensorplay.distributions.Binomial property) (tensorplay.distributions.Categorical property) (tensorplay.distributions.Cauchy property) (tensorplay.distributions.Chi2 property) (tensorplay.distributions.ContinuousBernoulli property) (tensorplay.distributions.Dirichlet property) (tensorplay.distributions.Distribution property) (tensorplay.distributions.Exponential property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.FisherSnedecor property) (tensorplay.distributions.Gamma property) (tensorplay.distributions.GeneralizedPareto property) (tensorplay.distributions.Geometric property) (tensorplay.distributions.Gumbel property) (tensorplay.distributions.HalfCauchy property) (tensorplay.distributions.HalfNormal property) (tensorplay.distributions.Independent property) (tensorplay.distributions.InverseGamma property) (tensorplay.distributions.Kumaraswamy property) (tensorplay.distributions.Laplace property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.LogNormal property) (tensorplay.distributions.LowRankMultivariateNormal property) (tensorplay.distributions.MixtureSameFamily property) (tensorplay.distributions.Multinomial property) (tensorplay.distributions.MultivariateNormal property) (tensorplay.distributions.NegativeBinomial property) (tensorplay.distributions.Normal property) (tensorplay.distributions.OneHotCategorical property) (tensorplay.distributions.OneHotCategoricalStraightThrough property) (tensorplay.distributions.Pareto property) (tensorplay.distributions.Poisson property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.StudentT property) (tensorplay.distributions.TransformedDistribution property) (tensorplay.distributions.Uniform property) (tensorplay.distributions.VonMises property) (tensorplay.distributions.Weibull property) (tensorplay.distributions.Wishart property) batch_sizes (tensorplay.nn.utils.rnn.PackedSequence attribute) batched_powerSGD_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) BatchNorm2d (class in tensorplay.ao.nn.quantized) BatchNorm3d (class in tensorplay.ao.nn.quantized) BatchSampler (class in tensorplay.utils.data) begin_capture_to_if_node() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) begin_capture_to_while_node() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) Bernoulli (class in tensorplay.distributions) bernoulli() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) bernoulli_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) bessel_j0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) bessel_j1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) bessel_y0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) bessel_y1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) Beta (class in tensorplay.distributions) bf16_compress_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks) bf16_compress_wrapper() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks) bfloat16() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |bilinear() (in module tensorplay.functional) (in module tensorplay.nn.functional) binary_cross_entropy() (in module tensorplay) (in module tensorplay.nn.functional) binary_cross_entropy_with_logits() (in module tensorplay) (in module tensorplay.nn.functional) bincount() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) Binomial (class in tensorplay.distributions) binomial() (in module tensorplay.functional) bit_count() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) bit_length() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) (tensorplay.SymInt method) bitwise_and() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_and_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_left_shift() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_left_shift_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_not() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_not_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_or() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_or_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_right_shift() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_right_shift_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_xor() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) bitwise_xor_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) blackman() (in module tensorplay.signal.windows) blackman_window() (in module tensorplay.functional) block_diag() (in module tensorplay._shape_funcs) BlockMask (class in tensorplay.nn.attention) bmm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) BNReLU2d (class in tensorplay.ao.nn.intrinsic) BNReLU3d (class in tensorplay.ao.nn.intrinsic) bootstrap_store_info (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) broadcast() (in module tensorplay.distributed.nn.functional) broadcast_shapes() (in module tensorplay._shape_funcs) broadcast_tensors() (in module tensorplay._shape_funcs) broadcast_to() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) bucketize() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) buffers() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.export.ExportedProgram method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) build() (tensorplay.distributed.elastic.rendezvous.RendezvousStoreInfo class method) build_stage() (in module tensorplay.distributed.pipelining) BytesIOWriteData (class in tensorplay.distributed.checkpoint) ByteSize() (tensorplay.onnx.TensorProto method) BytesStorageMetadata (class in tensorplay.distributed.checkpoint) |
| --- | --- |

## C

| caching_allocator_alloc() (in module tensorplay.cuda) caching_allocator_delete() (in module tensorplay.cuda) caching_allocator_disabled() (in module tensorplay.cuda.memory) caching_allocator_enable() (in module tensorplay.cuda.memory) calculate_gain() (in module tensorplay.nn.init) call_exported() (tensorplay.export.ExportedProgram static method) can_cast() (in module tensorplay.functional) can_device_access_peer() (in module tensorplay.cuda) can_use_cudnn_attention() (in module tensorplay.backends.cuda) can_use_efficient_attention() (in module tensorplay.backends.cuda) can_use_flash_attention() (in module tensorplay.backends.cuda) capitalize() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) capture_begin() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) capture_end() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) cartesian_prod() (in module tensorplay.functional) casefold() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) cast_like() (tensorplay.onnx.OpContext method) cat() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) Categorical (class in tensorplay.distributions) CatTransform (class in tensorplay.distributions) Cauchy (class in tensorplay.distributions) cauchy_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) causal_lower_right() (in module tensorplay.nn.attention.bias) causal_upper_left() (in module tensorplay.nn.attention.bias) CausalBias (class in tensorplay.nn.attention.bias) CausalVariant (class in tensorplay.nn.attention.bias) ccol_indices() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) ccol_indices_copy() (in module tensorplay.functional) cdf() (tensorplay.distributions.Bernoulli method) (tensorplay.distributions.Beta method) (tensorplay.distributions.Binomial method) (tensorplay.distributions.Categorical method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.Independent method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.OneHotCategorical method) (tensorplay.distributions.OneHotCategoricalStraightThrough method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) cdist() (in module tensorplay.functional) ceil() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) (tensorplay.SymInt method) ceil_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) celu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) celu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) center() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) chain_matmul() (in module tensorplay.functional) ChainDataset (class in tensorplay.utils.data) chalf() (tensorplay.nn.attention.bias.CausalBias method) change_current_allocator() (in module tensorplay.cuda) channel_shuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) chebyshev_polynomial_t() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_u() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_v() (in module tensorplay) (in module tensorplay.functional) chebyshev_polynomial_w() (in module tensorplay) (in module tensorplay.functional) check() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_count() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_dag() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_error() (in module tensorplay.cuda) check_next() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_not() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_regex() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_same() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) check_source_highlighted() (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) checkpoint() (in module tensorplay.utils.checkpoint) checkpoint_sequential() (in module tensorplay.utils.checkpoint) CheckpointableTensor (class in tensorplay.distributed.checkpoint) CheckpointException CheckpointPolicy (class in tensorplay.utils.checkpoint) Chi2 (class in tensorplay.distributions) ChildFailedError children() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) cholesky() (in module tensorplay) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) cholesky_ex() (in module tensorplay.linalg) cholesky_inverse() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) cholesky_solve() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) CholeskyExResult (class in tensorplay.linalg) choose_qparams_optimized() (in module tensorplay.functional) chunk() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) chunk_vmap() (in module tensorplay.func) ChunkStorageMetadata (class in tensorplay.distributed.checkpoint) circular_pad_nd() (in module tensorplay) clamp() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) clamp_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) clamp_max() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) clamp_max_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) clamp_min() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) clamp_min_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) classproperty() (in module tensorplay.cuda) clear() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.export.CustomDecompTable method) Clear() (tensorplay.onnx.TensorProto method) clear_autocast_cache() (in module tensorplay) clear_kernel_annotations() (in module tensorplay.cuda.graph_annotations) clear_safe_globals() (in module tensorplay.serialization) ClearExtension() (tensorplay.onnx.TensorProto method) ClearField() (tensorplay.onnx.TensorProto method) clip() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) clip_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) clock_rate() (in module tensorplay.cuda) clone() (in module tensorplay.functional) (tensorplay.autograd.grad_mode.inference_mode method) (tensorplay.autograd.grad_mode.set_grad_enabled method) (tensorplay.export.ExportGraphSignature method) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymInt method) close() (tensorplay.distributed.elastic.multiprocessing.PContext method) (tensorplay.package.PackageExporter method) coalesce() (in module tensorplay) code (tensorplay.export.ExportedProgram property) col2im() (in module tensorplay) (tensorplay.masked.MaskedTensor method) col_indices() (in module tensorplay) col_indices_copy() (in module tensorplay.functional) collect_all() (in module tensorplay.futures) column_stack() (in module tensorplay.functional) ColwiseParallel (class in tensorplay.distributed.tensor.parallel) combinations() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) compile() (in module tensorplay._stax.api) (in module tensorplay.compiler) (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) complex() (in module tensorplay.functional) ComposeTransform (class in tensorplay.distributions) compression_stats() (tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) compute_mask() (tensorplay.ao.pruning.BasePruningMethod method) (tensorplay.ao.pruning.LnStructured method) (tensorplay.ao.pruning.PruningContainer method) (tensorplay.ao.pruning.RandomStructured method) compute_nparams_to_prune() (in module tensorplay.ao.pruning) concat() (in module tensorplay.functional) |ConcatDataset (class in tensorplay.utils.data) concatenate() (in module tensorplay.functional) cond() (in module tensorplay.linalg) configure() (in module tensorplay.distributed.elastic.metrics) (in module tensorplay.distributed.elastic.timer) conj() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) conj_physical() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) conj_physical_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) conjugate() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) (tensorplay.SymFloat method) (tensorplay.SymInt method) consolidate_state_dict() (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) constant() (tensorplay.onnx.GraphBuilder method) constant_() (in module tensorplay.nn.init) constant_pad_nd() (in module tensorplay) ConstantArgument (class in tensorplay.export) Constraint (class in tensorplay.export) ConstraintsExceededError construct_rpc_backend_options() (in module tensorplay.distributed.rpc) construct_rpc_backend_options_handler (tensorplay.distributed.rpc.BackendValue attribute) context (class in tensorplay.distributed.autograd) contiguous() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) ContinuousBernoulli (class in tensorplay.distributions) Conv1d (class in tensorplay.ao.nn.qat) (class in tensorplay.ao.nn.quantized) conv1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv1d_grad_bias() (in module tensorplay) conv1d_grad_input() (in module tensorplay) conv1d_grad_weight() (in module tensorplay) Conv2d (class in tensorplay.ao.nn.qat) (class in tensorplay.ao.nn.quantized) conv2d() (in module tensorplay) (in module tensorplay.nn.functional) conv2d_grad_bias() (in module tensorplay) conv2d_grad_input() (in module tensorplay) conv2d_grad_weight() (in module tensorplay) conv2d_relu() (in module tensorplay) Conv3d (class in tensorplay.ao.nn.qat) (class in tensorplay.ao.nn.quantized) conv3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv3d_grad_bias() (in module tensorplay) conv3d_grad_input() (in module tensorplay) conv3d_grad_weight() (in module tensorplay) conv_depthwise3d() (in module tensorplay) conv_tbc() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose1d_grad_bias() (in module tensorplay) conv_transpose1d_grad_input() (in module tensorplay) conv_transpose1d_grad_weight() (in module tensorplay) conv_transpose2d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose2d_grad_bias() (in module tensorplay) conv_transpose2d_grad_input() (in module tensorplay) conv_transpose2d_grad_weight() (in module tensorplay) conv_transpose3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) conv_transpose3d_grad_bias() (in module tensorplay) conv_transpose3d_grad_input() (in module tensorplay) conv_transpose3d_grad_weight() (in module tensorplay) convert() (in module tensorplay.ao.quantization) convert_model() (in module tensorplay.serialization) convert_to_mega() (in module tensorplay.serialization) convolution() (in module tensorplay.functional) convolution_backward_overrideable() (in module tensorplay) convolution_overrideable() (in module tensorplay) ConvReLU1d (class in tensorplay.ao.nn.intrinsic) ConvReLU2d (class in tensorplay.ao.nn.intrinsic) ConvReLU3d (class in tensorplay.ao.nn.intrinsic) copy() (in module tensorplay) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.profiler.EventList method) copy_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) copy_sparse_to_sparse_() (in module tensorplay) CopyFrom() (tensorplay.onnx.TensorProto method) copysign() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) copysign_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) CorrCholeskyTransform (class in tensorplay.distributions) corrcoef() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) cos() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) cos_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) cosh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) cosh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) cosine() (in module tensorplay.signal.windows) cosine_embedding_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) cosine_similarity() (in module tensorplay.functional) (in module tensorplay.nn.functional) cot() (tensorplay.nn.attention.bias.CausalBias method) count() (tensorplay.ao.quantization.QConfig method) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.Kernel method) (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.distributed.rpc.BackendValue method) (tensorplay.linalg.CholeskyExResult method) (tensorplay.linalg.EighResult method) (tensorplay.linalg.EigResult method) (tensorplay.linalg.LstsqResult method) (tensorplay.linalg.QRResult method) (tensorplay.linalg.SlogdetResult method) (tensorplay.linalg.SVDResult method) (tensorplay.monitor.Stat method) (tensorplay.nn.utils.rnn.PackedSequence method) (tensorplay.profiler.EventList method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) (tensorplay.Size method) count_nonzero() (in module tensorplay.functional) cov() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) cpu() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) CPUOffload (class in tensorplay.distributed.fsdp) CPUOffloadPolicy (class in tensorplay.distributed.fsdp) create_block_mask() (in module tensorplay.nn.attention) create_handler() (in module tensorplay.distributed.elastic.rendezvous) create_mask() (in module tensorplay.nn.attention) create_selective_checkpoint_contexts() (in module tensorplay.utils.checkpoint) cross() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) cross_entropy() (in module tensorplay.nn.functional) cross_entropy_loss() (in module tensorplay) crow_indices() (in module tensorplay) crow_indices_copy() (in module tensorplay.functional) csc() (tensorplay.nn.attention.bias.CausalBias method) ctc_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) cuBLASModule (class in tensorplay.backends.cuda) cuda() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) CudaError CUDAGraph (class in tensorplay.cuda.graphs) CUDAPluggableAllocator (class in tensorplay.cuda) cudart() (in module tensorplay.cuda) cudnn_affine_grid_generator() (in module tensorplay.functional) cudnn_batch_norm() (in module tensorplay.functional) cudnn_convolution() (in module tensorplay.functional) cudnn_convolution_add_relu() (in module tensorplay.functional) cudnn_convolution_relu() (in module tensorplay.functional) cudnn_convolution_transpose() (in module tensorplay.functional) cudnn_grid_sampler() (in module tensorplay.functional) cudnn_is_acceptable() (in module tensorplay.functional) cuFFTPlanCache (class in tensorplay.backends.cuda) cummax() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) cummin() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) cumprod() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.nn.attention.bias.CausalBias method) cumprod_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) cumsum() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.nn.attention.bias.CausalBias method) cumsum_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) cumulative_trapezoid() (in module tensorplay.functional) CumulativeDistributionTransform (class in tensorplay.distributions) current_accelerator() (in module tensorplay.accelerator) current_blas_handle() (in module tensorplay.cuda) current_device() (in module tensorplay.cuda) current_device_index() (in module tensorplay.accelerator) current_flash_attention_impl() (in module tensorplay.nn.attention) current_solver_handle() (in module tensorplay.cuda) current_stream() (in module tensorplay.cuda) custom_bwd() (in module tensorplay) custom_from_mask() (in module tensorplay.ao.pruning) custom_fwd() (in module tensorplay) custom_op() (in module tensorplay.library) CustomDecompTable (class in tensorplay.export) CustomFromMask (class in tensorplay.ao.pruning) CustomObjArgument (class in tensorplay.export) CyclingIterator (class in tensorplay.distributed.elastic.utils.data) |
| --- | --- |

## D

| data (tensorplay.nn.utils.rnn.PackedSequence attribute) data() (in module tensorplay) data_ptr() (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.UntypedStorage method) DataLoader (class in tensorplay.utils.data) DataParallelMeshDims (class in tensorplay.distributed.fsdp) Dataset (class in tensorplay.utils.data) debug_dump() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) debug_unwrap() (in module tensorplay.func) default_collate() (in module tensorplay.utils.data) default_convert() (in module tensorplay.utils.data) default_decompositions() (in module tensorplay.export) default_dynamic_qconfig (in module tensorplay.ao.quantization) default_dynamic_quant_observer (in module tensorplay.ao.quantization) default_factory (tensorplay.autograd.profiler_util.StringTable attribute) default_generator() (in module tensorplay.random) default_observer (in module tensorplay.ao.quantization) default_per_channel_qconfig (in module tensorplay.ao.quantization) default_qconfig (in module tensorplay.ao.quantization) default_restore_location() (in module tensorplay.serialization) (in module tensorplay.serialization.api) default_stream() (in module tensorplay.cuda) default_tolerances() (in module tensorplay.testing) default_weight_observer (in module tensorplay.ao.quantization) default_weight_only_qconfig (in module tensorplay.ao.quantization) DefaultLoadPlanner (class in tensorplay.distributed.checkpoint) DefaultSavePlanner (class in tensorplay.distributed.checkpoint) DeferredCudaCallError define() (in module tensorplay.library) (tensorplay.library.Library method) defined() (tensorplay.nn.attention.bias.CausalBias method) deg2rad() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) deg2rad_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) denied_modules() (tensorplay.package.PackageExporter method) denominator (tensorplay.distributed.elastic.multiprocessing.Std attribute) (tensorplay.export.WeightType attribute) (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.nn.attention.bias.CausalVariant attribute) (tensorplay.QScheme attribute) dense_dim() (in module tensorplay) deny() (tensorplay.package.PackageExporter method) dependency_graph_string() (tensorplay.package.PackageExporter method) DeQuantize (class in tensorplay.ao.nn.quantized) dequantize() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) DeQuantStub (class in tensorplay.ao.quantization) DerivedDim (in module tensorplay.export) deserialize() (tensorplay.export.ExportedProgram class method) det() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) detach() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) detach_() (in module tensorplay.functional) detach_copy() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) detect_anomaly (class in tensorplay.autograd.anomaly_mode) Device (class in tensorplay) device (class in tensorplay.cuda) (tensorplay.autograd.profiler_util.Kernel attribute) (tensorplay.Generator property) device_count() (in module tensorplay.accelerator) (in module tensorplay.cuda) device_kernels() (tensorplay.profiler.EventList method) (tensorplay.profiler.profile method) (tensorplay.profiler.profiler.profile method) device_memory_used() (in module tensorplay.cuda) device_of (class in tensorplay.cuda) device_time (tensorplay.profiler.FunctionEvent property) (tensorplay.profiler.FunctionEventAvg property) device_time_by_span (tensorplay.profiler.EventList property) (tensorplay.profiler.profile property) (tensorplay.profiler.profiler.profile property) device_time_total (tensorplay.profiler.FunctionEvent property) DeviceMesh (class in tensorplay.distributed.device_mesh) DeviceMismatchError DeviceType (class in tensorplay) diag() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) diag_embed() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) diagflat() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) diagonal() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) diagonal_copy() (in module tensorplay.functional) diagonal_scatter() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) diff() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) digamma() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) digamma_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Dim (class in tensorplay.export) dim() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) |dim_order() (tensorplay.nn.attention.bias.CausalBias method) dims() (in module tensorplay.export) dirac_() (in module tensorplay.nn.init) Directory (class in tensorplay.package) Dirichlet (class in tensorplay.distributions) disable() (in module tensorplay.compiler) (in module tensorplay.cuda.tunable) disable_capture() (in module tensorplay.compiler) disallow_in_graph() (in module tensorplay.compiler) DiscardUnknownFields() (tensorplay.onnx.TensorProto method) DispatchKey (class in tensorplay) dist() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) DistAutogradContext (class in tensorplay.distributed.autograd) distribute_module() (in module tensorplay.distributed.tensor) distribute_tensor() (in module tensorplay.distributed.tensor) DistributedOptimizer (class in tensorplay.distributed.optim) Distribution (class in tensorplay.distributions) div() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) div_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) divide() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) divide_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) done() (tensorplay.Future method) (tensorplay.futures.Future method) dot() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) double() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) draft_export() (in module tensorplay.export) dropout() (in module tensorplay) (in module tensorplay.nn.functional) dropout1d() (in module tensorplay.nn.functional) dropout2d() (in module tensorplay.nn.functional) dropout3d() (in module tensorplay.nn.functional) dropout_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) dsplit() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) dstack() (in module tensorplay.functional) DTensor (class in tensorplay.distributed.tensor) DType (class in tensorplay) duration (tensorplay.autograd.profiler_util.Kernel attribute) DynamicRendezvousHandler (class in tensorplay.distributed.elastic.rendezvous) |
| --- | --- |

## E

| eig() (in module tensorplay.linalg) eigenvalues (tensorplay.linalg.EighResult attribute) (tensorplay.linalg.EigResult attribute) eigenvectors (tensorplay.linalg.EighResult attribute) (tensorplay.linalg.EigResult attribute) eigh() (in module tensorplay.linalg) EighResult (class in tensorplay.linalg) EigResult (class in tensorplay.linalg) eigvals() (in module tensorplay.linalg) eigvalsh() (in module tensorplay.linalg) einsum() (in module tensorplay._einsum) elapsed_time() (tensorplay.cuda.streams.Event method) elapsed_us() (tensorplay.autograd.profiler_util.Interval method) elastic_launch (class in tensorplay.distributed.launcher) ElasticAgent (class in tensorplay.distributed.elastic.agent.server) ElasticDistributedSampler (class in tensorplay.distributed.elastic.utils.data) element_size() (tensorplay.nn.attention.bias.CausalBias method) ELU (class in tensorplay.ao.nn.quantized) elu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) elu_() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) embedding() (in module tensorplay.functional) (in module tensorplay.nn.functional) embedding_bag() (in module tensorplay) (in module tensorplay.nn.functional) embedding_renorm_() (in module tensorplay.functional) emit_itt (class in tensorplay.profiler) (class in tensorplay.profiler.profiler) emit_nvtx (class in tensorplay.profiler) (class in tensorplay.profiler.profiler) empty() (in module tensorplay) (in module tensorplay.distributed.tensor) empty_cache() (in module tensorplay.cuda) empty_like() (in module tensorplay.functional) empty_permuted() (in module tensorplay.functional) empty_quantized() (in module tensorplay.functional) empty_strided() (in module tensorplay.functional) EmptyMatchError enable() (in module tensorplay.cuda.tunable) enable_grad (class in tensorplay.autograd.grad_mode) encode() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) end_capture_to_conditional_node() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) endswith() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) enter_dual_level() (in module tensorplay.autograd.forward_ad) entr() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) entropy() (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.VonMises method) enumerate_support() (tensorplay.distributions.Beta method) (tensorplay.distributions.Cauchy method) (tensorplay.distributions.Chi2 method) (tensorplay.distributions.ContinuousBernoulli method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.Exponential method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.Gamma method) (tensorplay.distributions.GeneralizedPareto method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.HalfCauchy method) (tensorplay.distributions.HalfNormal method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.Laplace method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.Normal method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.Uniform method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) epoch (tensorplay.distributed.elastic.utils.data.CyclingIterator property) eq() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) eq_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) equal() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) EqualityConstraint (class in tensorplay.export) erf() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) erf_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) erfc() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) erfc_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) erfcx() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) erfinv() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) erfinv_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) eval() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |Event (class in tensorplay.cuda.streams) (class in tensorplay.distributed.elastic.events) (class in tensorplay.monitor) event_shape (tensorplay.distributions.Bernoulli property) (tensorplay.distributions.Beta property) (tensorplay.distributions.Binomial property) (tensorplay.distributions.Categorical property) (tensorplay.distributions.Cauchy property) (tensorplay.distributions.Chi2 property) (tensorplay.distributions.ContinuousBernoulli property) (tensorplay.distributions.Dirichlet property) (tensorplay.distributions.Distribution property) (tensorplay.distributions.Exponential property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.FisherSnedecor property) (tensorplay.distributions.Gamma property) (tensorplay.distributions.GeneralizedPareto property) (tensorplay.distributions.Geometric property) (tensorplay.distributions.Gumbel property) (tensorplay.distributions.HalfCauchy property) (tensorplay.distributions.HalfNormal property) (tensorplay.distributions.Independent property) (tensorplay.distributions.InverseGamma property) (tensorplay.distributions.Kumaraswamy property) (tensorplay.distributions.Laplace property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.LogNormal property) (tensorplay.distributions.LowRankMultivariateNormal property) (tensorplay.distributions.MixtureSameFamily property) (tensorplay.distributions.Multinomial property) (tensorplay.distributions.MultivariateNormal property) (tensorplay.distributions.NegativeBinomial property) (tensorplay.distributions.Normal property) (tensorplay.distributions.OneHotCategorical property) (tensorplay.distributions.OneHotCategoricalStraightThrough property) (tensorplay.distributions.Pareto property) (tensorplay.distributions.Poisson property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.StudentT property) (tensorplay.distributions.TransformedDistribution property) (tensorplay.distributions.Uniform property) (tensorplay.distributions.VonMises property) (tensorplay.distributions.Weibull property) (tensorplay.distributions.Wishart property) EventList (class in tensorplay.autograd.profiler_util) (class in tensorplay.profiler) exit_dual_level() (in module tensorplay.autograd.forward_ad) exp() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) exp2() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) exp2_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) exp_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) expand() (in module tensorplay) (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) expand_as() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) expand_copy() (in module tensorplay.functional) expandtabs() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) expect_bool() (tensorplay.SymBool method) expect_float() (tensorplay.SymFloat method) expect_int() (tensorplay.SymInt method) expires() (in module tensorplay.distributed.elastic.timer) expit() (in module tensorplay.special._error) expm1() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) expm1_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Exponential (class in tensorplay.distributions) exponential() (in module tensorplay.signal.windows) exponential_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) ExponentialFamily (class in tensorplay.distributions) export() (in module tensorplay.compiler) (in module tensorplay.export) (in module tensorplay.onnx) export_chrome_trace() (tensorplay.autograd.profiler_util.EventList method) export_dot() (in module tensorplay.cuda.graphs) export_for_training() (in module tensorplay.export) ExportBackwardSignature (class in tensorplay.export) ExportedProgram (class in tensorplay.export) ExportGraphSignature (class in tensorplay.export) ExpTransform (class in tensorplay.distributions) extend() (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) Extensions (tensorplay.onnx.TensorProto attribute) extern() (tensorplay.package.PackageExporter method) ExternalStream (class in tensorplay.cuda.streams) externed_modules() (tensorplay.package.PackageExporter method) extra_repr() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) eye() (in module tensorplay.functional) eye_() (in module tensorplay.nn.init) |
| --- | --- |

## F

| fake_quantize_per_channel() (in module tensorplay.ao.quantization) fake_quantize_per_channel_affine() (in module tensorplay.functional) fake_quantize_per_channel_affine_cachemask() (in module tensorplay) fake_quantize_per_tensor() (in module tensorplay.ao.quantization) fake_quantize_per_tensor_affine() (in module tensorplay.functional) fake_quantize_per_tensor_affine_cachemask() (in module tensorplay) FakeQuantize (class in tensorplay.ao.quantization) fallback() (tensorplay.library.Library method) fbgemm_linear_fp16_weight() (in module tensorplay.functional) fbgemm_linear_fp16_weight_fp32_activation() (in module tensorplay.functional) fbgemm_linear_int8_weight() (in module tensorplay.functional) fbgemm_linear_int8_weight_fp32_activation() (in module tensorplay.functional) fbgemm_linear_quantize_weight() (in module tensorplay.functional) fbgemm_pack_gemm_matrix_fp16() (in module tensorplay.functional) fbgemm_pack_quantized_matrix() (in module tensorplay.functional) feature_alpha_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) feature_alpha_dropout_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) feature_dropout() (in module tensorplay.functional) (in module tensorplay.nn.functional) feature_dropout_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) fft() (in module tensorplay.fft._transforms) fft2() (in module tensorplay.fft._transforms) fft_fft() (in module tensorplay) fft_fft2() (in module tensorplay) fft_fftfreq() (in module tensorplay) fft_fftn() (in module tensorplay) fft_fftshift() (in module tensorplay) fft_hfft() (in module tensorplay) fft_hfft2() (in module tensorplay) fft_hfftn() (in module tensorplay) fft_ifft() (in module tensorplay) fft_ifft2() (in module tensorplay) fft_ifftn() (in module tensorplay) fft_ifftshift() (in module tensorplay) fft_ihfft() (in module tensorplay) fft_ihfft2() (in module tensorplay) fft_ihfftn() (in module tensorplay) fft_irfft() (in module tensorplay) fft_irfft2() (in module tensorplay) fft_irfftn() (in module tensorplay) fft_rfft() (in module tensorplay) fft_rfft2() (in module tensorplay) fft_rfftfreq() (in module tensorplay) fft_rfftn() (in module tensorplay) fftfreq() (in module tensorplay.fft._frequency) fftn() (in module tensorplay.fft._transforms) fftshift() (in module tensorplay.fft._frequency) file_structure() (tensorplay.package.PackageImporter method) FileCheck (class in tensorplay) (class in tensorplay.testing) FileSystem (class in tensorplay.distributed.checkpoint) FileSystemBase (class in tensorplay.distributed.checkpoint) FileSystemReader (class in tensorplay.distributed.checkpoint) FileSystemWriter (class in tensorplay.distributed.checkpoint) fill() (in module tensorplay.functional) fill_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) fill_diagonal_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) find() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) FindInitializationErrors() (tensorplay.onnx.TensorProto method) finfo (class in tensorplay) FisherSnedecor (class in tensorplay.distributions) fix() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fix_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) FixedQParamsFakeQuantize (class in tensorplay.ao.quantization) FixedQParamsObserver (class in tensorplay.ao.quantization) flags() (in module tensorplay.backends.nnpack) FlatArgsAdapter (class in tensorplay.export) Flatten (class in tensorplay.nn.modules.flatten) flatten() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) flatten_dense_tensors() (in module tensorplay) flip() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) fliplr() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) flipud() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) float() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |float_power() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) float_power_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) FloatFunctional (class in tensorplay.ao.nn.quantized) floor() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) (tensorplay.SymInt method) floor_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) floor_divide() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) floor_divide_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fmax() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fmin() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fmod() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fmod_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fold() (in module tensorplay.nn.functional) fork_rng() (in module tensorplay) (in module tensorplay.random) format() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) format_map() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) FormattedTimesMixin (class in tensorplay.autograd.profiler_util) forward() (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) forward_add() (in module tensorplay) forward_cos() (in module tensorplay) forward_div() (in module tensorplay) forward_exp() (in module tensorplay) forward_log() (in module tensorplay) forward_mm() (in module tensorplay) forward_mul() (in module tensorplay) forward_neg() (in module tensorplay) forward_pow() (in module tensorplay) forward_relu() (in module tensorplay) forward_shape() (tensorplay.distributions.AbsTransform method) (tensorplay.distributions.CatTransform method) (tensorplay.distributions.CumulativeDistributionTransform method) (tensorplay.distributions.ExpTransform method) (tensorplay.distributions.LowerCholeskyTransform method) (tensorplay.distributions.PositiveDefiniteTransform method) (tensorplay.distributions.SigmoidTransform method) (tensorplay.distributions.SoftplusTransform method) (tensorplay.distributions.StackTransform method) (tensorplay.distributions.TanhTransform method) (tensorplay.distributions.Transform method) forward_sigmoid() (in module tensorplay) forward_sin() (in module tensorplay) forward_sqrt() (in module tensorplay) forward_sub() (in module tensorplay) forward_tanh() (in module tensorplay) fp16_compress_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks) fp16_compress_wrapper() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks) frac() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) frac_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) fractional_max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) fractional_max_pool2d_with_indices() (in module tensorplay.nn.functional) fractional_max_pool3d() (in module tensorplay) (in module tensorplay.nn.functional) fractional_max_pool3d_with_indices() (in module tensorplay.nn.functional) frexp() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) frobenius_norm() (in module tensorplay) from_bytes() (tensorplay.distributed.elastic.multiprocessing.Std class method) (tensorplay.export.WeightType class method) (tensorplay.Layout class method) (tensorplay.MemoryFormat class method) (tensorplay.nn.attention.bias.CausalVariant class method) (tensorplay.QScheme class method) from_dlpack() (in module tensorplay) from_file() (in module tensorplay.functional) from_float() (tensorplay.ao.nn.intrinsic.ConvReLU1d class method) (tensorplay.ao.nn.intrinsic.ConvReLU2d class method) (tensorplay.ao.nn.intrinsic.ConvReLU3d class method) (tensorplay.ao.nn.intrinsic.LinearReLU class method) from_group() (tensorplay.distributed.device_mesh.DeviceMesh class method) from_ipc_handle() (tensorplay.cuda.streams.Event class method) from_kv_blocks() (tensorplay.nn.attention.BlockMask class method) from_local() (in module tensorplay.distributed.tensor) from_numpy() (in module tensorplay) from_reference() (tensorplay.ao.nn.quantized.Conv1d class method) (tensorplay.ao.nn.quantized.Conv2d class method) (tensorplay.ao.nn.quantized.Conv3d class method) (tensorplay.ao.nn.quantized.QuantizedLinear class method) from_str() (tensorplay.distributed.elastic.multiprocessing.Std class method) frombuffer() (in module tensorplay) fromkeys() (tensorplay.autograd.profiler_util.StringTable class method) (tensorplay.export.CustomDecompTable class method) FromString() (tensorplay.onnx.TensorProto class method) FSDPModule (class in tensorplay.distributed.fsdp) full() (in module tensorplay) (in module tensorplay.distributed.tensor) full_like() (in module tensorplay.functional) FullOptimStateDictConfig (class in tensorplay.distributed.fsdp) FullStateDictConfig (class in tensorplay.distributed.fsdp) fully_shard() (in module tensorplay.distributed.fsdp) FullyShardedDataParallel (class in tensorplay.distributed.fsdp) Function (class in tensorplay.autograd.function) functional_call() (in module tensorplay.func) (in module tensorplay.nn.utils.stateless) functionalize() (in module tensorplay.func) FunctionEvent (class in tensorplay.autograd.profiler_util) (class in tensorplay.profiler) FunctionEventAvg (class in tensorplay.autograd.profiler_util) (class in tensorplay.profiler) FunctionMeta (class in tensorplay.autograd.function) fuse_modules() (in module tensorplay.ao.quantization) fuse_modules_qat() (in module tensorplay.ao.quantization) fused_moving_avg_obs_fake_quant() (in module tensorplay.functional) fused_mul_add() (in module tensorplay) fused_rope() (in module tensorplay) fused_swiglu() (in module tensorplay) FusedMovingAvgObsFakeQuantize (class in tensorplay.ao.quantization) Future (class in tensorplay) (class in tensorplay.futures) (in module tensorplay.distributed.rpc) FXFloatFunctional (class in tensorplay.ao.nn.quantized) |
| --- | --- |

## G

| Gamma (class in tensorplay.distributions) gammainc() (in module tensorplay) (in module tensorplay.functional) gammaincc() (in module tensorplay) (in module tensorplay.functional) gammaln() (in module tensorplay.special._gamma) gather() (in module tensorplay.distributed.nn.functional) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) gaussian() (in module tensorplay.signal.windows) gaussian_nll_loss() (in module tensorplay.nn.functional) gcd() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) gcd_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) GdsFile (class in tensorplay.cuda.gds) ge() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) ge_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) gelu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) gelu_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) general_cosine() (in module tensorplay.signal.windows) general_hamming() (in module tensorplay.signal.windows) GeneralizedPareto (class in tensorplay.distributions) Generator (class in tensorplay) Geometric (class in tensorplay.distributions) geometric_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) geqrf() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ger() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) get() (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.distributed.elastic.rendezvous.RendezvousParameters method) (tensorplay.monitor.Stat method) get_all_sharing_strategies() (in module tensorplay.multiprocessing) get_allocator_backend() (in module tensorplay.cuda) get_arch_list() (in module tensorplay.cuda) get_as_bool() (tensorplay.distributed.elastic.rendezvous.RendezvousParameters method) get_as_int() (tensorplay.distributed.elastic.rendezvous.RendezvousParameters method) get_autocast_cpu_dtype() (in module tensorplay.amp.autocast_mode) get_autocast_dtype() (in module tensorplay) get_autocast_gpu_dtype() (in module tensorplay.amp.autocast_mode) get_backend() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) get_backend_capabilities() (in module tensorplay.compiler) get_backoff_factor() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) get_buffer() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_coordinate() (tensorplay.distributed.device_mesh.DeviceMesh method) get_cpu_capability() (in module tensorplay.backends.cpu) get_crc32_options() (in module tensorplay.serialization) get_default_backend() (in module tensorplay.compiler) get_default_device() (in module tensorplay) get_default_dtype() (in module tensorplay) get_default_load_endianness() (in module tensorplay.serialization) get_default_mmap_options() (in module tensorplay.serialization) get_default_qat_qconfig() (in module tensorplay.ao.quantization) get_default_qconfig() (in module tensorplay.ao.quantization) get_deterministic_debug_mode() (in module tensorplay) get_device() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) get_device_capability() (in module tensorplay.accelerator) (in module tensorplay.cuda) get_device_name() (in module tensorplay.cuda) get_device_properties() (in module tensorplay.cuda) get_entrypoint_name() (tensorplay.distributed.elastic.agent.server.WorkerSpec method) get_env_variable_or_raise() (in module tensorplay.distributed.elastic.utils) get_extra_state() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_filename() (in module tensorplay.cuda.tunable) get_float32_matmul_precision() (in module tensorplay) get_gencode_flags() (in module tensorplay.cuda) get_gradients() (in module tensorplay.distributed.autograd) get_growth_factor() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) get_growth_interval() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) get_ignored_functions() (in module tensorplay.overrides) get_kernel() (in module tensorplay.library) get_kernel_annotations() (in module tensorplay.cuda.graph_annotations) get_max_tuning_duration() (in module tensorplay.cuda.tunable) get_max_tuning_samples() (in module tensorplay.cuda.tunable) get_model_state_dict() (in module tensorplay.distributed.checkpoint) get_name() (tensorplay.package.Importer method) (tensorplay.package.PackageImporter method) get_num_threads() (in module tensorplay) get_numerical_jacobian_wrt_specific_input() (in module tensorplay.autograd.gradcheck) get_observer_state_dict() (in module tensorplay.ao.quantization) get_optimizer_state_dict() (in module tensorplay.distributed.checkpoint) get_overridable_functions() (in module tensorplay.overrides) |get_parallel_info() (in module tensorplay) get_param_to_buffer() (tensorplay.export.ExportGraphSignature method) get_parameter() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_per_process_memory_fraction() (in module tensorplay.cuda) get_qconv() (tensorplay.ao.nn.quantized.Conv1d class method) (tensorplay.ao.nn.quantized.Conv2d class method) (tensorplay.ao.nn.quantized.Conv3d class method) get_rdeps() (tensorplay.package.PackageExporter method) get_replace_hook() (tensorplay.export.ExportGraphSignature method) get_results() (in module tensorplay.cuda.tunable) get_rng_state() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) get_rng_state_all() (in module tensorplay.cuda.random) get_rpc_timeout() (in module tensorplay.distributed.rpc) get_run_id() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) get_safe_globals() (in module tensorplay.serialization) get_scale() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) get_sharing_strategy() (in module tensorplay.multiprocessing) get_socket_with_port() (in module tensorplay.distributed.elastic.utils) get_state() (tensorplay.Generator method) get_state_dict() (in module tensorplay.distributed.checkpoint) get_stream_from_external() (in module tensorplay.cuda) get_submodule() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) get_sync_debug_mode() (in module tensorplay.cuda) get_testing_overrides() (in module tensorplay.overrides) get_thread_num() (in module tensorplay) get_tolerances() (in module tensorplay.testing) get_unique_id() (tensorplay.package.PackageExporter method) get_unsafe_globals_in_checkpoint() (in module tensorplay.serialization) get_worker_group() (tensorplay.distributed.elastic.agent.server.ElasticAgent method) get_worker_info() (in module tensorplay.distributed.rpc) (in module tensorplay.utils.data) global_unstructured() (in module tensorplay.ao.pruning) GlobGroup (class in tensorplay.package) glu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) glu_backward_jvp() (in module tensorplay) glu_jvp() (in module tensorplay) grad() (in module tensorplay.autograd) (in module tensorplay.func) grad_and_value() (in module tensorplay.func) gradcheck() (in module tensorplay.autograd.gradcheck) GradcheckError gradgradcheck() (in module tensorplay.autograd.gradcheck) gradient() (in module tensorplay) GradScaler (class in tensorplay) Graph (class in tensorplay.accelerator) graph() (in module tensorplay.cuda.graphs) graph_pool_handle() (in module tensorplay.cuda.graphs) GraphBuilder (class in tensorplay.onnx) GraphCaptureError GraphExecGroup (class in tensorplay.utils.checkpoint) GraphSignature (class in tensorplay.export) greater() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) greater_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) greater_equal() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) greater_equal_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) GreenContext (class in tensorplay.cuda.green_contexts) grid_sample() (in module tensorplay.nn.functional) grid_sampler() (in module tensorplay) grid_sampler_2d() (in module tensorplay.functional) grid_sampler_3d() (in module tensorplay.functional) group_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) grouped_mm() (in module tensorplay) (in module tensorplay.nn.functional) GroupMember (in module tensorplay.distributed.distributed_core) GroupNorm (class in tensorplay.ao.nn.quantized) gru() (in module tensorplay.functional) gru_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) gt() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) gt_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) guard_bool() (tensorplay.SymBool method) guard_float() (tensorplay.SymFloat method) guard_int() (tensorplay.SymInt method) guard_or_false() (tensorplay.SymBool method) guard_or_true() (tensorplay.SymBool method) guard_size_oblivious() (tensorplay.SymBool method) Gumbel (class in tensorplay.distributions) gumbel_softmax() (in module tensorplay.nn.functional) |
| --- | --- |

## H

| half() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) HalfCauchy (class in tensorplay.distributions) HalfNormal (class in tensorplay.distributions) hamming() (in module tensorplay.signal.windows) hamming_window() (in module tensorplay.functional) handle_tensorplay_function() (in module tensorplay.overrides) hann() (in module tensorplay.signal.windows) hann_window() (in module tensorplay.functional) hardshrink() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) hardshrink_backward() (tensorplay.nn.attention.bias.CausalBias method) Hardsigmoid (class in tensorplay.ao.nn.quantized) |hardsigmoid() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) hardsigmoid_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) Hardswish (class in tensorplay.ao.nn.quantized) hardswish() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) hardswish_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) hardtanh() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) hardtanh_() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) has_file() (tensorplay.package.Directory method) has_hint() (tensorplay.SymBool method) (tensorplay.SymFloat method) (tensorplay.SymInt method) has_tensorplay_function() (in module tensorplay.overrides) has_uninitialized_params() (tensorplay.nn.modules.lazy.LazyModuleMixin method) HasExtension() (tensorplay.onnx.TensorProto method) HasField() (tensorplay.onnx.TensorProto method) hash_tensor() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) heaviside() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) heaviside_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) hermite_polynomial_h() (in module tensorplay) (in module tensorplay.functional) hermite_polynomial_he() (in module tensorplay) (in module tensorplay.functional) hessian() (in module tensorplay.autograd.functional) (in module tensorplay.func) hex() (tensorplay.SymFloat method) hfft() (in module tensorplay.fft._transforms) hfft2() (in module tensorplay.fft._transforms) hfftn() (in module tensorplay.fft._transforms) HierarchicalModelAverager (class in tensorplay.distributed.algorithms.model_averaging) hinge_embedding_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) histc() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) histogram() (in module tensorplay._composite_funcs) (tensorplay.nn.attention.bias.CausalBias method) histogramdd() (in module tensorplay.functional) HistogramObserver (class in tensorplay.ao.quantization) hook_with_zero_step() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook) hook_with_zero_step_interleaved() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook) host_memory_stats() (in module tensorplay.cuda) host_memory_stats_as_nested_dict() (in module tensorplay.cuda) householder_product() (in module tensorplay.linalg) hsplit() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) hspmm() (in module tensorplay) hstack() (in module tensorplay.functional) huber_loss() (in module tensorplay) (in module tensorplay.nn.functional) HuggingFaceStorageReader (class in tensorplay.distributed.checkpoint) HuggingFaceStorageWriter (class in tensorplay.distributed.checkpoint) hvp() (in module tensorplay.autograd.functional) hypot() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) hypot_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) |
| --- | --- |

## I

| i0() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) i0_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) i0e() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) i1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) i1e() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) icdf() (tensorplay.distributions.Bernoulli method) (tensorplay.distributions.Beta method) (tensorplay.distributions.Binomial method) (tensorplay.distributions.Categorical method) (tensorplay.distributions.Chi2 method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.Gamma method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.Independent method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.OneHotCategorical method) (tensorplay.distributions.OneHotCategoricalStraightThrough method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) id() (tensorplay.package.PackageImporter method) Identity (class in tensorplay.ao.pruning) identity() (in module tensorplay.ao.pruning) ifft() (in module tensorplay.fft._transforms) ifft2() (in module tensorplay.fft._transforms) ifftn() (in module tensorplay.fft._transforms) ifftshift() (in module tensorplay.fft._frequency) igamma() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) igamma_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) igammac() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) igammac_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) ihfft() (in module tensorplay.fft._transforms) ihfft2() (in module tensorplay.fft._transforms) ihfftn() (in module tensorplay.fft._transforms) iinfo (class in tensorplay) im2col() (in module tensorplay) (tensorplay.masked.MaskedTensor method) imag (tensorplay.distributed.elastic.multiprocessing.Std attribute) (tensorplay.export.WeightType attribute) (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.nn.attention.bias.CausalVariant attribute) (tensorplay.QScheme attribute) imag() (in module tensorplay.functional) impl() (in module tensorplay.library) (tensorplay.library.Library method) impl_abstract() (in module tensorplay.library) import_module() (tensorplay.package.Importer method) (tensorplay.package.PackageImporter method) Importer (class in tensorplay.package) in_interval() (tensorplay.autograd.profiler_util.MemRecordsAcc method) in_parallel_region() (in module tensorplay) Independent (class in tensorplay.distributions) IndependentTransform (class in tensorplay.distributions) index() (in module tensorplay) (tensorplay.ao.quantization.QConfig method) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.Kernel method) (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.distributed.rpc.BackendValue method) (tensorplay.linalg.CholeskyExResult method) (tensorplay.linalg.EighResult method) (tensorplay.linalg.EigResult method) (tensorplay.linalg.LstsqResult method) (tensorplay.linalg.QRResult method) (tensorplay.linalg.SlogdetResult method) (tensorplay.linalg.SVDResult method) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.utils.rnn.PackedSequence method) (tensorplay.profiler.EventList method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) (tensorplay.Size method) index_add() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) index_add_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_copy() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) index_copy_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_fill() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_fill_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_put() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_put_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) index_reduce() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) index_reduce_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) index_select() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) indices() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) indices_copy() (in module tensorplay.functional) infer_schema() (in module tensorplay.library) inference_mode (class in tensorplay.autograd.grad_mode) info (tensorplay.linalg.CholeskyExResult attribute) init() (in module tensorplay.cuda) init_backend() (in module tensorplay.distributed.rpc) init_backend_handler (tensorplay.distributed.rpc.BackendValue attribute) init_device_mesh() (in module tensorplay.distributed.device_mesh) init_rpc() (in module tensorplay.distributed.rpc) initial_seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) initialize_parameters() (tensorplay.nn.modules.lazy.LazyModuleMixin method) inner() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) inner_backward_other() (in module tensorplay) inner_backward_self() (in module tensorplay) InplaceFunction (class in tensorplay.autograd.function) InputKind (class in tensorplay.export) InputSpec (class in tensorplay.export) insert() (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) inspect_checkpoint() (in module tensorplay) (in module tensorplay.serialization) instance_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) InstanceNorm1d (class in tensorplay.ao.nn.quantized) InstanceNorm2d (class in tensorplay.ao.nn.quantized) InstanceNorm3d (class in tensorplay.ao.nn.quantized) instantiate() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) int_repr() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) interface() (in module tensorplay.distributed.nn.api) intern() (tensorplay.package.PackageExporter method) interned_modules() (tensorplay.package.PackageExporter method) interpolate() (in module tensorplay.nn.functional) Interval (class in tensorplay.autograd.profiler_util) (class in tensorplay.profiler) inv (tensorplay.distributions.AbsTransform property) (tensorplay.distributions.AffineTransform property) (tensorplay.distributions.CatTransform property) (tensorplay.distributions.CorrCholeskyTransform property) (tensorplay.distributions.CumulativeDistributionTransform property) (tensorplay.distributions.ExpTransform property) (tensorplay.distributions.IndependentTransform property) (tensorplay.distributions.LowerCholeskyTransform property) (tensorplay.distributions.PositiveDefiniteTransform property) (tensorplay.distributions.PowerTransform property) (tensorplay.distributions.ReshapeTransform property) (tensorplay.distributions.SigmoidTransform property) (tensorplay.distributions.SoftmaxTransform property) (tensorplay.distributions.SoftplusTransform property) (tensorplay.distributions.StackTransform property) (tensorplay.distributions.StickBreakingTransform property) (tensorplay.distributions.TanhTransform property) (tensorplay.distributions.Transform property) inv() (in module tensorplay.linalg) |inv_ex() (in module tensorplay.linalg) invalidate_unlifted() (tensorplay.export.ExportedProgram method) InvalidBackend inverse() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) inverse_shape() (tensorplay.distributions.AbsTransform method) (tensorplay.distributions.CatTransform method) (tensorplay.distributions.CumulativeDistributionTransform method) (tensorplay.distributions.ExpTransform method) (tensorplay.distributions.LowerCholeskyTransform method) (tensorplay.distributions.PositiveDefiniteTransform method) (tensorplay.distributions.SigmoidTransform method) (tensorplay.distributions.SoftplusTransform method) (tensorplay.distributions.StackTransform method) (tensorplay.distributions.TanhTransform method) (tensorplay.distributions.Transform method) InverseGamma (class in tensorplay.distributions) invert_permutation() (in module tensorplay.nn.utils.rnn) ipc_collect() (in module tensorplay.cuda) ipc_handle() (tensorplay.cuda.streams.Event method) irfft() (in module tensorplay.fft._transforms) irfft2() (in module tensorplay.fft._transforms) irfftn() (in module tensorplay.fft._transforms) is_autocast_available() (in module tensorplay) is_autocast_cache_enabled() (in module tensorplay) is_autocast_enabled() (in module tensorplay) is_available() (in module tensorplay.accelerator) (in module tensorplay.backends.mkl) (in module tensorplay.backends.nnpack) (in module tensorplay.backends.openmp) (in module tensorplay.cuda) (in module tensorplay.cuda.graph_annotations) (in module tensorplay.cuda.nccl) (in module tensorplay.distributed.autograd) (in module tensorplay.distributed.rpc) (in module tensorplay.stax) is_bf16_supported() (in module tensorplay.cuda) is_buffer() (tensorplay.export.ExportGraphSignature method) is_channels_last() (tensorplay.nn.attention.bias.CausalBias method) is_channels_last_2d() (tensorplay.nn.attention.bias.CausalBias method) is_channels_last_3d() (tensorplay.nn.attention.bias.CausalBias method) is_closed() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) is_coalesced() (in module tensorplay) is_compiling() (in module tensorplay.compiler) is_complex() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.Scalar method) is_conj() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_contiguous() (in module tensorplay) (tensorplay.masked.MaskedTensor method) is_cpu() (tensorplay.Device method) is_cuda (tensorplay.nn.utils.rnn.PackedSequence property) is_cuda() (tensorplay.Device method) is_current_stream_capturing() (in module tensorplay.cuda.graphs) is_deterministic_algorithms_warn_only_enabled() (in module tensorplay) is_distributed() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_done() (tensorplay.Future method) (tensorplay.futures.Future method) is_enabled() (in module tensorplay.cuda.tunable) (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) is_exporting() (in module tensorplay.compiler) is_failed() (tensorplay.distributed.elastic.multiprocessing.RunProcsResult method) is_floating_point() (in module tensorplay.functional) is_from_package() (in module tensorplay.package) is_gds_available() (in module tensorplay.cuda) is_grad_enabled() (in module tensorplay.autograd.grad_mode) is_heap_allocated() (tensorplay.SymInt method) is_inference() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_initialized() (in module tensorplay.cuda) (in module tensorplay.distributed.autograd) is_integer() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) (tensorplay.SymFloat method) is_lazy() (in module tensorplay.nn.parameter) is_leaf() (in module tensorplay) is_mangled() (in module tensorplay.package) is_masked_tensor() (in module tensorplay.masked) is_meta() (tensorplay.Device method) is_neg() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_nested() (tensorplay.nn.attention.bias.CausalBias method) is_nonzero() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_param() (tensorplay.export.ExportGraphSignature method) is_pinned() (in module tensorplay) (tensorplay.nn.utils.rnn.PackedSequence method) is_pruned() (in module tensorplay.ao.pruning) is_quantized() (in module tensorplay) (in module tensorplay.ao.quantization) (tensorplay.nn.attention.bias.CausalBias method) is_same() (tensorplay.SymInt method) is_same_size() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) is_set_to() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) is_shared() (tensorplay.nn.attention.bias.CausalBias method) is_signed() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) is_sparse_bsc() (tensorplay.nn.attention.bias.CausalBias method) is_sparse_bsr() (tensorplay.nn.attention.bias.CausalBias method) is_sparse_compressed() (tensorplay.nn.attention.bias.CausalBias method) is_sparse_csc() (tensorplay.nn.attention.bias.CausalBias method) is_sparse_csr() (tensorplay.nn.attention.bias.CausalBias method) is_storage() (in module tensorplay) is_supported() (in module tensorplay.onnx) is_symbolic() (tensorplay.SymBool method) (tensorplay.SymFloat method) (tensorplay.SymInt method) is_tensor() (in module tensorplay) is_tensor_like() (in module tensorplay.overrides) is_tensor_method_or_property() (in module tensorplay.overrides) is_tf32_supported() (in module tensorplay.cuda) is_verbose() (in module tensorplay.cuda.tunable) is_vulkan() (tensorplay.Device method) is_vulkan_available() (in module tensorplay.functional) isalnum() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isalpha() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isascii() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isclose() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) isdecimal() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isdigit() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isfinite() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) isidentifier() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isin() (in module tensorplay.functional) isinf() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) IsInitialized() (tensorplay.onnx.TensorProto method) islower() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isnan() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) isneginf() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) isnumeric() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isposinf() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) isprintable() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isreal() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) isspace() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) istft() (in module tensorplay.functional) istitle() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) isupper() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) item() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) items() (tensorplay.autograd.profiler_util.StringTable method) itemsize() (tensorplay.nn.attention.bias.CausalBias method) IterableDataset (class in tensorplay.utils.data) |
| --- | --- |

## J

| jacfwd() (in module tensorplay.func) jacobian() (in module tensorplay.autograd.functional) jacrev() (in module tensorplay.func) Join (class in tensorplay.distributed.algorithms) join() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) join_device (tensorplay.distributed.algorithms.Joinable property) (tensorplay.distributed.optim.ZeroRedundancyOptimizer property) join_hook() (tensorplay.distributed.algorithms.Joinable method) (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) |join_process_group (tensorplay.distributed.algorithms.Joinable property) (tensorplay.distributed.optim.ZeroRedundancyOptimizer property) Joinable (class in tensorplay.distributed.algorithms) JoinHook (class in tensorplay.distributed.algorithms) jvp() (in module tensorplay.autograd) (in module tensorplay.autograd.functional) (in module tensorplay.func) (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) |
| --- | --- |

## K

| kaiming_normal_() (in module tensorplay.nn.init) kaiming_uniform_() (in module tensorplay.nn.init) kaiser() (in module tensorplay.signal.windows) kaiser_window() (in module tensorplay.functional) Kernel (class in tensorplay.autograd.profiler_util) key_averages() (tensorplay.autograd.profiler_util.EventList method) keys() (tensorplay.autograd.profiler_util.StringTable method) kineto_results() (tensorplay.profiler.profile method) (tensorplay.profiler.profiler.profile method) |kl_div() (in module tensorplay.functional) (in module tensorplay.nn.functional) kl_divergence() (in module tensorplay.distributions) kron() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) kthvalue() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) Kumaraswamy (class in tensorplay.distributions) |
| --- | --- |

## L

| L (tensorplay.linalg.CholeskyExResult attribute) l1_loss() (in module tensorplay) (in module tensorplay.nn.functional) l1_unstructured() (in module tensorplay.ao.pruning) L1Unstructured (class in tensorplay.ao.pruning) laguerre_polynomial_l() (in module tensorplay) (in module tensorplay.functional) Laplace (class in tensorplay.distributions) launch_agent() (in module tensorplay.distributed.launcher) LaunchConfig (class in tensorplay.distributed.launcher) layer_norm() (in module tensorplay) (in module tensorplay.nn.functional) LayerNorm (class in tensorplay.ao.nn.quantized) Layout (class in tensorplay) LazyModuleMixin (class in tensorplay.nn.modules.lazy) lcm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) lcm_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ldexp() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ldexp_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ldl_factor() (in module tensorplay.linalg) ldl_factor_ex() (in module tensorplay.linalg) ldl_solve() (in module tensorplay.linalg) le() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) le_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) leaky_relu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) leaky_relu_() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) LeakyReLU (class in tensorplay.ao.nn.quantized) legendre_polynomial_p() (in module tensorplay) (in module tensorplay.functional) lerp() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) lerp_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) less() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) less_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) less_equal() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) less_equal_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) lgamma() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) lgamma_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Library (class in tensorplay.library) lift() (in module tensorplay) lift_fresh() (in module tensorplay) lift_fresh_copy() (in module tensorplay) LinAlgError Linear (class in tensorplay.ao.nn.qat) (class in tensorplay.ao.nn.quantized.dynamic) (in module tensorplay.ao.nn.quantized) linear() (in module tensorplay) (in module tensorplay.nn.functional) linear_cross_entropy() (in module tensorplay.nn.functional) linearize() (in module tensorplay.func) LinearReLU (class in tensorplay.ao.nn.intrinsic) linspace() (in module tensorplay.distributed.tensor) (in module tensorplay.functional) list_backends() (in module tensorplay.compiler) list_flash_attention_impls() (in module tensorplay.nn.attention) list_gpu_processes() (in module tensorplay.cuda) list_mode_options() (in module tensorplay.compiler) ListFields() (tensorplay.onnx.TensorProto method) ljust() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) LKJCholesky (class in tensorplay.distributions) ln_structured() (in module tensorplay.ao.pruning) LnStructured (class in tensorplay.ao.pruning) load() (in module tensorplay.distributed.checkpoint) (in module tensorplay.export) (in module tensorplay.serialization) (in module tensorplay.serialization.api) load_binary() (tensorplay.package.PackageImporter method) load_observer_state_dict() (in module tensorplay.ao.quantization) load_pickle() (tensorplay.package.PackageImporter method) load_pt2() (in module tensorplay.export) load_sharded_optimizer_state_dict() (in module tensorplay.distributed.checkpoint) load_state_dict() (in module tensorplay.distributed.checkpoint) (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.distributed.optim.PostLocalSGDOptimizer method) (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.GradScaler method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |load_text() (tensorplay.package.PackageImporter method) LoadEndianness (class in tensorplay.serialization) LoadItemType (class in tensorplay.distributed.checkpoint) LoadPlan (class in tensorplay.distributed.checkpoint) LoadPlanner (class in tensorplay.distributed.checkpoint) local_response_norm() (in module tensorplay.nn.functional) LocalElasticAgent (class in tensorplay.distributed.elastic.agent.server) LocalOptimStateDictConfig (class in tensorplay.distributed.fsdp) LocalStateDictConfig (class in tensorplay.distributed.fsdp) location_tag() (in module tensorplay.serialization) log() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log10() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log10_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log1p() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log1p_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log2() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) log2_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) log_abs_det_jacobian() (tensorplay.distributions.AbsTransform method) (tensorplay.distributions.LowerCholeskyTransform method) (tensorplay.distributions.PositiveDefiniteTransform method) (tensorplay.distributions.SoftmaxTransform method) (tensorplay.distributions.Transform method) log_event() (in module tensorplay.monitor) log_ndtr() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) log_normal_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) log_prob() (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.Weibull method) log_sigmoid() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) log_sigmoid_forward() (in module tensorplay) log_softmax() (in module tensorplay) (in module tensorplay.masked) (in module tensorplay.nn.functional) (in module tensorplay.special._elementwise) (tensorplay.nn.attention.bias.CausalBias method) logabsdet (tensorplay.linalg.SlogdetResult attribute) logaddexp() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) logaddexp2() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) logcumsumexp() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logdet() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logical_and() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logical_and_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) logical_not() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logical_not_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) logical_or() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logical_or_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) logical_xor() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) logical_xor_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) LogisticNormal (class in tensorplay.distributions) logit() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) logit_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) logit_backward() (tensorplay.nn.attention.bias.CausalBias method) LogNormal (class in tensorplay.distributions) logsigmoid() (in module tensorplay.nn.functional) logspace() (in module tensorplay.distributed.tensor) (in module tensorplay.functional) logsumexp() (in module tensorplay.functional) (in module tensorplay.masked) (in module tensorplay.special._elementwise) (tensorplay.nn.attention.bias.CausalBias method) lookup_backend() (in module tensorplay.compiler) lookup_function_handler() (in module tensorplay.onnx) lookup_method_handler() (in module tensorplay.onnx) loss_parallel() (in module tensorplay.distributed.tensor.parallel) lower() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) LowerCholeskyTransform (class in tensorplay.distributions) LowRankMultivariateNormal (class in tensorplay.distributions) lp_pool1d() (in module tensorplay.nn.functional) lp_pool2d() (in module tensorplay.nn.functional) lp_pool3d() (in module tensorplay.nn.functional) lstm() (in module tensorplay.functional) lstm_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) lstrip() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) lstsq() (in module tensorplay.linalg) LstsqResult (class in tensorplay.linalg) lt() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) lt_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) lu() (in module tensorplay.linalg) lu_factor() (in module tensorplay.linalg) lu_factor_ex() (in module tensorplay.linalg) lu_solve() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) lu_unpack() (in module tensorplay.functional) |
| --- | --- |

## M

| macros (class in tensorplay.distributed.elastic.utils) main_hook() (tensorplay.distributed.algorithms.JoinHook method) make_dot() (in module tensorplay.utils.viz) make_dual() (in module tensorplay.autograd.forward_ad) make_graphed_callables() (in module tensorplay.cuda.graphs) make_tensor() (in module tensorplay.testing) maketrans() (tensorplay.distributed.elastic.agent.server.WorkerState static method) (tensorplay.distributed.elastic.events.NodeState static method) (tensorplay.profiler.ProfilerAction static method) (tensorplay.profiler.ProfilerActivity static method) manual_seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) manual_seed_all() (in module tensorplay.cuda.random) map2_() (tensorplay.nn.attention.bias.CausalBias method) map_() (tensorplay.nn.attention.bias.CausalBias method) margin_ranking_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) mark() (in module tensorplay.cuda.nvtx) mark_kernels() (in module tensorplay.cuda.graph_annotations) mark_static() (in module tensorplay.compiler) masked_fill() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) masked_fill_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) masked_scatter() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) masked_scatter_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) masked_select() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) masked_tensor() (in module tensorplay.masked) MaskedTensor (class in tensorplay.masked) materialize() (tensorplay.export.CustomDecompTable method) matmul() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) matmul_backward_other() (in module tensorplay) matmul_backward_self() (in module tensorplay) matrix_exp() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) matrix_H() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) matrix_norm() (in module tensorplay.linalg) matrix_power() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) matrix_rank() (in module tensorplay.linalg) matrix_sqrth() (in module tensorplay.linalg) max() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) max_memory_allocated() (in module tensorplay.cuda) max_memory_reserved() (in module tensorplay.cuda) max_pool1d() (in module tensorplay.functional) (in module tensorplay.nn.functional) max_pool1d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_pool2d() (in module tensorplay) (in module tensorplay.nn.functional) max_pool2d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_pool3d() (in module tensorplay.functional) (in module tensorplay.nn.functional) max_pool3d_with_indices() (in module tensorplay) (in module tensorplay.nn.functional) max_unpool1d() (in module tensorplay.nn.functional) max_unpool2d() (in module tensorplay) (in module tensorplay.nn.functional) max_unpool3d() (in module tensorplay) (in module tensorplay.nn.functional) maximum() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) MaxPool1d (class in tensorplay.ao.nn.quantized) MaxPool2d (class in tensorplay.ao.nn.quantized) MaxPool3d (class in tensorplay.ao.nn.quantized) maybe_as_bool() (tensorplay.SymBool method) maybe_as_float() (tensorplay.SymFloat method) maybe_as_int() (tensorplay.SymInt method) maybe_increase_iter() (tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook.PostLocalSGDState method) (tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook.PowerSGDState method) mean (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.TransformedDistribution property) (tensorplay.distributions.VonMises property) mean() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) median() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.nn.attention.bias.CausalBias method) MegaStorageReader (class in tensorplay.distributed.checkpoint) MegaStorageWriter (class in tensorplay.distributed.checkpoint) mem_get_info() (in module tensorplay.cuda) memory_allocated() (in module tensorplay.cuda) memory_format() (tensorplay.nn.attention.bias.CausalBias method) memory_reserved() (in module tensorplay.cuda) memory_snapshot() (in module tensorplay.cuda) memory_stats() (in module tensorplay.cuda) memory_stats_as_nested_dict() (in module tensorplay.cuda) memory_summary() (in module tensorplay.cuda) memory_usage() (in module tensorplay.cuda) MemoryFormat (class in tensorplay) MemPool (class in tensorplay.cuda) MemRecordsAcc (class in tensorplay.autograd.profiler_util) merge_distributed_traces() (in module tensorplay.profiler) MergeFrom() (tensorplay.onnx.TensorProto method) MergeFromString() (tensorplay.onnx.TensorProto method) meshgrid() (in module tensorplay.functional) Metadata (class in tensorplay.distributed.checkpoint) MetadataIndex (class in tensorplay.distributed.checkpoint) method_factory() (in module tensorplay.distributed.rpc) MetricsConfig (class in tensorplay.distributed.elastic.metrics) mH() (in module tensorplay) min() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) minimum() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) MinMaxObserver (class in tensorplay.ao.quantization) miopen_batch_norm() (in module tensorplay.functional) miopen_convolution() (in module tensorplay.functional) miopen_convolution_add_relu() (in module tensorplay.functional) miopen_convolution_relu() (in module tensorplay.functional) miopen_convolution_transpose() (in module tensorplay.functional) miopen_ctc_loss() (in module tensorplay.functional) miopen_depthwise_convolution() (in module tensorplay.functional) miopen_rnn() (in module tensorplay.functional) mish() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) mish_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) MixedPrecision (class in tensorplay.distributed.fsdp) MixedPrecisionPolicy (class in tensorplay.distributed.fsdp) MixtureSameFamily (class in tensorplay.distributions) mkldnn_adaptive_avg_pool2d() (in module tensorplay.functional) mkldnn_convolution() (in module tensorplay.functional) mkldnn_linear() (in module tensorplay) mkldnn_linear_backward_input() (in module tensorplay) mkldnn_linear_backward_weights() (in module tensorplay.functional) mkldnn_max_pool2d() (in module tensorplay.functional) mkldnn_max_pool3d() (in module tensorplay.functional) mkldnn_reorder_conv2d_weight() (in module tensorplay) mkldnn_reorder_conv3d_weight() (in module tensorplay) mkldnn_rnn_layer() (in module tensorplay.functional) mm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) mock() (tensorplay.package.PackageExporter method) mocked_modules() (tensorplay.package.PackageExporter method) mode (tensorplay.distributions.ContinuousBernoulli property) (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.MixtureSameFamily property) (tensorplay.distributions.Multinomial property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.TransformedDistribution property) mode() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ModelAverager (class in tensorplay.distributed.algorithms.model_averaging) modified_bessel_i0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) modified_bessel_i1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) modified_bessel_k0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) modified_bessel_k1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) module tensorplay.accelerator.graphs tensorplay.accelerator.memory tensorplay.accelerator.random tensorplay.amp tensorplay.amp.autocast_mode tensorplay.amp.grad_scaler tensorplay.backends tensorplay.cuda.tunable tensorplay.distributed.algorithms tensorplay.distributed.algorithms.ddp_comm_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook tensorplay.distributed.algorithms.ddp_comm_hooks.debugging_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks tensorplay.distributed.algorithms.join tensorplay.distributed.algorithms.model_averaging tensorplay.distributed.algorithms.model_averaging.averagers tensorplay.distributed.algorithms.model_averaging.hierarchical_model_averager tensorplay.distributed.algorithms.model_averaging.utils tensorplay.distributed.argparse_util tensorplay.distributed.checkpoint.api tensorplay.distributed.checkpoint.default_planner tensorplay.distributed.checkpoint.filesystem tensorplay.distributed.checkpoint.hf_storage tensorplay.distributed.checkpoint.metadata tensorplay.distributed.checkpoint.optimizer tensorplay.distributed.checkpoint.planner tensorplay.distributed.checkpoint.planner_helpers tensorplay.distributed.checkpoint.quantized_hf_storage tensorplay.distributed.checkpoint.resharding tensorplay.distributed.checkpoint.state_dict tensorplay.distributed.checkpoint.state_dict_loader tensorplay.distributed.checkpoint.state_dict_saver tensorplay.distributed.checkpoint.stateful tensorplay.distributed.checkpoint.storage tensorplay.distributed.checkpoint.utils tensorplay.distributed.constants tensorplay.distributed.device_mesh tensorplay.distributed.distributed_core tensorplay.distributed.elastic tensorplay.distributed.elastic.agent.server.api tensorplay.distributed.elastic.agent.server.local_elastic_agent tensorplay.distributed.elastic.events.api tensorplay.distributed.elastic.events.handlers tensorplay.distributed.elastic.metrics.api tensorplay.distributed.elastic.multiprocessing.api tensorplay.distributed.elastic.multiprocessing.errors.error_handler tensorplay.distributed.elastic.multiprocessing.errors.handlers tensorplay.distributed.elastic.multiprocessing.redirects tensorplay.distributed.elastic.multiprocessing.tail_log tensorplay.distributed.elastic.rendezvous.api tensorplay.distributed.elastic.rendezvous.c10d_rendezvous_backend tensorplay.distributed.elastic.rendezvous.dynamic_rendezvous tensorplay.distributed.elastic.rendezvous.etcd_rendezvous tensorplay.distributed.elastic.rendezvous.etcd_rendezvous_backend tensorplay.distributed.elastic.rendezvous.etcd_server tensorplay.distributed.elastic.rendezvous.etcd_store tensorplay.distributed.elastic.rendezvous.static_tcp_rendezvous tensorplay.distributed.elastic.rendezvous.utils tensorplay.distributed.elastic.timer.api tensorplay.distributed.elastic.timer.file_based_local_timer tensorplay.distributed.elastic.timer.local_timer tensorplay.distributed.elastic.utils tensorplay.distributed.elastic.utils.api tensorplay.distributed.elastic.utils.data tensorplay.distributed.elastic.utils.data.cycling_iterator tensorplay.distributed.elastic.utils.data.elastic_distributed_sampler tensorplay.distributed.elastic.utils.distributed tensorplay.distributed.elastic.utils.log_level tensorplay.distributed.elastic.utils.logging tensorplay.distributed.elastic.utils.store tensorplay.distributed.fsdp.api tensorplay.distributed.fsdp.fully_sharded_data_parallel tensorplay.distributed.fsdp.sharded_grad_scaler tensorplay.distributed.fsdp.wrap tensorplay.distributed.launcher tensorplay.distributed.logging_handlers tensorplay.distributed.nn tensorplay.distributed.nn.api tensorplay.distributed.nn.api.remote_module tensorplay.distributed.nn.functional tensorplay.distributed.nn.jit tensorplay.distributed.nn.jit.instantiator tensorplay.distributed.nn.jit.templates tensorplay.distributed.nn.jit.templates.remote_module_template tensorplay.distributed.optim.apply_optimizer_in_backward tensorplay.distributed.optim.functional_adadelta tensorplay.distributed.optim.functional_adagrad tensorplay.distributed.optim.functional_adam tensorplay.distributed.optim.functional_adamax tensorplay.distributed.optim.functional_adamw tensorplay.distributed.optim.functional_rmsprop tensorplay.distributed.optim.functional_rprop tensorplay.distributed.optim.functional_sgd tensorplay.distributed.optim.named_optimizer tensorplay.distributed.optim.optimizer tensorplay.distributed.optim.post_localSGD_optimizer tensorplay.distributed.optim.utils tensorplay.distributed.optim.zero_redundancy_optimizer tensorplay.distributed.p10d_logger tensorplay.distributed.remote_device tensorplay.distributed.rendezvous tensorplay.distributed.rpc.api tensorplay.distributed.rpc.backend_registry tensorplay.distributed.rpc.constants tensorplay.distributed.rpc.functions tensorplay.distributed.rpc.internal tensorplay.distributed.rpc.options tensorplay.distributed.rpc.rref_proxy tensorplay.distributed.rpc.server_process_global_profiler tensorplay.distributed.tensor.parallel.api tensorplay.distributed.tensor.parallel.ddp tensorplay.distributed.tensor.parallel.fsdp tensorplay.distributed.tensor.parallel.input_reshard tensorplay.distributed.tensor.parallel.loss tensorplay.distributed.tensor.parallel.style tensorplay.distributed.utils tensorplay.func tensorplay.multiprocessing.pool tensorplay.multiprocessing.queue tensorplay.multiprocessing.reductions tensorplay.nn.attention tensorplay.nn.attention.bias tensorplay.nn.attention.omni_attention tensorplay.nn.attention.varlen tensorplay.nn.backends tensorplay.nn.backends.thnn tensorplay.nn.common_types tensorplay.nn.cpp tensorplay.nn.functional tensorplay.nn.grad tensorplay.nn.init tensorplay.nn.modules.activation tensorplay.nn.modules.adaptive tensorplay.nn.modules.batchnorm tensorplay.nn.modules.channelshuffle tensorplay.nn.modules.container tensorplay.nn.modules.conv tensorplay.nn.modules.distance tensorplay.nn.modules.dropout tensorplay.nn.modules.flatten tensorplay.nn.modules.folding tensorplay.nn.modules.instancenorm tensorplay.nn.modules.lazy tensorplay.nn.modules.linear tensorplay.nn.modules.loss tensorplay.nn.modules.module tensorplay.nn.modules.normalization tensorplay.nn.modules.padding tensorplay.nn.modules.pixelshuffle tensorplay.nn.modules.pooling tensorplay.nn.modules.rnn tensorplay.nn.modules.sparse tensorplay.nn.modules.transformer tensorplay.nn.modules.upsampling tensorplay.nn.modules.utils tensorplay.nn.parallel.comm tensorplay.nn.parallel.distributed tensorplay.nn.parallel.parallel_apply tensorplay.nn.parallel.replicate tensorplay.nn.parallel.scatter_gather tensorplay.nn.parameter tensorplay.nn.utils.clip_grad tensorplay.nn.utils.convert_parameters tensorplay.nn.utils.fusion tensorplay.nn.utils.init tensorplay.nn.utils.memory_format tensorplay.nn.utils.parametrizations tensorplay.nn.utils.parametrize tensorplay.nn.utils.prune tensorplay.nn.utils.rnn tensorplay.nn.utils.stateless tensorplay.optim.lr_scheduler tensorplay.optim.optimizer tensorplay.optim.swa_utils tensorplay.utils.data.datapipes tensorplay.utils.data.datapipes.dataframe tensorplay.utils.data.datapipes.iter tensorplay.utils.data.datapipes.map tensorplay.utils.data.datapipes.utils tensorplay.utils.serialization tensorplay.utils.serialization.config |module() (tensorplay.export.ExportedProgram method) module_load() (tensorplay.nn.attention.bias.CausalBias method) ModuleCallEntry (class in tensorplay.export) ModuleCallSignature (class in tensorplay.export) modules() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) ModuleType (in module tensorplay.package) moveaxis() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) movedim() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) MovingAverageMinMaxObserver (class in tensorplay.ao.quantization) MovingAveragePerChannelMinMaxObserver (class in tensorplay.ao.quantization) mro() (tensorplay.autograd.function.FunctionMeta method) (tensorplay.autograd.variable.VariableMeta method) mse_loss() (in module tensorplay) (in module tensorplay.nn.functional) msort() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) mT() (in module tensorplay) mul() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) mul_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) multi_dot() (in module tensorplay.linalg) multi_head_attention_forward() (in module tensorplay.nn.functional) multi_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) multigammaln() (in module tensorplay.special._gamma) multilabel_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) multilabel_margin_loss_forward() (in module tensorplay) multilabel_soft_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) Multinomial (class in tensorplay.distributions) multinomial() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) multiply() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) multiply_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) MultivariateNormal (class in tensorplay.distributions) mv() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) mvlgamma() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) mvlgamma_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) |
| --- | --- |

## N

| name (tensorplay.autograd.profiler_util.Kernel attribute) name() (tensorplay.Node method) (tensorplay.onnx.OpContext method) named_buffers() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_children() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_modules() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) named_parameters() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |NamedShape (tensorplay.nn.modules.flatten.Unflatten attribute) nan_to_num() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) nan_to_num_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) nanmean() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) nanmedian() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) nanquantile() (in module tensorplay._composite_funcs) (tensorplay.nn.attention.bias.CausalBias method) nansum() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) narrow() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) narrow_copy() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) native_alpha_dropout() (in module tensorplay) native_batch_norm() (in module tensorplay.functional) native_channel_shuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) native_dropout() (in module tensorplay) native_feature_dropout() (in module tensorplay) native_group_norm() (in module tensorplay.functional) native_layer_norm() (in module tensorplay.functional) native_norm() (in module tensorplay.functional) nbytes() (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.UntypedStorage method) ndimension() (tensorplay.nn.attention.bias.CausalBias method) ndtr() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ndtri() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ne() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) ne_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) neg() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) neg_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) negative() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) negative_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) NegativeBinomial (class in tensorplay.distributions) nelement() (tensorplay.nn.attention.bias.CausalBias method) nested_compile_region() (in module tensorplay.compiler) nested_tensor() (in module tensorplay.nested) nested_to_padded_tensor() (in module tensorplay) (in module tensorplay.nested) NestedCompileRegionOptions (class in tensorplay.compiler) NestedIOFunction (class in tensorplay.autograd.function) new() (tensorplay.nn.attention.bias.CausalBias method) new_empty() (in module tensorplay) new_empty_strided() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) new_full() (in module tensorplay) new_method() (in module tensorplay.distributed.rpc) new_ones() (in module tensorplay) new_zeros() (in module tensorplay) next_rendezvous() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) nextafter() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) nextafter_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) nll_loss() (in module tensorplay) (in module tensorplay.nn.functional) nll_loss2d() (in module tensorplay) nll_loss2d_forward() (in module tensorplay) nll_loss_forward() (in module tensorplay) nll_loss_nd() (in module tensorplay) no_grad (class in tensorplay.autograd.grad_mode) Node (class in tensorplay) NodeState (class in tensorplay.distributed.elastic.events) nonstrict_trace() (in module tensorplay.compiler) nonzero() (in module tensorplay.functional) nonzero_numpy() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) nonzero_static() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) noop_mask() (in module tensorplay.nn.attention) norm() (in module tensorplay.functional) (in module tensorplay.linalg) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) norm_except_dim() (in module tensorplay.functional) Normal (class in tensorplay.distributions) normal() (in module tensorplay.functional) normal_() (in module tensorplay) (in module tensorplay.nn.init) (tensorplay.nn.attention.bias.CausalBias method) normal_functional() (in module tensorplay) normalize() (in module tensorplay.masked) (in module tensorplay.nn.functional) normalize_storage_type() (in module tensorplay.serialization) not_equal() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) not_equal_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) notify_join_context() (tensorplay.distributed.algorithms.Join static method) nuclear_norm() (in module tensorplay.functional) num_nodes_waiting() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) numel() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.attention.BlockMask method) (tensorplay.Size method) numerator (tensorplay.distributed.elastic.multiprocessing.Std attribute) (tensorplay.export.WeightType attribute) (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.nn.attention.bias.CausalVariant attribute) (tensorplay.QScheme attribute) numpy() (tensorplay.nn.attention.bias.CausalBias method) numpy_T() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) nuttall() (in module tensorplay.signal.windows) |
| --- | --- |

## O

| ObjMismatchError ObjNotFoundError OffloadPolicy (class in tensorplay.distributed.fsdp) once_differentiable() (in module tensorplay.autograd.function) one_hot() (in module tensorplay) (in module tensorplay.nn.functional) OneHotCategorical (class in tensorplay.distributions) OneHotCategoricalStraightThrough (class in tensorplay.distributions) ones() (in module tensorplay) (in module tensorplay.distributed.tensor) ones_() (in module tensorplay.nn.init) ones_like() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) OnnxExporterError OnnxExporterWarning op() (tensorplay.onnx.GraphBuilder method) opcheck() (in module tensorplay.library) |OpContext (class in tensorplay.onnx) OptimStateDictConfig (class in tensorplay.distributed.fsdp) OptimStateKeyType (class in tensorplay.distributed.fsdp) or_masks() (in module tensorplay.nn.attention) OrderedImporter (class in tensorplay.package) orgqr() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) ormqr() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) orthogonal_() (in module tensorplay.nn.init) outer() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) OutOfMemoryError output_nr() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) OutputKind (class in tensorplay.export) OutputSpec (class in tensorplay.export) overload_method() (in module tensorplay.compiler) |
| --- | --- |

## P

| P10dRendezvousBackend (class in tensorplay.distributed.elastic.rendezvous) pack_padded_sequence() (in module tensorplay.nn.utils.rnn) pack_sequence() (in module tensorplay.nn.utils.rnn) package_pt2() (in module tensorplay.export) PackageExporter (class in tensorplay.package) PackageImporter (class in tensorplay.package) PackagingError PackagingErrorReason (class in tensorplay.package) PackedSequence (class in tensorplay.nn.utils.rnn) pad() (in module tensorplay) (in module tensorplay.nn.functional) pad_packed_sequence() (in module tensorplay.nn.utils.rnn) pad_sequence() (in module tensorplay) (in module tensorplay.nn.utils.rnn) pairwise_distance() (in module tensorplay.functional) (in module tensorplay.nn.functional) parallelize_module() (in module tensorplay.distributed.tensor.parallel) ParallelStyle (class in tensorplay.distributed.tensor.parallel) parameters() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.export.ExportedProgram method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) Pareto (class in tensorplay.distributions) parse_mega_header() (in module tensorplay.serialization) ParseFromString() (tensorplay.onnx.TensorProto method) Partial (class in tensorplay.distributed.tensor) participants (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) partition() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) PContext (class in tensorplay.distributed.elastic.multiprocessing) pdist() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) PerChannelFakeQuantize (class in tensorplay.ao.quantization) PerChannelMinMaxObserver (class in tensorplay.ao.quantization) PeriodicModelAverager (class in tensorplay.distributed.algorithms.model_averaging) permute() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) permute_copy() (in module tensorplay.functional) perplexity() (tensorplay.distributions.Bernoulli method) (tensorplay.distributions.Beta method) (tensorplay.distributions.Binomial method) (tensorplay.distributions.Categorical method) (tensorplay.distributions.Cauchy method) (tensorplay.distributions.Chi2 method) (tensorplay.distributions.ContinuousBernoulli method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.Exponential method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.Gamma method) (tensorplay.distributions.GeneralizedPareto method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.HalfCauchy method) (tensorplay.distributions.HalfNormal method) (tensorplay.distributions.Independent method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.Laplace method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.Normal method) (tensorplay.distributions.OneHotCategorical method) (tensorplay.distributions.OneHotCategoricalStraightThrough method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.Uniform method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) |pin_memory() (in module tensorplay) pinv() (in module tensorplay.linalg) pinverse() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) Pipe (class in tensorplay.distributed.pipelining) pipe_split() (in module tensorplay.distributed.pipelining) pipeline() (in module tensorplay.distributed.pipelining) PipelineScheduleMulti (class in tensorplay.distributed.pipelining) PipelineScheduleSingle (class in tensorplay.distributed.pipelining) PipelineStage (class in tensorplay.distributed.pipelining) pixel_shuffle() (in module tensorplay) (in module tensorplay.nn.functional) pixel_unshuffle() (in module tensorplay.functional) (in module tensorplay.nn.functional) PlaceholderObserver (class in tensorplay.ao.quantization) Placement (class in tensorplay.distributed.tensor) Poisson (class in tensorplay.distributions) poisson() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) poisson_nll_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) polar() (in module tensorplay.functional) (in module tensorplay.linalg) poll() (tensorplay.distributed.elastic.multiprocessing.PContext method) polygamma() (in module tensorplay.functional) (in module tensorplay.special._gamma) (tensorplay.nn.attention.bias.CausalBias method) polygamma_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) pool() (tensorplay.accelerator.Graph method) pool_id (tensorplay.accelerator.Graph property) (tensorplay.cuda.graphs.CUDAGraph property) pop() (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.profiler.EventList method) popitem() (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.export.CustomDecompTable method) positive() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) PositiveDefiniteTransform (class in tensorplay.distributions) post_hook() (tensorplay.distributed.algorithms.JoinHook method) post_localSGD_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook) PostLocalSGDOptimizer (class in tensorplay.distributed.optim) PostLocalSGDState (class in tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook) pow() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) pow_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) pow_by_natural() (tensorplay.SymInt method) power_draw() (in module tensorplay.cuda) powerSGD_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerSGDState (class in tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook) PowerTransform (class in tensorplay.distributions) pre_shutdown() (tensorplay.distributed.elastic.rendezvous.DynamicRendezvousHandler method) (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) (tensorplay.distributed.elastic.rendezvous.StaticTCPRendezvous method) prelu() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) prepare() (in module tensorplay.ao.quantization) prepare_qat() (in module tensorplay.ao.quantization) PrepareModuleInput (class in tensorplay.distributed.tensor.parallel) PrepareModuleInputOutput (class in tensorplay.distributed.tensor.parallel) PrepareModuleOutput (class in tensorplay.distributed.tensor.parallel) ProcessFailure (class in tensorplay.distributed.elastic.multiprocessing) prod() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) prof() (in module tensorplay.distributed.elastic.metrics) profile (class in tensorplay.profiler) (class in tensorplay.profiler.profiler) profile() (in module tensorplay.cuda.profiler) ProfilerAction (class in tensorplay.profiler) ProfilerActivity (class in tensorplay.profiler) promote_types() (in module tensorplay.functional) prune() (tensorplay.ao.pruning.BasePruningMethod method) (tensorplay.ao.pruning.CustomFromMask method) (tensorplay.ao.pruning.Identity method) (tensorplay.ao.pruning.L1Unstructured method) (tensorplay.ao.pruning.LnStructured method) (tensorplay.ao.pruning.PruningContainer method) (tensorplay.ao.pruning.RandomStructured method) (tensorplay.ao.pruning.RandomUnstructured method) PruningContainer (class in tensorplay.ao.pruning) psi() (in module tensorplay.special._gamma) PT2ArchiveContents (class in tensorplay.export) put() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) put_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) python_version() (tensorplay.package.PackageImporter method) |
| --- | --- |

## Q

| Q (tensorplay.linalg.QRResult attribute) q_per_channel_axis() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) q_per_channel_scales() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) q_per_channel_zero_points() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) q_scale() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) q_zero_point() (in module tensorplay.ao.quantization) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) QConfig (class in tensorplay.ao.quantization) QConfigMapping (class in tensorplay.ao.quantization) QFunctional (class in tensorplay.ao.nn.quantized) qr() (in module tensorplay.linalg) QRResult (class in tensorplay.linalg) QScheme (class in tensorplay) qscheme() (in module tensorplay) (in module tensorplay.ao.quantization) (tensorplay.nn.attention.bias.CausalBias method) quantile() (in module tensorplay._composite_funcs) (tensorplay.nn.attention.bias.CausalBias method) quantization_perchannel_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks) quantization_pertensor_hook() (in module tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks) Quantize (class in tensorplay.ao.nn.quantized) quantize() (in module tensorplay.ao.quantization) quantize_dynamic() (in module tensorplay.ao.quantization) |quantize_per_channel() (in module tensorplay.ao.quantization) (in module tensorplay.functional) quantize_per_tensor() (in module tensorplay.ao.quantization) (in module tensorplay.functional) quantize_per_tensor_dynamic() (in module tensorplay.ao.quantization) (in module tensorplay.functional) quantize_qat() (in module tensorplay.ao.quantization) quantized_add() (in module tensorplay) quantized_batch_norm() (in module tensorplay.functional) quantized_clamp() (in module tensorplay) quantized_conv2d() (in module tensorplay) quantized_conv2d_prepack() (in module tensorplay) quantized_conv2d_run() (in module tensorplay) quantized_conv2d_unpack() (in module tensorplay) quantized_div() (in module tensorplay) quantized_gru_cell() (in module tensorplay.functional) quantized_linear() (in module tensorplay) (in module tensorplay.ao.quantization) quantized_linear_dynamic() (in module tensorplay.ao.quantization) quantized_lstm_cell() (in module tensorplay.functional) quantized_max_pool1d() (in module tensorplay.functional) quantized_max_pool2d() (in module tensorplay.functional) quantized_max_pool3d() (in module tensorplay.functional) quantized_mul() (in module tensorplay) quantized_rnn_relu_cell() (in module tensorplay.functional) quantized_rnn_tanh_cell() (in module tensorplay.functional) quantized_sub() (in module tensorplay) QuantizedHuggingFaceStorageReader (class in tensorplay.distributed.checkpoint) QuantizedLinear (class in tensorplay.ao.nn.quantized) QuantStub (class in tensorplay.ao.quantization) query() (tensorplay.cuda.streams.Event method) (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) |
| --- | --- |

## R

| R (tensorplay.linalg.QRResult attribute) rad2deg() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) rad2deg_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) rand() (in module tensorplay) (in module tensorplay.distributed.tensor) rand_like() (in module tensorplay.functional) randint() (in module tensorplay.functional) randint_like() (in module tensorplay.functional) randn() (in module tensorplay) (in module tensorplay.distributed.tensor) randn_like() (in module tensorplay.functional) random_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) random_structured() (in module tensorplay.ao.pruning) random_unstructured() (in module tensorplay.ao.pruning) RandomSampler (class in tensorplay.utils.data) RandomStructured (class in tensorplay.ao.pruning) RandomUnstructured (class in tensorplay.ao.pruning) randperm() (in module tensorplay.functional) range() (in module tensorplay.cuda.nvtx) (in module tensorplay.functional) range_end() (in module tensorplay.cuda.nvtx) range_pop() (in module tensorplay.cuda.nvtx) range_push() (in module tensorplay.cuda.nvtx) range_start() (in module tensorplay.cuda.nvtx) rank (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) (tensorplay.linalg.LstsqResult attribute) ravel() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) RdzvEvent (class in tensorplay.distributed.elastic.events) read_file() (in module tensorplay.cuda.tunable) ReadItem (class in tensorplay.distributed.checkpoint) real (tensorplay.distributed.elastic.multiprocessing.Std attribute) (tensorplay.export.WeightType attribute) (tensorplay.Layout attribute) (tensorplay.MemoryFormat attribute) (tensorplay.nn.attention.bias.CausalVariant attribute) (tensorplay.QScheme attribute) real() (in module tensorplay.functional) rearrange() (in module tensorplay.func) reciprocal() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) reciprocal_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) record() (in module tensorplay.distributed.elastic.events) (tensorplay.ao.quantization.QuantStub method) (tensorplay.cuda.streams.Event method) record_event() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) record_function (class in tensorplay.profiler) (class in tensorplay.profiler.profiler) record_untuned_disable() (in module tensorplay.cuda.tunable) record_untuned_enable() (in module tensorplay.cuda.tunable) record_untuned_is_enabled() (in module tensorplay.cuda.tunable) Redirects (class in tensorplay.distributed.elastic.multiprocessing) redispatch_function() (in module tensorplay.overrides) reduce() (in module tensorplay.distributed.nn.functional) reduce_op (built-in class) reduce_scatter() (in module tensorplay.distributed.nn.functional) refine_dynamic_shapes_from_suggested_fixes() (in module tensorplay.export) reflection_pad1d() (in module tensorplay) reflection_pad2d() (in module tensorplay) reflection_pad3d() (in module tensorplay) reflection_pad_nd() (in module tensorplay) register_autocast() (in module tensorplay.library) register_autograd() (in module tensorplay.library) register_backend() (in module tensorplay.compiler) (in module tensorplay.distributed.rpc) register_buffer() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_dataclass() (in module tensorplay.export) register_debug_backend() (in module tensorplay.compiler) register_event_handler() (in module tensorplay.monitor) register_experimental_backend() (in module tensorplay.compiler) register_extern_hook() (tensorplay.package.PackageExporter method) register_fake() (in module tensorplay.library) register_flash_attention_impl() (in module tensorplay.nn.attention) register_forward_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_forward_pre_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_fsdp_forward_method() (in module tensorplay.distributed.fsdp) register_full_backward_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_full_backward_pre_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_hook() (tensorplay.nn.attention.bias.CausalBias method) register_intern_hook() (tensorplay.package.PackageExporter method) register_kernel() (in module tensorplay.library) register_kl() (in module tensorplay.distributions) register_load_state_dict_post_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_load_state_dict_pre_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_mock_hook() (tensorplay.package.PackageExporter method) register_module() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |register_module_buffer_registration_hook() (in module tensorplay.nn.modules.module) register_module_forward_hook() (in module tensorplay.nn.modules.module) register_module_forward_pre_hook() (in module tensorplay.nn.modules.module) register_module_full_backward_hook() (in module tensorplay.nn.modules.module) register_module_full_backward_pre_hook() (in module tensorplay.nn.modules.module) register_module_module_registration_hook() (in module tensorplay.nn.modules.module) register_module_parameter_registration_hook() (in module tensorplay.nn.modules.module) register_package() (in module tensorplay.serialization) register_parameter() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_post_accumulate_grad_hook() (tensorplay.nn.attention.bias.CausalBias method) register_state_dict_post_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_state_dict_pre_hook() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) register_vmap() (in module tensorplay.library) RelaxedBernoulli (class in tensorplay.distributions) RelaxedOneHotCategorical (class in tensorplay.distributions) ReLU (class in tensorplay.ao.nn.quantized) relu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) ReLU6 (class in tensorplay.ao.nn.quantized) relu6() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) relu6_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) relu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) remainder() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) remainder_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) remote() (in module tensorplay.distributed.rpc) RemoteModule (class in tensorplay.distributed.nn) remove() (in module tensorplay.ao.pruning) (tensorplay.ao.pruning.BasePruningMethod method) (tensorplay.ao.pruning.CustomFromMask method) (tensorplay.ao.pruning.Identity method) (tensorplay.ao.pruning.L1Unstructured method) (tensorplay.ao.pruning.LnStructured method) (tensorplay.ao.pruning.PruningContainer method) (tensorplay.ao.pruning.RandomStructured method) (tensorplay.ao.pruning.RandomUnstructured method) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.export.CustomDecompTable method) (tensorplay.profiler.EventList method) removeprefix() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) removesuffix() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) RendezvousClosedError RendezvousConnectionError RendezvousError RendezvousExhaustedError RendezvousGracefulExitError RendezvousHandler (class in tensorplay.distributed.elastic.rendezvous) RendezvousInfo (class in tensorplay.distributed.elastic.rendezvous) RendezvousParameters (class in tensorplay.distributed.elastic.rendezvous) RendezvousSettings (class in tensorplay.distributed.elastic.rendezvous) RendezvousStateError RendezvousStoreInfo (class in tensorplay.distributed.elastic.rendezvous) RendezvousTimeoutError renorm() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) renorm_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) repeat() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) repeat_interleave() (in module tensorplay.functional) replace() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) replace_all_batch_norm_modules_() (in module tensorplay.func) replace_all_uses() (tensorplay.export.ExportGraphSignature method) replay() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) Replicate (class in tensorplay.distributed.tensor) replication_pad1d() (in module tensorplay) replication_pad2d() (in module tensorplay) replication_pad3d() (in module tensorplay) replication_pad_nd() (in module tensorplay) requires_grad_() (in module tensorplay) (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) reset() (in module tensorplay.compiler) (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) reset_accumulated_host_memory_stats() (in module tensorplay.cuda) reset_accumulated_memory_stats() (in module tensorplay.cuda) reset_parameters() (tensorplay.ao.nn.qat.Linear method) reset_peak_host_memory_stats() (in module tensorplay.cuda) reset_peak_memory_stats() (in module tensorplay.cuda) reshape() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) reshape_as() (in module tensorplay) ReshapeTransform (class in tensorplay.distributions) residuals (tensorplay.linalg.LstsqResult attribute) resizable() (tensorplay.UntypedStorage method) resize_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.UntypedStorage method) resize_as_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) resize_as_sparse_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) resolve_conj() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) resolve_map_location() (in module tensorplay.serialization) resolve_name() (in module tensorplay.overrides) resolve_neg() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) restore_flash_attention_impl() (in module tensorplay.nn.attention) result_type() (in module tensorplay.functional) retain_grad() (in module tensorplay) retains_grad() (in module tensorplay) reverse() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.profiler.EventList method) rfft() (in module tensorplay.fft._transforms) rfft2() (in module tensorplay.fft._transforms) rfftfreq() (in module tensorplay.fft._frequency) rfftn() (in module tensorplay.fft._transforms) rfind() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rindex() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rjust() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rms_norm() (in module tensorplay.functional) (in module tensorplay.nn.functional) rnn_relu() (in module tensorplay.functional) rnn_relu_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) rnn_tanh() (in module tensorplay.functional) rnn_tanh_cell() (in module tensorplay.functional) (in module tensorplay.nn.functional) roll() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) rot90() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) rotary_embedding() (in module tensorplay) round() (in module tensorplay.functional) (in module tensorplay.special._elementwise) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) (tensorplay.SymInt method) round_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) row_indices() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) row_indices_copy() (in module tensorplay.functional) row_stack() (in module tensorplay.functional) RowwiseParallel (class in tensorplay.distributed.tensor.parallel) rpartition() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rpc_async() (in module tensorplay.distributed.rpc) rpc_sync() (in module tensorplay.distributed.rpc) RRef (class in tensorplay.distributed.rpc) rrelu() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) rrelu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) rrelu_with_noise() (in module tensorplay) rrelu_with_noise_() (in module tensorplay) rsample() (tensorplay.distributions.Bernoulli method) (tensorplay.distributions.Binomial method) (tensorplay.distributions.Categorical method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.HalfCauchy method) (tensorplay.distributions.HalfNormal method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.OneHotCategorical method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) rsplit() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rsqrt() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) rsqrt_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) rstrip() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) rsub() (in module tensorplay.functional) run() (tensorplay.distributed.elastic.agent.server.ElasticAgent method) (tensorplay.distributed.elastic.agent.server.LocalElasticAgent method) (tensorplay.distributed.elastic.agent.server.SimpleElasticAgent method) (tensorplay.FileCheck method) (tensorplay.testing.FileCheck method) run_decompositions() (tensorplay.export.ExportedProgram method) RunProcsResult (class in tensorplay.distributed.elastic.multiprocessing) RunResult (class in tensorplay.distributed.elastic.agent.server) |
| --- | --- |

## S

| S (tensorplay.linalg.SVDResult attribute) safe_globals (class in tensorplay.serialization) sample (tensorplay.onnx.OpContext attribute) sample() (in module tensorplay) (tensorplay.distributions.Beta method) (tensorplay.distributions.Cauchy method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.Exponential method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.GeneralizedPareto method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.HalfCauchy method) (tensorplay.distributions.HalfNormal method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.Laplace method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.Uniform method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) sample_n() (tensorplay.distributions.Bernoulli method) (tensorplay.distributions.Beta method) (tensorplay.distributions.Binomial method) (tensorplay.distributions.Categorical method) (tensorplay.distributions.Cauchy method) (tensorplay.distributions.Chi2 method) (tensorplay.distributions.ContinuousBernoulli method) (tensorplay.distributions.Dirichlet method) (tensorplay.distributions.Distribution method) (tensorplay.distributions.Exponential method) (tensorplay.distributions.ExponentialFamily method) (tensorplay.distributions.FisherSnedecor method) (tensorplay.distributions.Gamma method) (tensorplay.distributions.GeneralizedPareto method) (tensorplay.distributions.Geometric method) (tensorplay.distributions.Gumbel method) (tensorplay.distributions.HalfCauchy method) (tensorplay.distributions.HalfNormal method) (tensorplay.distributions.Independent method) (tensorplay.distributions.InverseGamma method) (tensorplay.distributions.Kumaraswamy method) (tensorplay.distributions.Laplace method) (tensorplay.distributions.LKJCholesky method) (tensorplay.distributions.LogisticNormal method) (tensorplay.distributions.LogNormal method) (tensorplay.distributions.LowRankMultivariateNormal method) (tensorplay.distributions.MixtureSameFamily method) (tensorplay.distributions.Multinomial method) (tensorplay.distributions.MultivariateNormal method) (tensorplay.distributions.NegativeBinomial method) (tensorplay.distributions.Normal method) (tensorplay.distributions.OneHotCategorical method) (tensorplay.distributions.OneHotCategoricalStraightThrough method) (tensorplay.distributions.Pareto method) (tensorplay.distributions.Poisson method) (tensorplay.distributions.RelaxedBernoulli method) (tensorplay.distributions.RelaxedOneHotCategorical method) (tensorplay.distributions.StudentT method) (tensorplay.distributions.TransformedDistribution method) (tensorplay.distributions.Uniform method) (tensorplay.distributions.VonMises method) (tensorplay.distributions.Weibull method) (tensorplay.distributions.Wishart method) Sampler (class in tensorplay.utils.data) save() (in module tensorplay.distributed.checkpoint) (in module tensorplay.export) (in module tensorplay.serialization) (in module tensorplay.serialization.api) save_binary() (tensorplay.package.PackageExporter method) save_module() (tensorplay.package.PackageExporter method) save_on_cpu() (in module tensorplay.autograd.graph) save_pickle() (tensorplay.package.PackageExporter method) save_source_file() (tensorplay.package.PackageExporter method) save_source_string() (tensorplay.package.PackageExporter method) save_state_dict() (in module tensorplay.distributed.checkpoint) save_text() (tensorplay.package.PackageExporter method) saved_tensors_hooks() (in module tensorplay.autograd.graph) SavePlan (class in tensorplay.distributed.checkpoint) SavePlanner (class in tensorplay.distributed.checkpoint) Scalar (class in tensorplay) scalar_tensor() (in module tensorplay.functional) scale() (tensorplay.GradScaler method) scaled_dot_product_attention() (in module tensorplay) (in module tensorplay.nn.functional) scaled_grouped_mm() (in module tensorplay.nn.functional) scaled_mm() (in module tensorplay.nn.functional) scaled_modified_bessel_k0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) scaled_modified_bessel_k1() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) scatter() (in module tensorplay.distributed.nn.functional) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) scatter_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) scatter_add() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) scatter_add_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) scatter_reduce() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) scatter_reduce_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) schedule() (in module tensorplay.profiler) Schedule1F1B (class in tensorplay.distributed.pipelining) ScheduleDualPipeV (class in tensorplay.distributed.pipelining) ScheduleGPipe (class in tensorplay.distributed.pipelining) ScheduleInterleaved1F1B (class in tensorplay.distributed.pipelining) ScheduleInterleavedZeroBubble (class in tensorplay.distributed.pipelining) ScheduleLoopedBFS (class in tensorplay.distributed.pipelining) ScheduleZBVZeroBubble (class in tensorplay.distributed.pipelining) sdpa_kernel() (in module tensorplay.nn.attention) SDPBackend (class in tensorplay.nn.attention) searchsorted() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) sec() (tensorplay.nn.attention.bias.CausalBias method) seed() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) (tensorplay.Generator method) seed_all() (in module tensorplay.cuda.random) segment_reduce() (in module tensorplay.functional) select() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) select_backward() (tensorplay.masked.MaskedTensor method) select_copy() (in module tensorplay.functional) select_scatter() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) SelectiveCheckpointContext (class in tensorplay.utils.checkpoint) selu() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) selu_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) SequenceParallel (class in tensorplay.distributed.tensor.parallel) SequentialSampler (class in tensorplay.utils.data) SerializationFormat (class in tensorplay.distributed.checkpoint) serialize() (tensorplay.export.ExportedProgram method) SerializePartialToString() (tensorplay.onnx.TensorProto method) SerializeToString() (tensorplay.onnx.TensorProto method) set_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) set_autocast_cache_enabled() (in module tensorplay) set_autocast_dtype() (in module tensorplay) set_autocast_enabled() (in module tensorplay) set_backoff_factor() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) set_checkpoint_debug_enabled() (in module tensorplay.utils.checkpoint) set_checkpoint_early_stop() (in module tensorplay.utils.checkpoint) set_closed() (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) set_conditional_handle_for_current_node() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) set_cpu_parent() (tensorplay.autograd.profiler_util.FunctionEvent method) set_crc32_options() (in module tensorplay.serialization) set_data() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) set_default_backend() (in module tensorplay.compiler) set_default_device() (in module tensorplay) set_default_dtype() (in module tensorplay) set_default_load_endianness() (in module tensorplay.serialization) set_default_mmap_options (class in tensorplay.serialization) set_default_validate_args() (tensorplay.distributions.Bernoulli static method) (tensorplay.distributions.Beta static method) (tensorplay.distributions.Binomial static method) (tensorplay.distributions.Categorical static method) (tensorplay.distributions.Cauchy static method) (tensorplay.distributions.Chi2 static method) (tensorplay.distributions.ContinuousBernoulli static method) (tensorplay.distributions.Dirichlet static method) (tensorplay.distributions.Distribution static method) (tensorplay.distributions.Exponential static method) (tensorplay.distributions.ExponentialFamily static method) (tensorplay.distributions.FisherSnedecor static method) (tensorplay.distributions.Gamma static method) (tensorplay.distributions.GeneralizedPareto static method) (tensorplay.distributions.Geometric static method) (tensorplay.distributions.Gumbel static method) (tensorplay.distributions.HalfCauchy static method) (tensorplay.distributions.HalfNormal static method) (tensorplay.distributions.Independent static method) (tensorplay.distributions.InverseGamma static method) (tensorplay.distributions.Kumaraswamy static method) (tensorplay.distributions.Laplace static method) (tensorplay.distributions.LKJCholesky static method) (tensorplay.distributions.LogisticNormal static method) (tensorplay.distributions.LogNormal static method) (tensorplay.distributions.LowRankMultivariateNormal static method) (tensorplay.distributions.MixtureSameFamily static method) (tensorplay.distributions.Multinomial static method) (tensorplay.distributions.MultivariateNormal static method) (tensorplay.distributions.NegativeBinomial static method) (tensorplay.distributions.Normal static method) (tensorplay.distributions.OneHotCategorical static method) (tensorplay.distributions.OneHotCategoricalStraightThrough static method) (tensorplay.distributions.Pareto static method) (tensorplay.distributions.Poisson static method) (tensorplay.distributions.RelaxedBernoulli static method) (tensorplay.distributions.RelaxedOneHotCategorical static method) (tensorplay.distributions.StudentT static method) (tensorplay.distributions.TransformedDistribution static method) (tensorplay.distributions.Uniform static method) (tensorplay.distributions.VonMises static method) (tensorplay.distributions.Weibull static method) (tensorplay.distributions.Wishart static method) set_detect_anomaly (class in tensorplay.autograd.anomaly_mode) set_deterministic_debug_mode() (in module tensorplay) set_device() (in module tensorplay.cuda) set_device_index() (in module tensorplay.accelerator) set_device_states() (in module tensorplay.utils.checkpoint) set_exception() (tensorplay.Future method) set_extra_state() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) set_filename() (in module tensorplay.cuda.tunable) set_flags() (in module tensorplay.backends.nnpack) set_float32_matmul_precision() (in module tensorplay) set_grad_enabled (class in tensorplay.autograd.grad_mode) set_growth_factor() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) set_growth_interval() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.GradScaler method) set_max_tuning_duration() (in module tensorplay.cuda.tunable) set_max_tuning_samples() (in module tensorplay.cuda.tunable) set_model_state_dict() (in module tensorplay.distributed.checkpoint) set_num_threads() (in module tensorplay) set_optimizer_state_dict() (in module tensorplay.distributed.checkpoint) set_per_process_memory_fraction() (in module tensorplay.cuda) set_printoptions() (in module tensorplay) set_result() (tensorplay.Future method) (tensorplay.futures.Future method) set_rng_state() (in module tensorplay) (in module tensorplay.cuda.random) (in module tensorplay.random) set_rng_state_all() (in module tensorplay.cuda.random) set_sharing_strategy() (in module tensorplay.multiprocessing) set_state() (tensorplay.Generator method) set_state_dict() (in module tensorplay.distributed.checkpoint) set_stream() (in module tensorplay.cuda) set_submodule() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) set_sync_debug_mode() (in module tensorplay.cuda) set_verbose() (in module tensorplay.cuda.tunable) setdefault() (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.export.CustomDecompTable method) SetInParent() (tensorplay.onnx.TensorProto method) setup_context() (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) sgn() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sgn_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) shallow_copy_data() (tensorplay.nn.attention.bias.CausalBias method) ShapesCollection (class in tensorplay.export) Shard (class in tensorplay.distributed.tensor) ShardedGradScaler (class in tensorplay.distributed.fsdp.sharded_grad_scaler) ShardedOptimStateDictConfig (class in tensorplay.distributed.fsdp) ShardedStateDictConfig (class in tensorplay.distributed.fsdp) ShardingStrategy (class in tensorplay.distributed.fsdp) share_comm_ctx() (in module tensorplay.distributed.fsdp) share_memory() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) share_memory_() (tensorplay.nn.attention.bias.CausalBias method) shifted_chebyshev_polynomial_t() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_u() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_v() (in module tensorplay) (in module tensorplay.functional) shifted_chebyshev_polynomial_w() (in module tensorplay) (in module tensorplay.functional) shutdown() (in module tensorplay.distributed.rpc) (tensorplay.distributed.elastic.rendezvous.RendezvousHandler method) Sigmoid (class in tensorplay.ao.nn.quantized) sigmoid() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sigmoid_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sigmoid_backward() (tensorplay.nn.attention.bias.CausalBias method) SigmoidTransform (class in tensorplay.distributions) sign (tensorplay.distributions.AbsTransform property) (tensorplay.distributions.CatTransform property) (tensorplay.distributions.CorrCholeskyTransform property) (tensorplay.distributions.LowerCholeskyTransform property) (tensorplay.distributions.PositiveDefiniteTransform property) (tensorplay.distributions.ReshapeTransform property) (tensorplay.distributions.SoftmaxTransform property) (tensorplay.distributions.StackTransform property) (tensorplay.distributions.StickBreakingTransform property) (tensorplay.distributions.Transform property) (tensorplay.linalg.SlogdetResult attribute) |sign() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sign_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) SignalException signbit() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) silu() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) silu_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) silu_and_mul() (in module tensorplay) silu_mul() (in module tensorplay) SimpleElasticAgent (class in tensorplay.distributed.elastic.agent.server) sin() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) sin_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sinc() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sinc_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) singular_values (tensorplay.linalg.LstsqResult attribute) sinh() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) sinh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Size (class in tensorplay) size() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.UntypedStorage method) skip_data (class in tensorplay.serialization) slice() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) slice_backward() (tensorplay.masked.MaskedTensor method) slice_copy() (in module tensorplay.functional) slice_inverse() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) slice_scatter() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) slogdet() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) SlogdetResult (class in tensorplay.linalg) slow_conv3d() (in module tensorplay) slow_conv3d_forward() (in module tensorplay) slow_conv_dilated2d() (in module tensorplay) slow_conv_dilated3d() (in module tensorplay) slow_conv_transpose2d() (in module tensorplay) slow_conv_transpose3d() (in module tensorplay) smm() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) smooth_l1_loss() (in module tensorplay) (in module tensorplay.nn.functional) soft_margin_loss() (in module tensorplay) (in module tensorplay.nn.functional) softmax() (in module tensorplay.functional) (in module tensorplay.masked) (in module tensorplay.nn.functional) (in module tensorplay.special._elementwise) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) SoftmaxTransform (class in tensorplay.distributions) softmin() (in module tensorplay.masked) (in module tensorplay.nn.functional) softplus() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) SoftplusTransform (class in tensorplay.distributions) softshrink() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) softshrink_backward() (tensorplay.nn.attention.bias.CausalBias method) softsign() (in module tensorplay.nn.functional) solution (tensorplay.linalg.LstsqResult attribute) solve() (in module tensorplay.linalg) solve_ex() (in module tensorplay.linalg) solve_triangular() (in module tensorplay.linalg) sort() (in module tensorplay.functional) (tensorplay.autograd.profiler_util.EventList method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.profiler.EventList method) sorted_indices (tensorplay.nn.utils.rnn.PackedSequence attribute) sparse_() (in module tensorplay.nn.init) sparse_add() (in module tensorplay) sparse_blocksize() (tensorplay.nn.attention.bias.CausalBias method) sparse_bsc_tensor() (in module tensorplay.functional) sparse_bsr_tensor() (in module tensorplay.functional) sparse_compressed_tensor() (in module tensorplay) sparse_coo_tensor() (in module tensorplay.functional) sparse_csc_tensor() (in module tensorplay.functional) sparse_csr_tensor() (in module tensorplay.functional) sparse_dim() (in module tensorplay) sparse_mask() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) sparse_mm() (in module tensorplay) sparse_mul() (in module tensorplay) sparse_resize_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) sparse_resize_and_clear_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) sparse_sampled_addmm() (in module tensorplay) sparse_sum() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) sparsity() (tensorplay.nn.attention.BlockMask method) spdiags() (in module tensorplay) special_airy_ai() (in module tensorplay) special_bessel_j0() (in module tensorplay) special_bessel_j1() (in module tensorplay) special_bessel_y0() (in module tensorplay) special_bessel_y1() (in module tensorplay) special_chebyshev_polynomial_t() (in module tensorplay) special_chebyshev_polynomial_u() (in module tensorplay) special_chebyshev_polynomial_v() (in module tensorplay) special_chebyshev_polynomial_w() (in module tensorplay) special_digamma() (in module tensorplay) special_entr() (in module tensorplay) special_erf() (in module tensorplay) special_erfc() (in module tensorplay) special_erfcx() (in module tensorplay) special_erfinv() (in module tensorplay) special_exp2() (in module tensorplay) special_expit() (in module tensorplay) special_expm1() (in module tensorplay) special_gammainc() (in module tensorplay) special_gammaincc() (in module tensorplay) special_gammaln() (in module tensorplay) special_hermite_polynomial_h() (in module tensorplay) special_hermite_polynomial_he() (in module tensorplay) special_i0() (in module tensorplay) special_i0e() (in module tensorplay) special_i1() (in module tensorplay) special_i1e() (in module tensorplay) special_laguerre_polynomial_l() (in module tensorplay) special_legendre_polynomial_p() (in module tensorplay) special_log1p() (in module tensorplay) special_log_ndtr() (in module tensorplay) special_log_softmax() (in module tensorplay) special_logit() (in module tensorplay) special_logsumexp() (in module tensorplay) special_modified_bessel_i0() (in module tensorplay) special_modified_bessel_i1() (in module tensorplay) special_modified_bessel_k0() (in module tensorplay) special_modified_bessel_k1() (in module tensorplay) special_multigammaln() (in module tensorplay) special_ndtr() (in module tensorplay) special_ndtri() (in module tensorplay) special_polygamma() (in module tensorplay) special_psi() (in module tensorplay) special_round() (in module tensorplay) special_scaled_modified_bessel_k0() (in module tensorplay) special_scaled_modified_bessel_k1() (in module tensorplay) special_shifted_chebyshev_polynomial_t() (in module tensorplay) special_shifted_chebyshev_polynomial_u() (in module tensorplay) special_shifted_chebyshev_polynomial_v() (in module tensorplay) special_shifted_chebyshev_polynomial_w() (in module tensorplay) special_sinc() (in module tensorplay) special_softmax() (in module tensorplay) special_spherical_bessel_j0() (in module tensorplay) special_xlog1py() (in module tensorplay) special_xlogy() (in module tensorplay) special_zeta() (in module tensorplay) spherical_bessel_j0() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) split() (in module tensorplay.functional) (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) split_copy() (in module tensorplay.functional) split_with_sizes() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) split_with_sizes_copy() (in module tensorplay.functional) splitlines() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) SplitPoint (class in tensorplay.distributed.pipelining) sqrt() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) sqrt_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) square() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) square_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) squared_difference() (tensorplay.nn.attention.bias.CausalBias method) squeeze() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) squeeze_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) squeeze_copy() (in module tensorplay.functional) sspaddmm() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) stack() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) stack_module_state() (in module tensorplay.func) StackDataset (class in tensorplay.utils.data) StackTransform (class in tensorplay.distributions) stage_and_launch() (tensorplay.accelerator.Graph method) (tensorplay.cuda.graphs.CUDAGraph method) start() (in module tensorplay.cuda.profiler) (tensorplay.distributed.elastic.multiprocessing.PContext method) start_processes() (in module tensorplay.distributed.elastic.multiprocessing) startswith() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) Stat (class in tensorplay.monitor) STAT_EVENT (in module tensorplay.monitor) state_dict (tensorplay.export.ExportedProgram property) state_dict() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.distributed.optim.PostLocalSGDOptimizer method) (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.GradScaler method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) StateDictConfig (class in tensorplay.distributed.fsdp) StateDictOptions (class in tensorplay.distributed.checkpoint) StateDictSettings (class in tensorplay.distributed.fsdp) StateDictType (class in tensorplay.distributed.fsdp) Stateful (class in tensorplay.distributed.checkpoint) static_dims (tensorplay.onnx.GraphBuilder attribute) statically_known_true() (tensorplay.SymBool method) StaticTCPRendezvous (class in tensorplay.distributed.elastic.rendezvous) StatResult (class in tensorplay.monitor) stax() (in module tensorplay.stax) Std (class in tensorplay.distributed.elastic.multiprocessing) std() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) std_mean() (in module tensorplay.functional) stddev (tensorplay.distributions.Bernoulli property) (tensorplay.distributions.Beta property) (tensorplay.distributions.Binomial property) (tensorplay.distributions.Categorical property) (tensorplay.distributions.Cauchy property) (tensorplay.distributions.Chi2 property) (tensorplay.distributions.Dirichlet property) (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.FisherSnedecor property) (tensorplay.distributions.Gamma property) (tensorplay.distributions.GeneralizedPareto property) (tensorplay.distributions.Geometric property) (tensorplay.distributions.HalfCauchy property) (tensorplay.distributions.HalfNormal property) (tensorplay.distributions.Independent property) (tensorplay.distributions.InverseGamma property) (tensorplay.distributions.Kumaraswamy property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.LogNormal property) (tensorplay.distributions.LowRankMultivariateNormal property) (tensorplay.distributions.MixtureSameFamily property) (tensorplay.distributions.Multinomial property) (tensorplay.distributions.MultivariateNormal property) (tensorplay.distributions.NegativeBinomial property) (tensorplay.distributions.OneHotCategorical property) (tensorplay.distributions.OneHotCategoricalStraightThrough property) (tensorplay.distributions.Pareto property) (tensorplay.distributions.Poisson property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.StudentT property) (tensorplay.distributions.TransformedDistribution property) (tensorplay.distributions.VonMises property) (tensorplay.distributions.Weibull property) (tensorplay.distributions.Wishart property) step() (tensorplay.distributed.fsdp.sharded_grad_scaler.ShardedGradScaler method) (tensorplay.distributed.optim.PostLocalSGDOptimizer method) (tensorplay.distributed.optim.ZeroRedundancyOptimizer method) (tensorplay.GradScaler method) (tensorplay.profiler.profile method) (tensorplay.profiler.profiler.profile method) stft() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) StickBreakingTransform (class in tensorplay.distributions) stop() (in module tensorplay.cuda.profiler) storage_offset() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) storage_to_tensor_type() (in module tensorplay.serialization) StorageMeta (class in tensorplay.distributed.checkpoint) StorageReader (class in tensorplay.distributed.checkpoint) StorageType (in module tensorplay.serialization) StorageWriter (class in tensorplay.distributed.checkpoint) store (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) Stream (class in tensorplay.cuda.streams) stream() (in module tensorplay.cuda) StreamContext (class in tensorplay.cuda) stride() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) StringTable (class in tensorplay.autograd.profiler_util) strip() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) StudentT (class in tensorplay.distributions) sub() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sub_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Subset (class in tensorplay.utils.data) SubsetRandomSampler (class in tensorplay.utils.data) substitute() (tensorplay.distributed.elastic.utils.macros class method) substitute_in_graph() (in module tensorplay.compiler) subtract() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) subtract_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sum() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) sum_to_size() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) support (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) supported_activities() (in module tensorplay.profiler) svd() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) SVDResult (class in tensorplay.linalg) svdvals() (in module tensorplay.linalg) swapaxes() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) swapaxes_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) swapcase() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) swapdims() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) swapdims_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) swish() (tensorplay.nn.attention.bias.CausalBias method) sym_and() (tensorplay.SymBool method) sym_constrain_range() (in module tensorplay.functional) sym_constrain_range_for_size() (in module tensorplay.functional) sym_eq() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_float() (in module tensorplay) (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_ge() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_gt() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_int() (in module tensorplay) (tensorplay.SymFloat method) sym_is_contiguous() (in module tensorplay) sym_ite() (in module tensorplay) sym_le() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_lt() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_max() (in module tensorplay) sym_min() (in module tensorplay) sym_ne() (tensorplay.SymFloat method) (tensorplay.SymInt method) sym_not() (in module tensorplay) (tensorplay.SymBool method) sym_numel() (in module tensorplay) sym_or() (tensorplay.SymBool method) sym_size() (in module tensorplay) sym_sqrt() (in module tensorplay) sym_storage_offset() (in module tensorplay) sym_stride() (in module tensorplay) sym_sum() (in module tensorplay) sym_xor() (tensorplay.SymBool method) symbolic() (tensorplay.SymBool static method) (tensorplay.SymFloat static method) (tensorplay.SymInt static method) SymBool (class in tensorplay) SymBoolArgument (class in tensorplay.export) SymFloat (class in tensorplay) SymFloatArgument (class in tensorplay.export) SymInt (class in tensorplay) SymIntArgument (class in tensorplay.export) synchronize() (in module tensorplay.accelerator) (in module tensorplay.cuda) (tensorplay.cuda.streams.Event method) (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) |
| --- | --- |

## T

| t() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) t_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) t_copy() (in module tensorplay.functional) table() (tensorplay.autograd.profiler_util.EventList method) (tensorplay.profiler.EventList method) (tensorplay.profiler.profile method) (tensorplay.profiler.profiler.profile method) take() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) take_along_dim() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) tan() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) tan_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) Tanh (class in tensorplay.ao.nn.quantized) tanh() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) tanh_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) tanh_backward() (tensorplay.nn.attention.bias.CausalBias method) tanhshrink() (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) TanhTransform (class in tensorplay.distributions) temperature() (in module tensorplay.cuda) Tensor (in module tensorplay) tensor() (in module tensorplay) tensor_constants (tensorplay.export.ExportedProgram property) tensor_split() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) TensorArgument (class in tensorplay.export) tensorboard_trace_handler() (in module tensorplay.profiler) TensorboardEventHandler (class in tensorplay.monitor) TensorDataset (class in tensorplay.utils.data) tensordot() (in module tensorplay._shape_funcs) tensorinv() (in module tensorplay.linalg) TensorPipeRpcBackendOptions (class in tensorplay.distributed.rpc) tensorplay.accelerator.graphs module tensorplay.accelerator.memory module tensorplay.accelerator.random module tensorplay.amp module tensorplay.amp.autocast_mode module tensorplay.amp.grad_scaler module tensorplay.backends module tensorplay.cuda.tunable module tensorplay.distributed.algorithms module tensorplay.distributed.algorithms.ddp_comm_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.ddp_zero_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.debugging_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.default_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.mixed_precision_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.optimizer_overlap_hooks module tensorplay.distributed.algorithms.ddp_comm_hooks.post_localSGD_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.powerSGD_hook module tensorplay.distributed.algorithms.ddp_comm_hooks.quantization_hooks module tensorplay.distributed.algorithms.join module tensorplay.distributed.algorithms.model_averaging module tensorplay.distributed.algorithms.model_averaging.averagers module tensorplay.distributed.algorithms.model_averaging.hierarchical_model_averager module tensorplay.distributed.algorithms.model_averaging.utils module tensorplay.distributed.argparse_util module tensorplay.distributed.checkpoint.api module tensorplay.distributed.checkpoint.default_planner module tensorplay.distributed.checkpoint.filesystem module tensorplay.distributed.checkpoint.hf_storage module tensorplay.distributed.checkpoint.metadata module tensorplay.distributed.checkpoint.optimizer module tensorplay.distributed.checkpoint.planner module tensorplay.distributed.checkpoint.planner_helpers module tensorplay.distributed.checkpoint.quantized_hf_storage module tensorplay.distributed.checkpoint.resharding module tensorplay.distributed.checkpoint.state_dict module tensorplay.distributed.checkpoint.state_dict_loader module tensorplay.distributed.checkpoint.state_dict_saver module tensorplay.distributed.checkpoint.stateful module tensorplay.distributed.checkpoint.storage module tensorplay.distributed.checkpoint.utils module tensorplay.distributed.constants module tensorplay.distributed.device_mesh module tensorplay.distributed.distributed_core module tensorplay.distributed.elastic module tensorplay.distributed.elastic.agent.server.api module tensorplay.distributed.elastic.agent.server.local_elastic_agent module tensorplay.distributed.elastic.events.api module tensorplay.distributed.elastic.events.handlers module tensorplay.distributed.elastic.metrics.api module tensorplay.distributed.elastic.multiprocessing.api module tensorplay.distributed.elastic.multiprocessing.errors.error_handler module tensorplay.distributed.elastic.multiprocessing.errors.handlers module tensorplay.distributed.elastic.multiprocessing.redirects module tensorplay.distributed.elastic.multiprocessing.tail_log module tensorplay.distributed.elastic.rendezvous.api module tensorplay.distributed.elastic.rendezvous.c10d_rendezvous_backend module tensorplay.distributed.elastic.rendezvous.dynamic_rendezvous module tensorplay.distributed.elastic.rendezvous.etcd_rendezvous module tensorplay.distributed.elastic.rendezvous.etcd_rendezvous_backend module tensorplay.distributed.elastic.rendezvous.etcd_server module tensorplay.distributed.elastic.rendezvous.etcd_store module tensorplay.distributed.elastic.rendezvous.static_tcp_rendezvous module tensorplay.distributed.elastic.rendezvous.utils module tensorplay.distributed.elastic.timer.api module tensorplay.distributed.elastic.timer.file_based_local_timer module tensorplay.distributed.elastic.timer.local_timer module tensorplay.distributed.elastic.utils module tensorplay.distributed.elastic.utils.api module tensorplay.distributed.elastic.utils.data module tensorplay.distributed.elastic.utils.data.cycling_iterator module tensorplay.distributed.elastic.utils.data.elastic_distributed_sampler module tensorplay.distributed.elastic.utils.distributed module tensorplay.distributed.elastic.utils.log_level module tensorplay.distributed.elastic.utils.logging module tensorplay.distributed.elastic.utils.store module tensorplay.distributed.fsdp.api module tensorplay.distributed.fsdp.fully_sharded_data_parallel module tensorplay.distributed.fsdp.sharded_grad_scaler module tensorplay.distributed.fsdp.wrap module tensorplay.distributed.launcher module tensorplay.distributed.logging_handlers module tensorplay.distributed.nn module tensorplay.distributed.nn.api module tensorplay.distributed.nn.api.remote_module module tensorplay.distributed.nn.functional module tensorplay.distributed.nn.jit module tensorplay.distributed.nn.jit.instantiator module tensorplay.distributed.nn.jit.templates module tensorplay.distributed.nn.jit.templates.remote_module_template module tensorplay.distributed.optim.apply_optimizer_in_backward module tensorplay.distributed.optim.functional_adadelta module tensorplay.distributed.optim.functional_adagrad module tensorplay.distributed.optim.functional_adam module tensorplay.distributed.optim.functional_adamax module tensorplay.distributed.optim.functional_adamw module tensorplay.distributed.optim.functional_rmsprop module tensorplay.distributed.optim.functional_rprop module tensorplay.distributed.optim.functional_sgd module tensorplay.distributed.optim.named_optimizer module tensorplay.distributed.optim.optimizer module tensorplay.distributed.optim.post_localSGD_optimizer module tensorplay.distributed.optim.utils module tensorplay.distributed.optim.zero_redundancy_optimizer module tensorplay.distributed.p10d_logger module tensorplay.distributed.remote_device module tensorplay.distributed.rendezvous module tensorplay.distributed.rpc.api module tensorplay.distributed.rpc.backend_registry module tensorplay.distributed.rpc.constants module tensorplay.distributed.rpc.functions module tensorplay.distributed.rpc.internal module tensorplay.distributed.rpc.options module tensorplay.distributed.rpc.rref_proxy module tensorplay.distributed.rpc.server_process_global_profiler module tensorplay.distributed.tensor.parallel.api module tensorplay.distributed.tensor.parallel.ddp module tensorplay.distributed.tensor.parallel.fsdp module tensorplay.distributed.tensor.parallel.input_reshard module tensorplay.distributed.tensor.parallel.loss module tensorplay.distributed.tensor.parallel.style module tensorplay.distributed.utils module tensorplay.func module tensorplay.multiprocessing.pool module tensorplay.multiprocessing.queue module tensorplay.multiprocessing.reductions module tensorplay.nn.attention module tensorplay.nn.attention.bias module tensorplay.nn.attention.omni_attention module tensorplay.nn.attention.varlen module tensorplay.nn.backends module tensorplay.nn.backends.thnn module tensorplay.nn.common_types module tensorplay.nn.cpp module tensorplay.nn.functional module tensorplay.nn.grad module tensorplay.nn.init module tensorplay.nn.modules.activation module tensorplay.nn.modules.adaptive module tensorplay.nn.modules.batchnorm module tensorplay.nn.modules.channelshuffle module tensorplay.nn.modules.container module tensorplay.nn.modules.conv module tensorplay.nn.modules.distance module tensorplay.nn.modules.dropout module tensorplay.nn.modules.flatten module tensorplay.nn.modules.folding module tensorplay.nn.modules.instancenorm module tensorplay.nn.modules.lazy module tensorplay.nn.modules.linear module tensorplay.nn.modules.loss module tensorplay.nn.modules.module module tensorplay.nn.modules.normalization module tensorplay.nn.modules.padding module tensorplay.nn.modules.pixelshuffle module tensorplay.nn.modules.pooling module tensorplay.nn.modules.rnn module tensorplay.nn.modules.sparse module tensorplay.nn.modules.transformer module tensorplay.nn.modules.upsampling module tensorplay.nn.modules.utils module tensorplay.nn.parallel.comm module tensorplay.nn.parallel.distributed module tensorplay.nn.parallel.parallel_apply module tensorplay.nn.parallel.replicate module tensorplay.nn.parallel.scatter_gather module tensorplay.nn.parameter module tensorplay.nn.utils.clip_grad module tensorplay.nn.utils.convert_parameters module tensorplay.nn.utils.fusion module |tensorplay.nn.utils.init module tensorplay.nn.utils.memory_format module tensorplay.nn.utils.parametrizations module tensorplay.nn.utils.parametrize module tensorplay.nn.utils.prune module tensorplay.nn.utils.rnn module tensorplay.nn.utils.stateless module tensorplay.optim.lr_scheduler module tensorplay.optim.optimizer module tensorplay.optim.swa_utils module tensorplay.utils.data.datapipes module tensorplay.utils.data.datapipes.dataframe module tensorplay.utils.data.datapipes.iter module tensorplay.utils.data.datapipes.map module tensorplay.utils.data.datapipes.utils module tensorplay.utils.serialization module tensorplay.utils.serialization.config module TensorPlayFunctionMode (class in tensorplay.overrides) TensorProperties (class in tensorplay.distributed.checkpoint) TensorProto (class in tensorplay.onnx) tensorsolve() (in module tensorplay.linalg) TensorStorageMetadata (class in tensorplay.distributed.checkpoint) TensorWriteData (class in tensorplay.distributed.checkpoint) then() (tensorplay.Future method) (tensorplay.futures.Future method) thnn_conv2d() (in module tensorplay) thread_safe_generator() (in module tensorplay.random) threshold() (in module tensorplay.functional) (in module tensorplay.nn.functional) threshold_() (in module tensorplay.functional) (in module tensorplay.nn.functional) (tensorplay.nn.attention.bias.CausalBias method) tile() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) timeout (tensorplay.distributed.elastic.rendezvous.RendezvousParameters property) TimerClient (class in tensorplay.distributed.elastic.timer) TimerServer (class in tensorplay.distributed.elastic.timer) title() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) to() (in module tensorplay) (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.BlockMask method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) (tensorplay.nn.utils.rnn.PackedSequence method) to_bytes() (tensorplay.distributed.elastic.multiprocessing.Std method) (tensorplay.export.WeightType method) (tensorplay.Layout method) (tensorplay.MemoryFormat method) (tensorplay.nn.attention.bias.CausalVariant method) (tensorplay.QScheme method) to_dense() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.attention.BlockMask method) to_empty() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.nn.RemoteModule method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) to_map() (in module tensorplay.distributed.elastic.multiprocessing) (tensorplay.distributed.elastic.utils.macros class method) to_mkldnn() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) to_padded_tensor() (in module tensorplay) (in module tensorplay.nested) (tensorplay.nn.attention.bias.CausalBias method) to_sparse() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) to_sparse_bsc() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) to_sparse_bsr() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) to_sparse_coo() (tensorplay.nn.attention.bias.CausalBias method) to_sparse_csc() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) to_sparse_csr() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) to_string() (tensorplay.nn.attention.BlockMask method) to_sym_float() (tensorplay.SymBool method) (tensorplay.SymInt method) to_sym_int() (tensorplay.SymBool method) TokenArgument (class in tensorplay.export) tolist() (tensorplay.nn.attention.bias.CausalBias method) topk() (in module tensorplay.functional) total_average() (tensorplay.autograd.profiler_util.EventList method) tp_cosine_embedding_loss() (in module tensorplay) tp_export() (in module tensorplay.onnx) tp_hinge_embedding_loss() (in module tensorplay) tp_kl_div() (in module tensorplay) tp_l1_loss() (in module tensorplay) tp_margin_ranking_loss() (in module tensorplay) tp_poisson_nll_loss() (in module tensorplay) tp_soft_margin_loss() (in module tensorplay) TpRendezvousBackend (class in tensorplay.distributed.elastic.rendezvous) trace() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) train() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) Transform (class in tensorplay.distributions) TransformedDistribution (class in tensorplay.distributions) translate() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) transpose() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) transpose_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) transpose_copy() (in module tensorplay.functional) trapezoid() (in module tensorplay.functional) trapz() (in module tensorplay.functional) triangular_solve() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) tril() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) tril_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) tril_indices() (in module tensorplay.functional) triplet_margin_loss() (in module tensorplay.functional) (in module tensorplay.nn.functional) triplet_margin_with_distance_loss() (in module tensorplay.nn.functional) triton_op() (in module tensorplay.library) triu() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) triu_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) triu_indices() (in module tensorplay.functional) true_divide() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) true_divide_() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) trunc() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.SymFloat method) (tensorplay.SymInt method) trunc_() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) trunc_normal_() (in module tensorplay.nn.init) tuning_disable() (in module tensorplay.cuda.tunable) tuning_enable() (in module tensorplay.cuda.tunable) tuning_is_enabled() (in module tensorplay.cuda.tunable) type() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.attention.bias.CausalBias method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) type_as() (in module tensorplay) typename() (in module tensorplay) |
| --- | --- |

## U

| U (tensorplay.linalg.SVDResult attribute) unbind() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) unbind_copy() (in module tensorplay.functional) Unflatten (class in tensorplay.nn.modules.flatten) unflatten() (in module tensorplay.export) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) unflatten_dense_tensors() (in module tensorplay) UnflattenedModule (class in tensorplay.export) unfold() (in module tensorplay) (in module tensorplay.nn.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) unfold_backward() (tensorplay.masked.MaskedTensor method) unfold_copy() (in module tensorplay.functional) Uniform (class in tensorplay.distributions) uniform_() (in module tensorplay) (in module tensorplay.nn.init) (tensorplay.nn.attention.bias.CausalBias method) unique() (in module tensorplay.functional) unique_consecutive() (in module tensorplay.functional) unique_dim() (in module tensorplay) unique_dim_consecutive() (in module tensorplay) UnknownFields() (tensorplay.onnx.TensorProto method) unpack_dual() (in module tensorplay.autograd.forward_ad) unpack_sequence() (in module tensorplay.nn.utils.rnn) unpad_sequence() (in module tensorplay.nn.utils.rnn) unravel_index() (in module tensorplay._shape_funcs) unregister_backend() (in module tensorplay.compiler) unregister_event_handler() (in module tensorplay.monitor) unsafe_chunk() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) |unsafe_split() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) unsafe_split_with_sizes() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) unscale_() (tensorplay.GradScaler method) UnshardHandle (class in tensorplay.distributed.fsdp) unsorted_indices (tensorplay.nn.utils.rnn.PackedSequence attribute) unsqueeze() (in module tensorplay.functional) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) unsqueeze_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) unsqueeze_copy() (in module tensorplay.functional) UnsupportedOperatorError untyped_storage() (tensorplay.nn.attention.bias.CausalBias method) UntypedStorage (class in tensorplay) unused() (in module tensorplay.compiler) update() (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.GradScaler method) upper() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) upsample_bicubic2d() (in module tensorplay) upsample_bilinear2d() (in module tensorplay) upsample_linear1d() (in module tensorplay) upsample_nearest1d() (in module tensorplay) upsample_nearest2d() (in module tensorplay) upsample_nearest3d() (in module tensorplay) upsample_trilinear3d() (in module tensorplay) use_agent_store (tensorplay.distributed.elastic.rendezvous.RendezvousHandler property) use_deterministic_algorithms() (in module tensorplay) use_mem_pool() (in module tensorplay.cuda.memory) utilization() (in module tensorplay.cuda) |
| --- | --- |

## V

| validate_pruning_amount() (in module tensorplay.ao.pruning) Value (class in tensorplay.onnx) value() (tensorplay.Future method) (tensorplay.futures.Future method) values() (in module tensorplay) (tensorplay.autograd.profiler_util.StringTable method) (tensorplay.masked.MaskedTensor method) values_copy() (in module tensorplay.functional) vander() (in module tensorplay.functional) (in module tensorplay.linalg) var() (in module tensorplay.functional) (in module tensorplay.masked) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) var_mean() (in module tensorplay.functional) Variable (class in tensorplay.autograd.variable) VariableMeta (class in tensorplay.autograd.variable) variance (tensorplay.distributions.Distribution property) (tensorplay.distributions.ExponentialFamily property) (tensorplay.distributions.LKJCholesky property) (tensorplay.distributions.LogisticNormal property) (tensorplay.distributions.RelaxedBernoulli property) (tensorplay.distributions.RelaxedOneHotCategorical property) (tensorplay.distributions.TransformedDistribution property) (tensorplay.distributions.VonMises property) varlen_attn() (in module tensorplay.nn.attention.varlen) varlen_attn_out() (in module tensorplay.nn.attention.varlen) vdot() (in module tensorplay.functional) (in module tensorplay.linalg) (tensorplay.nn.attention.bias.CausalBias method) |vecdot() (in module tensorplay.linalg) vector_norm() (in module tensorplay.linalg) VerificationError VerificationResult (class in tensorplay.onnx) verify_model() (in module tensorplay.onnx) version() (in module tensorplay.cuda.nccl) Vh (tensorplay.linalg.SVDResult attribute) vhp() (in module tensorplay.autograd.functional) view() (in module tensorplay) (tensorplay.masked.MaskedTensor method) (tensorplay.nn.attention.bias.CausalBias method) view_as() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) view_as_complex() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) view_as_complex_copy() (in module tensorplay.functional) view_as_real() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) view_as_real_copy() (in module tensorplay.functional) view_copy() (in module tensorplay.functional) vjp() (in module tensorplay.autograd.functional) (in module tensorplay.func) vmap() (in module tensorplay.func) (tensorplay.autograd.function.Function static method) (tensorplay.autograd.function.InplaceFunction static method) (tensorplay.autograd.function.NestedIOFunction static method) VonMises (class in tensorplay.distributions) vsplit() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) vstack() (in module tensorplay.functional) |
| --- | --- |

## W

| wait() (tensorplay.cuda.streams.Event method) (tensorplay.distributed.elastic.multiprocessing.PContext method) (tensorplay.Future method) (tensorplay.futures.Future method) wait_all() (in module tensorplay.futures) wait_event() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) wait_list (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) wait_stream() (tensorplay.cuda.streams.ExternalStream method) (tensorplay.cuda.streams.Stream method) Weibull (class in tensorplay.distributions) weight (tensorplay.ao.quantization.QConfig attribute) WeightedRandomSampler (class in tensorplay.utils.data) WeightType (class in tensorplay.export) where() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) |whichmodule() (tensorplay.package.Importer method) (tensorplay.package.PackageImporter method) WhichOneof() (tensorplay.onnx.TensorProto method) Wishart (class in tensorplay.distributions) Worker (class in tensorplay.distributed.elastic.agent.server) WorkerGroup (class in tensorplay.distributed.elastic.agent.server) WorkerSpec (class in tensorplay.distributed.elastic.agent.server) WorkerState (class in tensorplay.distributed.elastic.agent.server) world_size (tensorplay.distributed.elastic.rendezvous.RendezvousInfo property) wrap_tensorplay_function() (in module tensorplay.overrides) wrap_triton() (in module tensorplay.library) write_file() (in module tensorplay.cuda.tunable) WriteItem (class in tensorplay.distributed.checkpoint) WriteItemType (class in tensorplay.distributed.checkpoint) |
| --- | --- |

## X

| x (tensorplay.onnx.OpContext property) xavier_normal_() (in module tensorplay.nn.init) xavier_uniform_() (in module tensorplay.nn.init) xlog1py() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) |xlog1py_() (in module tensorplay) (tensorplay.nn.attention.bias.CausalBias method) xlogy() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) xlogy_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) |
| --- | --- |

## Z

| zero_() (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) zero_grad() (tensorplay.ao.nn.intrinsic.BNReLU2d method) (tensorplay.ao.nn.intrinsic.BNReLU3d method) (tensorplay.ao.nn.intrinsic.ConvReLU1d method) (tensorplay.ao.nn.intrinsic.ConvReLU2d method) (tensorplay.ao.nn.intrinsic.ConvReLU3d method) (tensorplay.ao.nn.intrinsic.LinearReLU method) (tensorplay.ao.nn.qat.Conv1d method) (tensorplay.ao.nn.qat.Conv2d method) (tensorplay.ao.nn.qat.Conv3d method) (tensorplay.ao.nn.qat.Linear method) (tensorplay.ao.nn.quantized.BatchNorm2d method) (tensorplay.ao.nn.quantized.BatchNorm3d method) (tensorplay.ao.nn.quantized.Conv1d method) (tensorplay.ao.nn.quantized.Conv2d method) (tensorplay.ao.nn.quantized.Conv3d method) (tensorplay.ao.nn.quantized.DeQuantize method) (tensorplay.ao.nn.quantized.dynamic.Linear method) (tensorplay.ao.nn.quantized.ELU method) (tensorplay.ao.nn.quantized.FloatFunctional method) (tensorplay.ao.nn.quantized.FXFloatFunctional method) (tensorplay.ao.nn.quantized.GroupNorm method) (tensorplay.ao.nn.quantized.Hardsigmoid method) (tensorplay.ao.nn.quantized.Hardswish method) (tensorplay.ao.nn.quantized.InstanceNorm1d method) (tensorplay.ao.nn.quantized.InstanceNorm2d method) (tensorplay.ao.nn.quantized.InstanceNorm3d method) (tensorplay.ao.nn.quantized.LayerNorm method) (tensorplay.ao.nn.quantized.LeakyReLU method) (tensorplay.ao.nn.quantized.MaxPool1d method) (tensorplay.ao.nn.quantized.MaxPool2d method) (tensorplay.ao.nn.quantized.MaxPool3d method) (tensorplay.ao.nn.quantized.QFunctional method) (tensorplay.ao.nn.quantized.Quantize method) (tensorplay.ao.nn.quantized.QuantizedLinear method) (tensorplay.ao.nn.quantized.ReLU method) (tensorplay.ao.nn.quantized.ReLU6 method) (tensorplay.ao.nn.quantized.Sigmoid method) (tensorplay.ao.nn.quantized.Tanh method) (tensorplay.ao.quantization.DeQuantStub method) (tensorplay.ao.quantization.FakeQuantize method) (tensorplay.ao.quantization.FixedQParamsFakeQuantize method) (tensorplay.ao.quantization.FixedQParamsObserver method) (tensorplay.ao.quantization.FusedMovingAvgObsFakeQuantize method) (tensorplay.ao.quantization.HistogramObserver method) (tensorplay.ao.quantization.MinMaxObserver method) (tensorplay.ao.quantization.MovingAverageMinMaxObserver method) (tensorplay.ao.quantization.MovingAveragePerChannelMinMaxObserver method) (tensorplay.ao.quantization.PerChannelFakeQuantize method) (tensorplay.ao.quantization.PerChannelMinMaxObserver method) (tensorplay.ao.quantization.PlaceholderObserver method) (tensorplay.ao.quantization.QuantStub method) (tensorplay.distributed.fsdp.FullyShardedDataParallel method) (tensorplay.distributed.pipelining.Pipe method) (tensorplay.nn.modules.flatten.Flatten method) (tensorplay.nn.modules.flatten.Unflatten method) |ZeroRedundancyOptimizer (class in tensorplay.distributed.optim) zeros() (in module tensorplay) (in module tensorplay.distributed.tensor) zeros_() (in module tensorplay.nn.init) zeros_like() (in module tensorplay.functional) zeta() (in module tensorplay) (in module tensorplay.functional) (tensorplay.nn.attention.bias.CausalBias method) zfill() (tensorplay.distributed.elastic.agent.server.WorkerState method) (tensorplay.distributed.elastic.events.NodeState method) (tensorplay.profiler.ProfilerAction method) (tensorplay.profiler.ProfilerActivity method) |
| --- | --- |