Operator List#

No.NameKindStageSinceLabelsDescription
1_adaptive_avg_pool3d_backwardNeuralNetworkAlpha5.1aten, KernelGen, reductionComputes the gradient of 3D adaptive average pooling.
2_flash_attention_forwardNeuralNetworkAlpha5.4aten, NoCPU, KernelGenRun the canonical FlashAttention forward operator.
3_reshape_aliasTensorAlpha5.4aten, KernelGenCreates a view of the input tensor with the given size and stride, sharing the same storage as the input. This is an internal reshape helper that skips the validity checks a normal reshape performs.
4_weight_normNeuralNetworkStable3.0aten, fused, KernelGenReparameterizes a module's weight tensor by decoupling its magnitude (g) from its direction (v). It is a hook that compute the actual weight before each forward pass.
5absMathStable1.0aten, pointwiseComputes the absolute value of each element in input. This is a simple wrapper of the existing torch abs operator.
6abs_MathStable2.2aten, pointwiseThe in-place version of abs(), which is a simple wrapper of the Torch abs operator.
7absoluteMathStable5.3aaten, KernelGenThis is an alias for abs() with the low-level operations implemented by invoking low-level Torch operators.
8absolute_MathAlpha5.4aten, KernelGenThe in-place version of absolute().
9acosMathStable5.0aten, pointwiseReturns a new tensor with the arccosine (in radians) of each element in input.
10acos_MathAlpha5.4aten, pointwise, inplace, KernelGenIn-place version of acos. Computes the arccosine (in radians) of each element of input and writes the result back to the input tensor.
11acoshMathAlpha5.4aten, KernelGenTriton kernel implementation for acosh.
12acosh_MathAlpha5.4aten, KernelGenThe in-place version of acosh().
13act_quant_tritonQuantizationBeta5.3fusedThis is a fused operator.
14adaptive_avg_pool1dNeuralNetworkAlpha5.4aten, KernelGenApplies a one-dimensional adaptive average pooling over an input signal, producing an output of the requested length.
15adaptive_avg_pool2dNeuralNetworkBeta5.3aten, nn.functional, KernelGenApply a 2D adaptive average pooling over an input signal composed of several input planes.
16adaptive_avg_pool2d_backwardNeuralNetworkAlpha5.4aten, KernelGen, reductionComputes the gradient of two-dimensional adaptive average pooling with respect to the input tensor.
17adaptive_avg_pool3d_backwardNeuralNetworkAlpha5.1aten, KernelGen, reductionComputes the gradient of 3D adaptive average pooling.
18adaptive_max_pool2dNeuralNetworkAlpha5.4aten, KernelGenApplies two-dimensional adaptive max pooling and returns both pooled values and the selected input indices.
19adaptive_max_pool2d_backwardNeuralNetworkAlpha5.4aten, KernelGenComputes the input gradient for two-dimensional adaptive max pooling by scattering output gradients to the saved maximum indices.
20adaptive_max_pool3d_backwardNeuralNetworkAlpha5.4aten, KernelGenComputes the gradient for adaptive max pooling 3D.
21addMathStable1.0aten, pointwiseAdd a scalar or tensor to self tensor. If both alpha and other are specified, each element of other is scaled by alpha before being used.
22add_MathStable2.2aten, pointwiseThe in-place version of add().
23add_reluMathAlpha5.4aten, KernelGenTriton kernel implementation for _add_relu.
24add_relu_MathAlpha5.4aten, KernelGenTriton kernel implementation for the in-place _add_relu_().
25add_rms_normNeuralNetworkAlpha5.4aten, KernelGen, NormalizationAdd two inputs element-wise and apply Root Mean Square Layer Normalization.
26addbmmLinearAlgAlpha5.4aten, KernelGenPerforms a batch matrix-matrix product of matrices in batch1 and batch2, with a reduced add step (sum over batch dimension), and adds input to the result. Formula: out = beta * input + alpha * (sum_i batch1_i @ batch2_i).
27addbmm_LinearAlgAlpha5.4aten, KernelGenInplace variant of addbmm. Performs a batch matrix-matrix product of matrices in batch1 and batch2, with a reduced add step, and stores the result in self. Formula: self = beta * self + alpha * (sum_i batch1_i @ batch2_i).
28addcdivLinearAlgStable4.0aten, pointwisePerforms the element-wise division of tensor1 by tensor2, multiplies the result by the scalar value and adds it to input.
29addcdiv_MathAlpha5.1aten, KernelGenThe in-place version of addcdiv(). Performs the element-wise division of tensor1 by tensor2, multiplies the result by the scalar value and adds it to input, storing the result in input.
30addcdiv_outLinearAlgStable5.3aten, pointwise, KernelGenA variant of addcdiv() that assigns the output to the given out parameter..
31addcmulLinearAlgStable4.0aten, pointwisePerforms the element-wise multiplication of tensor1 by tensor2, multiplies the result by the scalar value and adds it to input.
32addcmul_LinearAlgAlpha5.4aten, pointwise, KernelGenThe in-place version of addcmul().
33addcmul_outLinearAlgBeta5.3aten, pointwise, KernelGenA variant of addcmul that allows the output to be assigned to out.
34addmmBLASStable1.0atenPerforms a matrix multiplication of the matrices mat1 and mat2. The matrix input is added to the final result.
35addmm_BLASAlpha5.4aten, KernelGenThe in-place version of addmm. Performs the matrix multiplication of mat1 and mat2, multiplied by alpha, then adds beta times input to the result in-place.
36addmm_dtypeBLASBeta5.3atenA variant of addmm that allows the dtype of the output tensor to be specified. This is supported only on CUDA and for torch.float32 given torch.float16 or torch.bfloat16 input dtypes.
37addmm_dtype_outBLASBeta5.3atenA variant of addmm_dtype() that allows the output to be saved to the provided out parameter.
38addmm_outBLASStable4.0atenA variant of addmm that assigns to the output to the provided out parameter.
39addmvBLASStable4.0atenPerforms a matrix-vector product of the matrix mat and the vector vec. The vector input is added to the final result.
40addmv_BLASAlpha5.4aten, KernelGenPerforms matrix-vector product with accumulation in-place.
41addmv_outBLASStable4.0atenPerforms a matrix-vector product of the matrix mat and the vector vec. The vector input is added to the final result.
42addrBLASStable4.0atenPerforms the outer-product of vectors vec1 and vec2 and adds it to the matrix input.
43addr_BLASStable4.0aten, KernelGenIn-place version of addr. Performs the outer-product of vectors vec1 and vec2 and adds it to the matrix input in-place.
44affine_grid_generatorTensorAlpha5.3aten, KernelGen, pointwiseGenerates a 2D or 3D flow field (sampling grid), given a batch of affine matrices theta.
45aliasMathAlpha5.4aten, KernelGenReturn a view tensor sharing the same storage (zero-copy alias).
46alias_copyTensorStable5.3aten, KernelGenCreates a new tensor that shares the same storage data as the original tensor, but without preserving the original tensor's metadata (like shape or strides) in a way that links future mutations.
47alias_copy_outTensorStable5.3aten, KernelGenA variant of alias_copy() that assigns the output to the out tensor.
48allMathStable2.0aten, ReductionTests if all elements in input evaluate to True.
49all_dimMathStable2.0aten, ReductionFor each row of input in the given dimension dim, returns True if all elements in the row evaluate to True and False otherwise.
50all_dimsMathStable2.0aten, ReductionA variant of all.
51allcloseMathStable2.1atenThis function checks if input and other satisfy a condition specified via atol and rtol elementwise, for all elements of input and other.
52alpha_dropoutNeuralNetworkAlpha5.4aten, KernelGenApplies alpha dropout to the input. Alpha Dropout is a type of Dropout that maintains the self-normalizing property by scaling activations with SELU activation function parameters.
53alpha_dropout_NeuralNetworkAlpha5.4aten, KernelGenApplies alpha dropout to the input inplace. Alpha Dropout is a type of Dropout that maintains the self-normalizing property by scaling activations with SELU activation function parameters.
54amaxLinearAlgStable2.0aten, ReductionReturns the maximum value of each slice of the input tensor in the given dimension(s) dim.
55aminReductionAlpha5.4aten, KernelGenReturns the minimum value of each row of the input tensor in the given dimension dim.
56amin_ReductionAlpha5.4aten, KernelGenThe in-place version of amin, returns the minimum value of each row in the given dimension.
57aminmaxTensorBeta5.3atenComputes the minimum and maximum values of the input tensor.
58amp_foreach_non_finite_check_and_unscale_MathAlpha5.4aten, KernelGenChecks gradient tensors for non-finite values (inf or nan) and unscales each tensor in-place by inv_scale. If any non-finite value is detected, found_inf is set to 1.0.
59amp_update_scale_MathAlpha5.4aten, KernelGenUpdates the AMP loss scale in-place: when found_inf is non-zero the scale is multiplied by scale_backoff_factor and the growth tracker is reset to 0; otherwise the growth tracker is incremented and, once it reaches growth_interval, the scale is multiplied by scale_growth_factor and the tracker is reset to 0.
60and_scalarMathBeta5.4aten, KernelGenComputes the bitwise AND operation through the tensor dunder method with a scalar operand.
61and_tensorMathBeta5.4aten, KernelGenComputes the bitwise AND operation through the tensor dunder method with tensor operands.
62angleMathStable3.0aten, pointwiseComputes the element-wise angle (in radians) of the given input tensor.
63anyMathStable2.0aten, ReductionTests if any element in input evaluates to True.
64any_dimMathStable2.0aten, ReductionFor each row of input in the given dimension dim, returns True if any element in the row evaluate to True and False otherwise.
65any_dimsMathStable2.0aten, ReductionFor each row of input in the given dimensions in dims, returns True if any element in the row evaluate to True and False otherwise. The dims contains tuple of ints indicating the dimensions to reduce.
66apply_repetition_penaltiesNeuralNetworkStable5.0fused, vLLMModifies logit tensors in place to penalize tokens that have already appeared in the generated sequence.
67apply_rotary_pos_embNeuralNetworkStable2.0fusedA method to incorporate positional information into the Transformer architecture. Rotary Positional Embedding (RoPE) applies position-dependent rotation to the query (Q) and key (K) vectors before computing the attention score.
68arangeTensorStable2.1atenReturns a 1-D tensor of size ceiling((end−start)/step) with values from the interval [start, end) taken with common difference step beginning from start.
69arange_starttensorStable2.1atenA variant of arange, with start and/or step specified.
70arange_start_steptensorStable2.1atenA variant of arange, with start and/or step specified.
71arccosMathAlpha5.4aten, KernelGenReturns a new tensor with the arccosine of the elements of input.
72arccos_MathAlpha5.4aten, KernelGenThe in-place version of arccos().
73arccoshMathAlpha5.4aten, KernelGenReturns a new tensor with the inverse hyperbolic cosine of the elements of input.
74arccosh_MathAlpha5.4aten, KernelGenComputes the element-wise inverse hyperbolic cosine of a given input tensor. This is an in-place version.
75arccosh_outMathAlpha5.4aten, KernelGenA variant of arccosh that allows the output to be assigned to the out tensor.
76arcsinMathAlpha5.4aten, KernelGenReturns a new tensor with the arcsine of the elements of input.
77arcsin_MathAlpha5.4aten, KernelGenThe in-place version of arcsin().
78arcsin_outMathAlpha5.4aten, KernelGenA variant of arcsin that allows the output to be assigned to the out tensor.
79arcsinhMathStable5.4aten, KernelGenPerforms an element-wise inverse hyperbolic sine computation on the given tensor.
80arcsinh_MathStable5.4aten, KernelGenThe in-place version of arcsinh().
81arcsinh_outMathStable5.4aten, KernelGenA variant of arcsinh that allows the output to be assigned to the out tensor.
82arctanMathAlpha5.4aten, KernelGenReturns a new tensor with the element-wise inverse tangent (arctangent) of the input tensor.
83arctan2MathAlpha5.4aten, KernelGenComputes the element-wise arctangent of input / other. Alias for atan2.
84arctan2_MathAlpha5.4aten, KernelGenThe in-place version of arctan2(). Computes the element-wise arc tangent of input/other(y/x), returning angles in radians between -PI and PI and stores the result in input.
85arctan_MathAlpha5.4aten, KernelGenComputes the element-wise inverse tangent (arctangent) of a given input tensor. This is an in-place version.
86arctanhMathAlpha5.4aten, KernelGenComputes the element-wise inverse hyperbolic tangent of a given input tensor.
87arctanh_MathStable5.4aten, KernelGenComputes the element-wise inverse hyperbolic tangent of a given input tensor. This is an in-place version.
88arctanh_outMathAlpha5.4aten, KernelGenA variant of arctanh that assigns the output to the provided out parameter.
89argmaxLinearAlgStable2.0aten, ReductionReturns the indices of the maximum value of all elements in the input tensor.
90argminLinearAlgStable2.2aten, ReductionReturns the indices of the minimum value(s) of the flattened tensor or along a dimension.
91argsortTensorAlpha5.4skip_precision_check, KernelGenSorting/selection operator (argsort).
92as_strided_copyTensorBeta5.3aten, KernelGenCreates a contiguous copy of an as_strided view of the input tensor.
93as_strided_copy_outTensorBeta5.3aten, KernelGenA variant of as_strided_copy() that assigns the output to the out tensor.
94as_strided_scatterTensorAlpha5.4aten, KernelGenReturns a clone of the input storage with source values written through a view defined by the requested size, stride, and storage offset.
95asinMathAlpha5.4aten, KernelGenReturns a new tensor with the arcsine of the elements of input.
96asin_MathAlpha5.4aten, KernelGenThe in-place version of asin().
97asinhMathAlpha5.3aten, KernelGenReturns a new tensor with the inverse hyperbolic sine of the elements of input.
98asinh_MathStable5.3aten, KernelGenComputes the inverse hyperbolic sine for each element of a tensor in-place.
99assert_asyncTensorStable5.3utilityA utility used to perform data-dependent assertions on GPU tensors without triggering an immediate, performance-heavy GPU-to-CPU synchronization.
100atanMathStable4.0aten, pointwiseReturns a new tensor with the arctangent of the elements (in radians) in the input tensor.
101atan2MathStable5.3aten, pointwiseComputes the element-wise arc tangent of input/other(y/x), returning angles in radians between -PI and PI.
102atan2_MathAlpha5.4aten, KernelGen, pointwiseComputes the element-wise arctangent of input/other in-place.
103atan2_outMathBeta5.3aten, pointwiseA variant of atan2 that allows the output to be saved into out.
104atan_MathStable4.0aten, pointwiseThe in-place version of atan().
105atanhMathAlpha5.4aten, KernelGenTriton kernel implementation for atanh.
106atanh_MathAlpha5.4aten, KernelGenThe in-place version of atanh().
107avg_pool1dNeuralNetworkAlpha5.4aten, KernelGenApplies 1D average-pooling operation in kL regions by step size sL steps. The number of output features is equal to the number of input planes.
108avg_pool2dNeuralNetworkStable4.1nn.functionalApplies 2D average-pooling operation in kH \mul kW regions by step size sH \mul sW steps. The number of output features is equal to the number of input planes. This is for the forward case.
109avg_pool2d_backwardNeuralNetworkStable4.1atenThe backward version of avg_pool2d().
110avg_pool3dNeuralNetworkBeta5.3atenApplies 3D average-pooling operation in kD \times kH \times kW regions by step size sD \times sH \times sW steps.
111avg_pool3d_backwardNeuralNetworkAlpha5.3atenThis is the backward version of avg_pool3d().
112baddbmmBLASStable4.1atenPerforms a batch matrix-matrix product of matrices in batch1 and batch2. input is added to the final result. batch1 and batch2 must be 3-D tensors each containing the same number of matrices.
113baddbmm_BLASAlpha5.4aten, KernelGenPerforms batched matrix-matrix product with accumulation in-place.
114baddbmm_outBLASBeta5.3atenThis is a variant of baddbmm().
115batch_normNeuralNetworkStable3.0atenAn internal operator used for implementing the BatchNorm functionality.
116batch_norm_backwardNeuralNetworkStable3.0atenThe backward version of batch_norm().
117batch_norm_impl_indexNeuralNetworkAlpha5.4aten, KernelGenApplies Batch Normalization and returns the normalized output along with saved statistics, a reserve tensor, and the backend implementation index.
118batch_norm_impl_index_backwardNeuralNetworkAlpha5.4aten, KernelGenBackward pass dispatcher for _batch_norm_impl_index. Selects the matching batch-norm backward backend (native / cudnn / miopen) from the impl_index returned by the forward and produces the gradients with respect to the input, weight and bias.
119batch_norm_no_updateNeuralNetworkAlpha5.4aten, KernelGenApplies Batch Normalization using precomputed running statistics without updating running_mean/running_var.
120batch_norm_with_update_functionalNeuralNetworkAlpha5.4aten, KernelGenFunctional variant of _batch_norm_with_update that applies training-mode Batch Normalization and updates running statistics.
121beam_search_scoreMathAlpha5.4KernelGen, fusedComputes beam search scores by adding cumulative log probabilities with new token scores.
122beam_search_score_MathAlpha5.4KernelGen, fusedIn-place version of beam search score computation.
123bernoulliTensorAlpha5.4aten, KernelGenDraws binary random numbers (0 or 1) from a Bernoulli distribution with per-element probabilities.
124bernoulli_TensorBeta5.3aten, skip_precision_check, KernelGenDraws binary random numbers (0 or 1) from a Bernoulli distribution.
125bf16_paged_mqa_logitsNeuralNetworkAlpha5.4fused, KernelGenCompute multi-head weighted ReLU attention logits on paged BF16 KV cache. Uses shape-specialized Triton kernels for H=32 and H=64 with zero-constexpr dispatch for decode-phase inference.
126bilinearBLASAlpha5.4aten, KernelGenApplies a bilinear transformation to the incoming data: y = x1^T A x2 + b.
127binary_cross_entropyNeuralNetworkAlpha5.4aten, pointwise, nn.functional, KernelGenMeasures the binary cross entropy between the target and input probabilities, with optional per-element weighting.
128binary_cross_entropy_backwardMathAlpha5.4aten, KernelGenComputes the gradient of binary cross entropy loss with respect to the input, with optional per-element weight.
129binary_cross_entropy_outNeuralNetworkAlpha5.4aten, pointwise, nn.functional, KernelGenThe out variant of binary_cross_entropy. Measures the binary cross entropy between the target and input probabilities and writes the result to the provided output tensor.
130binary_cross_entropy_with_logitsNeuralNetworkAlpha5.4nn.functional, KernelGen, atenCompute the binary cross entropy loss with logits.
131bincountReductionStable5.0aten, pointwise, KernelGenCount the frequency of each value in an array of non-negative integers.
132binomialMathAlpha5.4aten, KernelGenDraws samples from a binomial distribution with the number of trials given by count and the success probability given by prob.
133binomial_outMathAlpha5.4aten, KernelGenA variant of binomial() that writes the result into the out tensor.
134bitwise_and_scalarMathStable2.0aten, pointwiseComputes the bitwise AND of input and other scalar.
135bitwise_and_scalar_MathStable2.2aten, pointwiseThe in-place, scalar version of bitwise_and().
136bitwise_and_scalar_tensorMathStable2.0aten, pointwiseA variant of bitwise_and().
137bitwise_and_tensorMathStable2.0aten, pointwiseThe Tensor method version of bitwise_and().
138bitwise_and_tensor_MathStable2.2aten, pointwiseThe in-place, Tensor method version of bitwise_and().
139bitwise_left_shiftMathStable4.0aten, pointwiseComputes the left arithmetic shift of input by other bits.
140bitwise_left_shift_MathAlpha5.4aten, KernelGenThe in-place version of bitwise_left_shift().
141bitwise_notMathStable2.0aten, pointwiseComputes the bitwise NOT of the given input tensor.
142bitwise_not_MathStable2.2aten, pointwiseThe in-place version of bitwise_not().
143bitwise_or_scalarMathStable2.0aten, pointwiseComputes the bitwise OR of scalars input and other.
144bitwise_or_scalar_MathStable2.2aten, pointwiseThe in-place version of bitwise_or_scalar.
145bitwise_or_scalar_tensorMathStable2.0aten, pointwiseComputes the bitwise OR of input and other.
146bitwise_or_tensorMathStable2.0aten, pointwiseComputes the bitwise OR of input and other, this is the Tensor method variant.
147bitwise_or_tensor_MathStable2.2aten, pointwiseThe in-place version of bitwise_or_tensor().
148bitwise_right_shiftMathStable4.0aten, pointwiseComputes the right arithmetic shift of input by other bits.
149bitwise_right_shift_MathAlpha5.4aten, KernelGen, pointwiseThe in-place version of bitwise_right_shift().
150bitwise_xor_scalarMathAlpha5.4aten, pointwise, KernelGenComputes the bitwise XOR of tensor input and scalar other.
151bitwise_xor_scalar_MathAlpha5.4aten, pointwise, KernelGenThe in-place version of bitwise_xor_scalar().
152bitwise_xor_scalar_tensorMathAlpha5.4aten, pointwise, KernelGenComputes the bitwise XOR of scalar input and tensor other.
153bitwise_xor_tensorMathAlpha5.4aten, pointwise, KernelGenComputes the element-wise bitwise XOR of tensor input and tensor other.
154bitwise_xor_tensor_MathAlpha5.4aten, pointwise, KernelGenThe in-place version of bitwise_xor_tensor().
155blackman_windowMathAlpha5.4aten, KernelGenComputes the Blackman window function.
156blackman_window_periodicMathAlpha5.4aten, KernelGenA variant of blackman_window() that takes an explicit periodic flag.
157block_diagLinearAlgAlpha5.4aten, KernelGenCreates a block diagonal matrix from provided tensors. Each input tensor forms a diagonal block in the output, with zeros elsewhere.
158bmmBLASStable1.0atenPerforms a batch matrix-matrix product of matrices stored in input and mat2.
159bmm_outBLASStable5.0atenPerforms a batch matrix-matrix product of matrices stored in input and mat2. This is a variant of bmm with out specified.
160bmm_w8a8_fp8BLASAlpha5.4NoCPU, QuantizationPerforms batched matrix multiplication with block-wise FP8 E4M3 quantized activations and weights using FP32 accumulation.
161broadcast_tensorsTensorAlpha5.4aten, KernelGenBroadcasts the given tensors to a common shape according to broadcasting semantics, returning a list of tensors.
162broadcast_toTensorAlpha5.4aten, KernelGenBroadcasts input to the shape size. Equivalent to calling input.expand(size).
163bucket_sort_topkNeuralNetworkBeta5.3fused, DSAA wrapper of the TLE version and the Triton version bucket-sort topk operation.
164bucketizeMathAlpha5.4aten, KernelGenReturns the indices of the buckets to which each input value belongs.
165catTensorStable2.2atenConcatenates the given sequence of tensors in tensors in the given dimension.
166cat_outTensorStable2.2atenA variant of cat that assigns the result to the provided out parameter.
167cauchyDistributionBeta5.3atenDraws random numbers from a Cauchy distribution.
168cauchy_DistributionBeta5.3atenFills the tensor with numbers drawn from the Cauchy distribution.
169cdistMathAlpha5.4aten, KernelGenComputes the p-norm distance between each pair of the two collections of row vectors. This is the canonical user-facing API for cdist computation.
170cdist_backwardMathAlpha5.4aten, KernelGenComputes the backward pass for the batched pairwise distance (cdist) computation, computing the gradient with respect to the input tensors.
171cdist_forwardMathAlpha5.4aten, KernelGenComputes the batched pairwise distance (cdist) between two collections of row vectors, supporting p=2.0 (L2 norm).
172ceilMathStable5.0aten, pointwiseReturns a new tensor with the ceil of the elements of input, the smallest integer greater than or equal to each element.
173ceil_MathStable5.0aten, pointwiseThe in-place version of ceil().
174ceil_outMathStable5.0aten, pointwiseA variant of ceil() with out specified.
175celuNeuralNetworkStable4.0aten, nn.functional, pointwiseApplies the quantized CELU (Continuously Differentiable Exponential Linear Unit) activation function element-wise.
176celu_NeuralNetworkStable4.0aten, nn.functional, pointwiseThe in-place version of celu().
177chalfMathAlpha5.4aten, KernelGenReturns a copy of the tensor cast to the complex32 dtype. Real inputs are down-cast to a complex tensor with zero imaginary part, and complex inputs have both their real and imaginary components cast to half precision.
178channel_shuffleNeuralNetworkAlpha5.4aten, KernelGenDivides the channels into groups and rearranges each group.
179cholesky_inverseMathAlpha5.4aten, KernelGenCompute inverse of symmetric positive-definite matrix from Cholesky factor.
180cholesky_solveBLASAlpha5.4atenSolves a system of linear equations with a symmetric positive-definite matrix using its Cholesky factorization.
181cholesky_solve_helperMathAlpha5.4aten, KernelGenSolves a positive-definite linear system using its lower- or upper-triangular Cholesky factor.
182cholesky_solve_outBLASAlpha5.4atenSolves a system of linear equations with a symmetric positive-definite matrix using its Cholesky factorization.
183choose_qparams_optimizedMathAlpha5.4aten, KernelGenChooses optimized quantization parameters (scale and zero-point) for a tensor by searching candidate value ranges and selecting the pair that minimizes the quantization error.
184chunkMathAlpha5.4aten, KernelGenSplit a tensor into a specific number of chunks along a given dimension.
185chunk_catMathAlpha5.4aten, KernelGenTriton kernel implementation for _chunk_cat.
186chunk_gated_delta_rule_fwdAttentionBeta5.3fused, FLAThe forward case for ChunkGatedDeltaRuleFunction with Flash Linear Attention (FLA).
187clampMathStable2.0aten, pointwiseClamps all elements in input into the range [min, max].
188clamp_MathStable2.2aten, pointwiseThe in-place version of clamp().
189clamp_maxMathAlpha5.4aten, KernelGen, pointwiseClamps all elements in input to be smaller or equal max.
190clamp_max_MathAlpha5.4aten, KernelGen, pointwiseThe in-place version of clamp_max().
191clamp_minMathStable4.0aten, pointwiseA variant of clamp() with min set to min.
192clamp_min_MathStable4.0aten, pointwiseThe in-place version of clamp_().
193clamp_tensorMathStable2.0aten, pointwiseThe tensor version of clamp().
194clamp_tensor_MathStable2.2aten, pointwiseThe in-place, tensor version of clamp().
195clipMathBeta5.3aten, KernelGenThis is identical to clamp().
196clip_MathBeta5.3aten, KernelGenThis is identical to clamp_().
197cloneTensorAlpha5.4KernelGen, aten, skip_precision_checkPure layout/memory operation (clone).
198coalesced_TensorAlpha5.4aten, KernelGenSets the coalesced flag of a sparse coordinate (COO) tensor in place and returns self. This is a metadata-only mutation: the underlying indices/values data is not modified, only the is_coalesced bit is toggled. The flag flip is dispatched through the native backend kernel (below the autograd key); a Triton identity kernel is provided over the values data as the per-element implementation.
199col2imMathAlpha5.3aten, KernelGenRearranges column blocks back into a multidimensional tensor (inverse of im2col).
200column_stackTensorAlpha5.4aten, KernelGenCreates a new tensor by horizontally stacking the tensors in tensors. Equivalent to torch.hstack after 0-D and 1-D tensors have been reshaped into (numel, 1) columns.
201column_stack_outTensorAlpha5.4aten, KernelGenA variant of column_stack() that writes the result into the out tensor.
202combine_topk_swa_indicesNeuralNetworkBeta5.3fused, Attention, vLLM, DeepSeekV4Combines compressed top-k sparse attention indices with sliding-window attention indices for DeepSeekV4 attention.
203compute_global_topk_indices_and_lensNeuralNetworkBeta5.3fused, Attention, vLLM, DeepSeekV4Converts local top-k sparse attention indices to global KV-cache indices and computes valid top-k lengths for DeepSeekV4 attention.
204compute_linear_combinationBLASAlpha5.4aten, KernelGenComputes a batched linear combination: given coefficients of shape [m, n] and input of shape [n, ...], returns an output of shape [m, ...] where output[i, ...] = sum_j coefficients[i, j] * input[j, ...] (equivalent to coefficients @ input.flatten(1)). Backs the string-padding form dispatched by torch._compute_linear_combination.
205compute_linear_combination_outBLASAlpha5.4aten, KernelGenA variant of _compute_linear_combination() that writes the result into the out tensor.
206concatTensorAlpha5.4aten, KernelGenAn alias of cat().
207concat_and_cache_mlaAttentionBeta3.0fused, MLAWrites the latent and RoPE value into KV cache for Multi-head Latent Attention forward case.
208concatenateTensorAlpha5.3aten, KernelGenAn alias of cat().
209conjMathAlpha5.4aten, KernelGenComputes the element-wise conjugate of a complex tensor. For non-complex tensors, returns the input unchanged.
210conj_copyMathAlpha5.4aten, KernelGenComputes the element-wise conjugate of a complex input tensor and returns it as a new (copied) tensor. For non-complex tensors this op is not supported.
211conj_copy_outMathAlpha5.4aten, KernelGenA variant of _conj_copy() that writes the result into the out tensor.
212conj_physicalLinearAlgBeta5.3atenComputes the element-wise conjugate of the given input tensor. If input has a non-complex dtype, this function just returns input.
213conj_physical_LinearAlgAlpha5.4atenIn-place version of conj_physical. Computes the element-wise conjugate of the given input tensor in place. If input has a non-complex dtype, this function just returns input.
214constant_pad_ndNeuralNetworkStable2.2aten, IRPads the input tensor boundaries with a constant value. This is an IR representation, not a public API.
215contiguousTensorRemoved4.1aten, skip_precision_checkReturns a contiguous in memory tensor containing the same data as self tensor.
216conv1dConvolutionStable4.2atenApplies a 1D convolution over a quantized 1D input composed of several input planes.
217conv1d_paddingConvolutionStable4.2atenApplies a 1D convolution over a quantized 1D input composed of several input planes.
218conv2dConvolutionStable4.2atenApplies a 2D convolution over a quantized 2D input composed of several input planes.
219conv2d_paddingConvolutionStable4.2atenApplies a 2D convolution over a quantized 2D input composed of several input planes.
220conv3dConvolutionStable4.2atenApplies a 3D convolution over a quantized 3D input composed of several input planes.
221conv3d_paddingConvolutionStable4.2atenApplies a 3D convolution over a quantized 3D input composed of several input planes.
222conv_depthwise2dNeuralNetworkStable5.3aten, Convolution, NoCPUA depthwise convolution for the conv2d neural network function.
223conv_tbc_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for conv_tbc_backward.
224conv_transpose1dConvolutionBeta5.3aten, KernelGenApplies a 1D transposed convolution operator over an input image composed of several input planes.
225conv_transpose2dConvolutionAlpha5.3aten, KernelGenApplies a 2D transposed convolution operator over an input image composed of several input planes.
226conv_transpose3dNeuralNetworkAlpha5.4aten, KernelGenApplies a 3D transposed convolution operator over an input volume composed of several input planes.
227convert_weight_to_int4packMathAlpha5.4aten, KernelGenConverts a weight tensor to int4 packed format.
228convolution_double_backwardConvolutionAlpha5.4aten, KernelGenSecond-order backward of a convolution, supporting both 1D (3-D operands [N, C, L]) and 2D (4-D operands [N, C, H, W]) cases. Given the gradients of the first-order gradients (ggI, ggW, ggb) and the output gradient (gO), returns the gradients of the scalar <gX, ggI> + <gW, ggW> + <gB, ggb> with respect to (gO, input, weight). The 2D path runs on Triton conv / conv-transpose kernels plus a custom Triton weight-gradient correlation kernel; 1D reuses the 2D path by unsqueezing the lone spatial axis (mirroring how conv1d delegates to conv2d). 3D (5-D operands) is not yet supported and raises NotImplementedError.
229convolution_modeConvolutionAlpha5.4aten, KernelGenLow-level dispatcher backing the string-padding form of convolution. Given an input, a weight, an optional bias, stride, a string padding mode (valid or same), dilation and groups, it computes the N-D convolution by dispatching to the FlagGems conv1d/conv2d/conv3d Triton kernels based on the spatial dimensionality of the weight tensor. same padding requires unit strides.
230copyTensorStable5.3aten, pointwiseAs a wrapper of copy_, this operator copies elements from src to out using given template for shapes.
231copy_TensorStable4.1aten, pointwise, skip_precision_checkCopies the elements from src into self tensor and returns self.
232copysignTensorBeta5.3aten, pointwiseCreate a new floating-point tensor with the magnitude of input and the sign of other, elementwise.
233copysign_MathAlpha5.3aten, KernelGenTriton kernel implementation for copysign_.
234copysign_outTensorBeta5.3aten, pointwiseA variant of copysign that allows the output to be saved into out.
235corrcoefReductionAlpha5.4aten, KernelGenEstimates the Pearson product-moment correlation coefficient matrix of the variables given by the input matrix, where rows are the variables and columns are the observations. The correlation matrix R is computed from the covariance matrix C as R_ij = C_ij / sqrt(C_ii * C_jj), with the internal computation carried out in float32 for numerical stability.
236cosMathStable2.0aten, pointwiseReturns a new tensor with the cosine of the elements of input given in radians.
237cos_MathStable2.2aten, pointwiseThe in-place version of cos().
238coshMathStable5.3aten, pointwiseReturns a new tensor with the hyperbolic cosine of the elements of input.
239cosh_MathStable5.3aten, pointwiseThis is the in-place version of cosh().
240cosh_outMathStable5.3aten, pointwiseThis is an variant of cosh() that assigns the output to the provided out.
241cosine_embedding_lossNeuralNetworkAlpha5.4aten, nn.functional, KernelGenCompute the cosine embedding loss.
242count_nonzeroTensorStable2.2aten, ReductionCounts the number of non-zero values in the tensor input along the given dim. If no dim is specified then all non-zeros in the tensor are counted.
243covReductionAlpha5.4aten, KernelGenEstimates the covariance matrix of the variables given by the input matrix, where rows are the variables and columns are the observations. Supports Bessel's correction and optional frequency/analytic observation weights, with the internal computation carried out in float32 for numerical stability.
244cp_gather_indexer_k_quant_cacheQuantizationBeta5.3fused, vLLMThis is a fused operator that gathers FP8 K cache values and scales.
245cross_entropy_lossNeuralNetworkRemoved3.0fused, ReductionComputes the cross entropy loss between input logits and target.
246ctc_lossNeuralNetworkAlpha5.4aten, KernelGenConnectionist Temporal Classification loss for sequence-to-sequence models.
247cudnn_attention_forwardNeuralNetworkBeta5.5aten, NoCPUForward kernel for cuDNN attention, computing scaled dot-product attention outputs and log-sum-exp statistics. Uses a Triton FlashAttention-2 kernel in BHSD layout.
248cudnn_batch_norm_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for cudnn_batch_norm_backward.
249cudnn_convolutionNeuralNetworkBeta5.3aten, KernelGenA wrapper for cuDNN convolution backend.
250cudnn_convolution_transposeMathAlpha5.4aten, KernelGenWrapper for torch.cudnn_convolution_transpose, providing autograd support and device-agnostic interface for cuDNN transposed convolution.
251cudnn_rnn_backwardMathAlpha5.4aten, KernelGenBackward pass of a single-layer unidirectional cuDNN LSTM. Reimplements the RNN forward in pure PyTorch and applies autograd to compute gradients, avoiding the opaque cuDNN reserve tensor.
252cummaxMathStable3.0aten, ReductionReturns a named tuple (values, indices) where values is the cumulative maximum of elements of input in the dimension dim. And indices is the index location of each maximum value found in the dimension dim.
253cummax_helperMathAlpha5.4aten, KernelGenTriton kernel implementation for _cummax_helper, the out-of-place helper used by aten::cummax / aten::cummax.out. It writes the cumulative maximum of elements of input along dim into the pre-allocated values tensor and the index location of each maximum value into the pre-allocated indices tensor.
254cummaxmin_backwardMathAlpha5.4aten, ReductionBackward pass shared by cummax and cummin. Scatter-adds the output gradient back to the input positions selected during the forward pass (given by indices), accumulating in float32 for numerical stability. Equivalent to grad_input.scatter_add_(dim, indices, grad_output).
255cumminMathStable2.2aten, ReductionReturns a named tuple (values, indices) where values is the cumulative minimum of elements of input in the dimension dim. And indices is the index location of each minimum value found in the dimension dim.
256cummin_helperMathAlpha5.4aten, KernelGenTriton kernel implementation for _cummin_helper, the out-of-place helper used by aten::cummin / aten::cummin.out. It writes the cumulative minimum of elements of input along dim into the pre-allocated values tensor and the index location of each minimum value into the pre-allocated indices tensor.
257cumprodMathBeta5.3aten, ReductionReturns the cumulative product of elements of input in the dimension dim.
258cumprod_MathBeta5.3aten, ReductionThis is the in-place version of cumprod().
259cumsumLinearAlgStable1.0aten
260cumsum_ReductionAlpha5.4aten, KernelGenIn-place version of cumsum.
261cumsum_outReductionStable3.0aten
262cumulative_trapezoidReductionAlpha5.4aten, KernelGenCumulatively integrate y using the composite trapezoidal rule with uniform spacing dx along the given dimension.
263cutlass_scaled_mmLinearAlgBeta5.0fused, vLLM
264deg2radMathAlpha5.4aten, KernelGenConverts angles from degrees to radians.
265deg2rad_MathAlpha5.4aten, KernelGenIn-place version of deg2rad.
266deg2rad_outMathAlpha5.4aten, KernelGenOut-of-place variant of deg2rad with pre-allocated output tensor.
267dequantizeMathAlpha5.4aten, KernelGenReturns an fp32 Tensor by dequantizing a quantized Tensor.
268dequantize_and_gather_k_cacheNeuralNetworkBeta5.3fused, Attention, vLLM, DeepSeekV4Dequantizes FP8 K-cache entries and gathers them into a BF16 tensor for DeepSeekV4 attention.
269dgegluNeuralNetworkStable5.0fused, TransformerGaussian Error Gated Linear Unit with GELU activation instead of sigmoid function. This is for the backward case.
270diagTensorStable2.2aten
  • If input is a vector (1-D tensor), then returns a 2-D square tensor with the elements of input as the diagonal.
  • If input is a matrix (2-D tensor), then returns a 1-D tensor with the diagonal elements of input.
271diag_embedTensorStable2.2aten, pointwiseCreates a tensor whose diagonals of certain 2D planes (specified by dim1 and dim2) are filled by input. To facilitate creating batched diagonal matrices, the 2D planes formed by the last two dimensions of the returned tensor are chosen by default.
272diagonal_backwardLinearAlgStable2.2aten, pointwiseA diagonal operation returns a partial view of input with the its diagonal elements with respect to dim1 and dim2 appended as a dimension at the end of the shape. This is the backward case for diagonal().
273diagonal_copyTensorAlpha5.4aten, KernelGenPerforms the same operation as torch.diagonal, but all output tensors are freshly created instead of aliasing the input.
274diagonal_scatterMathAlpha5.4aten, KernelGenScatter source values into the diagonal of a tensor with optional offset.
275diffMathBeta5.3aten, KernelGenComputes the n-th forward difference along the given dimension.
276digammaMathAlpha5.3aten, KernelGenComputes the digamma function, which is the logarithmic derivative of the Gamma function.
277digamma_MathStable5.3aten, KernelGenComputes the in-place digamma function, which is the logarithmic derivative of the Gamma function.
278dispatch_fused_moe_kernelMoEStable5.3fused, Activation, vLLMAccelerates neural network training by combining token routing (dispatch/all-to-all communication), expert computation (GEMM), and result aggregation into a single GPU kernel.
279distMathAlpha5.4aten, ReductionComputes the p-norm distance between two tensors.
280div_modeMathStable1.0aten, pointwiseDivides each element of the input by the corresponding element of other. An optional rounding_mode can be specified.
281div_mode_MathStable2.2aten, pointwiseThe in-place version of div_mode().
282div_outMathStable4.2atenThis is an variant of div() with an out argument.
283div_scalarMathStable2.1atenThis is the scalar version of div().
284div_scalar_MathStable2.1atenThis is the in-place version of div_scalar().
285div_tensorMathStable2.1aten, pointwiseDivides each element of the input input by the corresponding element of other. Note that torch.divide() is an alias of torch.div() and torch.true_divide() is an alias of torch.div() with rounding_mode=None.
286div_tensor_MathStable2.1atenThis is the in-place version of div_tensor().
287divideMathAlpha5.4aten, pointwise, KernelGenAn alias of div() for element-wise division.
288dotBLASStable3.0atenComputes the dot product of two 1D tensors.
289dregluNeuralNetworkBeta5.3fused, TransformerRectified Gated Linear Unit is a variant of GLU that uses ReLU instead of the sigmoid function for gating. This is the backward case.
290dropoutNeuralNetworkStable1.0aten, nn.functionalAn internal IR for implementing torch.nn.functional.dropout.
291dsplitTensorAlpha5.4aten, KernelGen, skip_precision_checkSplit a tensor along the third axis (depth-wise). Pure layout operation returning zero-copy views.
292dswigluNeuralNetworkBeta5.3fused, TransformerSwish-Gated Linear Unit, a variant of GLU with the Swish activation function. This is for the backward case.
293dunder_ior_scalarMathBeta5.3aten, KernelGenThe scalar version of dunder_ior_tensor.
294dunder_ior_tensorMathBeta5.3aten, KernelGenThe in-place version of bitwise or operation for tensor and scalar.
295dunder_or_scalarMathBeta5.3aten, KernelGenThe scalar version of dunder_or_tensor.
296dunder_or_tensorMathBeta5.3aten, KernelGenThe in-place version of bitwise or operation for tensor and scalar.
297dyn_quant_pack_4bit_weightTensorAlpha5.4aten, KernelGenPacks dynamic 4-bit quantized weights, scale/zero values, and optional bias into ATen's portable packed representation.
298efficient_attention_backwardNeuralNetworkBeta5.4aten, NoCPUBackward kernel for FlashAttention, computing gradients of queries, keys, values, and attention outputs efficiently.
299einsumReductionAlpha5.3aten, KernelGenSums the product of the elements of the input operands along dimensions specified using a notation based on the Einstein summation convention.
300eluNeuralNetworkStable2.2aten, nn.functional, pointwiseApply the Exponential Linear Unit (ELU) function element-wise.
301elu_NeuralNetworkStable4.0aten, pointwiseThe in-place version of elu().
302elu_backwardNeuralNetworkStable4.0aten, pointwiseThe backward version of elu().
303embeddingNeuralNetworkStable2.1aten, nn.functionalGenerate a simple lookup table that looks up embeddings in a fixed dictionary and size. Note that the parameter sequence differs from torch.nn.functional.embedding.
304embedding_backwardNeuralNetworkStable3.0aten, NoCPUThe backward version of embedding().
305embedding_bag_dense_backwardNeuralNetworkAlpha5.4aten, KernelGenComputes the gradient for the backward pass of embedding_bag in dense mode.
306embedding_bag_per_sample_weights_backwardNeuralNetworkAlpha5.4aten, KernelGenTriton kernel implementation for _embedding_bag_per_sample_weights_backward.
307embedding_dense_backwardNeuralNetworkStable5.0aten, NoCPUCalculates the gradient of the weight matrix for a dense embedding layer during backpropagation.
308embedding_renorm_MathAlpha5.4aten, KernelGenIn-place renormalization of the embedding rows selected by indices: each indexed row whose norm_type-norm exceeds max_norm is scaled down so its norm equals max_norm.
309emptyTensorAlpha5.4aten, KernelGen, skip_precision_checkTensor factory operator — verifies dtype inference for newly allocated tensors.
310empty_permutedTensorAlpha5.4aten, KernelGen, skip_precision_checkTensor factory operator — allocates an uninitialized tensor whose memory follows the given physical layout.
311eqMathStable2.0aten, pointwiseComputes element-wise equality.
312eq_MathAlpha5.4aten, KernelGenThe in-place version of eq().
313eq_scalarMathStable2.0aten, pointwiseComputes equality between scalars.
314eq_scalar_MathAlpha5.4aten, KernelGenThe scalar version of eq_().
315equalMathStable5.0aten, ReductionReturns True if two tensors have the same size and elements, False otherwise.
316erfScienceStable2.1aten, pointwiseComputes the error function of input.
317erf_ScienceStable2.2aten, pointwiseThe in-place version of erf().
318erfcMathAlpha5.4aten, pointwise, KernelGenComputes the complementary error function.
319erfc_MathAlpha5.4aten, pointwise, KernelGenIn-place version of erfc.
320erfinvMathAlpha5.4aten, KernelGenComputes the inverse error function of input.
321erfinv_MathAlpha5.4aten, KernelGenThe in-place version of erfinv().
322euclidean_distMathAlpha5.3aten, KernelGen, pointwiseComputes pairwise Euclidean distances between rows of two 2D tensors.
323expMathStable1.0aten, pointwiseReturns a new tensor with the exponential of the elements of the input tensor input.
324exp2MathStable4.0aten, pointwiseComputes the base two exponential function of input.
325exp2_MathStable4.0aten, pointwiseThe in-place version of exp2().
326exp_MathStable2.2aten, pointwiseThe in-place version of exp().
327exp_outMathStable4.1aten, pointwiseA variant of exp2(), with out specified.
328expandTensorAlpha5.4aten, KernelGen, skip_precision_checkPure layout operation (expand).
329expand_TensorAlpha5.4aten, KernelGen, skip_precision_checkPure layout operation (expand, in-place).
330expand_asMathAlpha5.4aten, KernelGenExpand tensor to the same size as another tensor (view operation).
331expand_copyMathAlpha5.4aten, KernelGenReturns a copy of the input tensor expanded by broadcasting singleton dimensions.
332expm1MathBeta5.3atenComputes the exponential of the elements minus 1 of input.
333expm1_MathBeta5.3atenThe inplace version of expm1.
334expm1_outMathBeta5.3atenA variant of expm1 that saves the output to the specified out.
335exponentialDistributionAlpha5.4aten, skip_precision_checkReturns a new tensor with elements drawn from an exponential PDF (probability density function).
336exponential_DistributionStable2.1aten, skip_precision_checkFills self tensor with elements drawn from a PDF (probability density function).
337eyeLinearAlgStable3.0aten, ReductionReturns a 2-D tensor with ones on the diagonal and zeros elsewhere.
338eye_mLinearAlgStable3.0aten, ReductionTriton-based implementation of torch.eye_m(n, m), using 2D tiles to split the matrix into blocks.
339fake_quantize_learnable_per_channel_affine_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for _fake_quantize_learnable_per_channel_affine_backward, the backward of learnable per-channel fake quantization. Computes the gradients with respect to the input, scale and zero-point, accounting for the saturation branch where the quantized value is clamped to [quant_min, quant_max].
340fake_quantize_learnable_per_tensor_affineMathAlpha5.4aten, KernelGen, pointwiseFake-quantizes the input tensor per-tensor with learnable scale and zero_point.
341fake_quantize_learnable_per_tensor_affine_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for the backward pass of learnable per-tensor affine fake quantization. Computes gradients with respect to the input, scale, and zero-point.
342fake_quantize_per_channel_affineQuantizationAlpha5.4aten, pointwise, KernelGenApplies fake quantization per channel with affine parameters (scale and zero_point).
343fake_quantize_per_channel_affine_cachemaskQuantizationAlpha5.4aten, pointwise, KernelGenApplies per-channel affine fake quantization and returns the quantization-range mask.
344fake_quantize_per_channel_affine_cachemask_backwardQuantizationAlpha5.4aten, pointwise, KernelGenApplies the cache mask to the gradient of per-channel affine fake quantization.
345fake_quantize_per_channel_affine_cachemask_outQuantizationAlpha5.4aten, pointwise, KernelGenA variant of fake_quantize_per_channel_affine_cachemask that assigns the results to the provided output tensors.
346fake_quantize_per_tensor_affineQuantizationAlpha5.4aten, pointwise, KernelGenApplies per-tensor affine fake quantization using shared scale and zero-point parameters.
347fake_quantize_per_tensor_affine_cachemask_backwardMathAlpha5.4aten, pointwise, KernelGenComputes the gradient for fake_quantize_per_tensor_affine_cachemask by multiplying the incoming gradient by the cached boolean mask.
348feature_dropoutNeuralNetworkAlpha5.3aten, KernelGenApplies feature dropout to the input tensor. Randomly zeroes out entire channels of the input tensor with probability p. Each batch element has its own independent channel mask.
349feature_dropout_NeuralNetworkAlpha5.3aten, KernelGenThe in-place version of feature_dropout().
350fft_irfftnMathAlpha5.4aten, KernelGenComputes the N-dimensional inverse of torch.fft.rfftn.
351fill_diagonal_TensorAlpha5.4aten, pointwise, KernelGenFills the main diagonal of a tensor that has at least two dimensions with the specified scalar value, in-place.
352fill_mem_eff_dropout_mask_NeuralNetworkAlpha5.4aten, KernelGenTriton kernel implementation for _fill_mem_eff_dropout_mask_, the in-place helper used by the memory efficient attention dropout path. It fills a contiguous 4D float32 tensor of shape (batch, heads, queries, keys) with random uniform values in [0, 1) drawn from a Philox4x32-10 stream identified by seed and offset.
353fill_scalarTensorStable2.2aten, pointwiseFills a scalar with the specified value.
354fill_scalar_TensorStable2.2aten, pointwiseThe in-place version of fill_scalar().
355fill_scalar_outTensorStable5.0aten, pointwise, KernelGenA variant of fill_scalar() that assigns the output to an out tensor.
356fill_tensorTensorStable2.2aten, pointwiseFills a tensor with the specified value.
357fill_tensor_TensorStable2.2aten, pointwiseThe in-place version of fill_tensor().
358fill_tensor_outTensorStable5.0aten, pointwise, KernelGenA variant of fill_tensor() that assigns the output to an out tensor.
359fixMathAlpha5.4aten, KernelGenfix operator
360fix_MathAlpha5.4aten, KernelGenIn-place version of fix. Truncates each element toward zero (floor for non-negative values, ceil for negative values). Integer tensors are left unchanged.
361flash_attention_backwardNeuralNetworkBeta5.4aten, NoCPUBackward kernel for FlashAttention, computing gradients of queries, keys, values, and attention outputs efficiently.
362flash_attention_forwardNeuralNetworkStable3.0aten, NoCPU
363flash_attn_varlen_funcNeuralNetworkStable3.1aten, Attention, FlashAttentionCompute attention for sequences of variable lengths within a single batch. Eliminating the need for padding.
364flash_attn_varlen_opt_funcNeuralNetworkBeta5.3aten, Attention, FlashAttentionA variant of flash_attn_varlen_func that has lse as an optional parameter.
365flash_mlaNeuralNetworkStable3.0fused, Attention, vLLMA variant of Multi-head Latent Attention (MLA).
366flash_mla_sparse_fwdNeuralNetworkAlpha5.3fused, Attention, vLLMPart of the FlashMLA.
367flattenMathAlpha5.4aten, KernelGenFlatten a contiguous range of dimensions into a single dimension (view operation).
368flipTensorStable2.1aten, pointwiseReverse the order of an n-D tensor along given axis in dims.
369fliplrTensorAlpha5.4aten, pointwise, KernelGenFlips a tensor in the left-to-right direction by reversing dimension 1. The input must have at least two dimensions.
370flipudTensorAlpha5.4aten, pointwise, KernelGenFlips a tensor in the up-to-down direction by reversing dimension 0. The input must have at least one dimension.
371float_power_MathAlpha5.4aten, KernelGenTriton kernel implementation for float_power_.
372float_power_scalar_tensorMathAlpha5.4aten, KernelGenComputes a scalar base raised element-wise to tensor exponents in float64.
373float_power_scalar_tensor_outMathAlpha5.4aten, KernelGenOut variant of float_power.Scalar.
374float_power_tensor_scalarMathAlpha5.4aten, KernelGenRaises tensor elements to a scalar exponent in float64.
375float_power_tensor_scalar_outMathAlpha5.4aten, KernelGenOut variant of float_power.Tensor_Scalar.
376float_power_tensor_tensorMathAlpha5.4aten, KernelGenRaises tensor elements to tensor exponents in float64 with broadcasting.
377float_power_tensor_tensor_outMathAlpha5.4aten, KernelGenOut variant of float_power.Tensor_Tensor.
378floorMathStable5.3aten, KernelGenPerforms an element-wise floor operation, rounding each element of a tensor down to the nearest integer less than or equal to itself.
379floor_MathStable5.3aten, KernelGenPerforms an in-place element-wise floor operation, rounding each element of a tensor down to the nearest integer less than or equal to itself.
380floor_divide_scalarMathStable2.1atenComputes input divided by other, elementwise, and floors the result.
381floor_divide_scalar_MathStable2.2atenComputes input divided by other, elementwise, and floors the result.
382floor_divide_tensorMathStable2.1atenComputes input divided by other, elementwise, and floors the result.
383floor_divide_tensor_MathStable2.2atenComputes input divided by other, elementwise, and floors the result.
384floor_outMathStable5.3aten, KernelGenPerforms an element-wise floor operation with output tensor, rounding each element down to the nearest integer less than or equal to itself.
385fmaxMathAlpha5.4aten, KernelGenComputes the element-wise maximum of input and other, ignoring NaNs.
386fmax_outMathAlpha5.4aten, KernelGenComputes the element-wise maximum of input and other, ignoring NaNs (out variant).
387fminMathStable5.3aten, KernelGenComputes the element-wise minimum of two tensors, specially handling NaN values by prioritizing the numerical value. Unlike minimum(), if one input is NaN and the other is a number, fmin() returns the number. It supports broadcasting, type promotion, and operates on both CPU and GPU.
388fmin_outMathStable5.3aten, KernelGenA variant of fmin() that assigns the output to the out tensor.
389fmod_MathAlpha5.4aten, KernelGenIn-place remainder of division (fmod_), computes element-wise remainder with truncation toward zero.
390fmod_scalarMathAlpha5.3aten, KernelGenComputes the element-wise remainder of division of input by a scalar divisor.
391fmod_scalar_MathAlpha5.3aten, KernelGenIn-place version of fmod with a scalar divisor.
392fmod_tensorMathAlpha5.3aten, KernelGenComputes the element-wise remainder of division of input by a tensor divisor.
393fmod_tensor_MathAlpha5.3aten, KernelGenIn-place version of fmod with a tensor divisor.
394fp8_fp4_mega_moeMoE NeuralNetworkAlpha5.4fused, vLLM, TritonFunctional Triton fallback for local FP8 x FP4 MegaMoE. It consumes staged FP8 activations, packed FP4 expert weights, scales, and top-k routing tensors, then computes the two-layer SwiGLU MoE output.
395fp8_fp4_mqa_logitsNeuralNetworkAlpha5.1fused, vLLM, KernelGenCompute weighted MQA logits with FP8 quantized Q and K tensors. Uses head-batched tiled dot products with K reuse for high throughput on DeepSeek V4 sparse attention indexer workloads.
396fp8_fp4_paged_mqa_logitsNeuralNetworkBeta5.3fused, vLLM, DeepSeekV4, KernelGenCompute paged multi-query attention logits from FP8 queries against FP8/FP4 paged KV cache with per-token scaling. Used in DeepSeek-V4 decode-phase inference with adaptive tile sizing.
397fp8_mqa_logitsNeuralNetworkBeta5.3fused, vLLMFor each token in the given E4M3 tensor, iterate all tokens from two other given tensors, calculate the logit.
398fracMathAlpha5.4aten, KernelGenComputes the fractional part of each element, returning x - trunc(x).
399frac_MathAlpha5.4aten, KernelGenIn-place fractional part computation, returning x - trunc(x).
400fractional_max_pool2dMathAlpha5.4aten, KernelGenApplies 2D fractional max pooling over an input signal composed of several input planes.
401fractional_max_pool2d_backwardMathAlpha5.4aten, KernelGenComputes the gradient of fractional_max_pool2d.
402frexpMathAlpha5.4aten, KernelGenReturn mantissa and exponent of input, such that x = mantissa * 2^exponent, with mantissa in (-1, 1).
403fullTensorStable2.1aten, pointwise, skip_precision_checkCreates a tensor of size size filled with fill_value. The tensor's dtype is inferred from fill_value.
404full_likeTensorStable2.1aten, pointwiseReturns a tensor with the same size as input filled with fill_value.
405functional_assert_asyncTensorAlpha5.4utility, KernelGenFunctional version of _assert_async that takes a dependency token and returns a new token. Part of PyTorch's functionalization framework for tracking operation dependencies.
406functional_sym_constrain_rangeTensorAlpha5.4aten, KernelGenA low-level function used in symbolic shape analysis to restrict the possible numerical range (min/max) of a symbolic integer based on tensor input arguments.
407functional_sym_constrain_range_for_sizeTensorBeta5.0aten, KernelGenA low-level function used in symbolic shape analysis to restrict the possible numerical range (min/max) of an unbacked symbolic integer.
408fused_adamMathAlpha5.4aten, KernelGenFused Adam optimizer step with optional AdamW weight decay, AMSGrad, and gradient maximize support.
409fused_adam_MathAlpha5.4aten, KernelGenThe in-place version of fused_adam().
410fused_add_rms_normNeuralNetworkStable2.0fused, Normalization
411fused_deepseek_v4_qnorm_rope_kv_rope_insertNeuralNetworkAlpha5.4fused, KernelGenHorizontally-fused DeepseekV4-MLA BF16 variant. per-head RMSNorm + GPT-J RoPE for Q, and GPT-J RoPE + bf16 paged cache insert for KV, all in one kernel launch. No FP8 quantization.
412fused_deepseek_v4_qnorm_rope_kv_rope_quant_insertNeuralNetworkBeta5.3fused, vLLM, DeepSeekV4Horizontally-fused DeepseekV4-MLA. per-head RMSNorm + GPT-J RoPE for Q, and GPT-J RoPE + UE8M0 FP8 quant + paged cache insert for KV, all in one kernel launch.
413fused_experts_implNeuralNetworkBeta5.3fused, vLLM, MoEAn implementation of fused MoE.
414fused_indexer_q_rope_quantNeuralNetworkBeta5.4fused, Attention, vLLM, DeepSeekV4Applies RoPE to sparse indexer Q, quantizes it to FP8 or MXFP4, and folds indexer weights for DeepSeekV4 attention.
415fused_moeNeuralNetworkBeta5.3fused, vLLM, MoEThe generic interface for fused MoE.
416fused_moving_avg_obs_fq_helperTensorAlpha5.4aten, quantization, KernelGenFused moving-average observer with fake-quantize for QAT; updates running min/max, recomputes scale/zero_point, and fake-quantizes the input for both per-tensor and per-channel modes.
417fused_q_kv_rmsnormNeuralNetworkBeta5.3fused, Attention, vLLM, DeepSeekV4Applies RMSNorm to Q and KV tensors in a single fused kernel for DeepSeekV4 attention.
418fused_recurrent_gated_delta_rule_fwdAttentionBeta5.3fused, FLAThe forward case for fused_recurrent_gated_delta_rule used in Flash Linear Attention (FLA).
419fused_rms_normNeuralNetworkAlpha5.4aten, KernelGenFused RMS normalization returning the normalized output and inverse RMS.
420fused_rms_norm_backwardNeuralNetworkAlpha5.4aten, KernelGenComputes input and weight gradients for fused RMS normalization using the saved reciprocal root-mean-square values.
421gatherTensorStable2.2aten, ReductionGathers values along an axis specified by dim.
422gather_backwardTensorStable2.2aten, ReductionThe backward version of gather().
423gather_block_quantizedMathAlpha5.4aten, KernelGenTriton kernel implementation for gather_block_quantized.
424gcdMathBeta5.3atenComputes the element-wise greatest common divisor (GCD) of input and other.
425gcd_MathAlpha5.4aten, KernelGenComputes the element-wise greatest common divisor (GCD) of input and other in-place.
426gcd_outMathBeta5.3atenA variant of gcd() that allows the output to be assigned to the specified out.
427geMathStable2.0aten, pointwiseComputes input is greater or equal to other element-wise.
428ge_scalarMathStable2.0aten, pointwiseThe scalar version of ge().
429gegluNeuralNetworkBeta5.3fused, Activation, TransformerGaussian Error Gated Linear Unit with GELU activation instead of sigmoid function.
430geluNeuralNetworkStable1.0aten, pointwise, Activation, nn.functionalApply Cumulative Distribution Function for Gaussian Distribution function element-wise.
431gelu_NeuralNetworkStable2.2aten, Activation, pointwiseThe in-place version of gelu().
432gelu_and_mulNeuralNetworkStable2.0fused, pointwise, ActivationAn activation function for GeGLU.
433gelu_backwardNeuralNetworkStable3.0aten, Activation, pointwiseThe backward version of gelu().
434geometricDistributionBeta5.1atenDraws random numbers from a geometric distribution.
435geometric_DistributionBeta5.1atenFills self tensor with elements drawn from the geometric distribution.
436get_paged_mqa_logits_metadataNeuralNetworkBeta5.3vLLMBuild scheduling metadata for paged MQA logits.
437get_scheduler_metadataAttentionStable4.0NoCPU, vLLMComputes scheduling metadata for attention work partitioning so that CPU computations can be routed to ISA-specific kernel implmentations. The metadata is stored in a tensor.
438gluNeuralNetworkStable3.0aten, Activation, pointwiseGated Linear Unit activation for modulating the output of a linear transformation with a gate.
439glu_backwardNeuralNetworkStable4.0aten, Activation, pointwiseThe backward version of glu().
440greaterMathStable5.3atenTest if input is greater than other elementwise.
441greater_equal_MathAlpha5.4aten, pointwise, KernelGenThe in-place version of greater_equal(), which is an alias for ge().
442greater_outMathStable5.3atenA variant of greater that saves the output to the specified out.
443greater_scalarMathStable5.3atenA variant of greater for scalar variables.
444greater_scalar_outMathStable5.3atenA variant of greater_out that saves the output to the specified out.
445grid_sampleNeuralNetworkAlpha5.3aten, nn.functionalGiven an input and a flow-field grid, computes the output using input values and pixel locations from grid.
446grid_sampler_3dNeuralNetworkAlpha5.4aten, KernelGenGiven an input and a flow-field grid in 3D, computes the output using input values and pixel locations from grid with trilinear or nearest interpolation.
447grid_sampler_3d_backwardMathAlpha5.4aten, KernelGenBackward pass for 3D grid sampling with bilinear/nearest interpolation and multiple padding modes.
448group_normNeuralNetworkStable2.0aten, ReductionAn internal IR for applying Group Normalization for last certain number of dimensions.
449group_norm_backwardNeuralNetworkStable3.0aten, ReductionThe backward case for group_norm().
450grouped_mmBLASBeta5.3atenGrouped matrix multiply is a functional operator designed to accelerate Mixture-of-Experts (MoE) models by computing multiple matrix multiplications in a single kernel launch.
451grouped_topkMoEStable5.0fused, NoCPU, vLLMA specialized routing mechanism used in Mixture-of-Experts (MoE) models (like DeepSeek-V3/R1) to select top-k experts by first grouping them, rather than selecting globally.
452gruNeuralNetworkBeta5.4atenGated recurrent unit (GRU). At each time step, computes a new hidden state from the input and the previous hidden state using a reset gate and an update gate, across one or more layers and optionally bidirectionally.
453gru_dataNeuralNetworkBeta5.4atenGated recurrent unit (GRU) over packed-sequence data.
454gtMathStable2.0aten, pointwiseComputes that input is greater than other element-wise.
455gt_scalarMathStable2.0aten, pointwiseThe scalar version of gt().
456gt_scalar_MathAlpha5.4aten, pointwise, KernelGenIn-place version of gt for scalar comparison, computes element-wise greater-than with a scalar storing result in input.
457gt_tensor_MathAlpha5.4aten, pointwise, KernelGenIn-place version of gt for tensor comparison, computes element-wise greater-than with a tensor storing result in input.
458hardshrinkNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenApplies the hard shrinkage function element-wise: returns x if |x| > lambd, else 0.
459hardshrink_outNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenOut-variant of hardshrink. Writes the element-wise result into the out tensor.
460hardsigmoidNeuralNetworkBeta5.0aten, pointwise, nn.functional, Activation, KernelGenAn activation function that provides a piecewise linear approximation of the standard sigmoid function, mapping inputs to a range between 0 and 1.
461hardsigmoid_NeuralNetworkAlpha5.4aten, pointwise, KernelGen, ActivationAn in-place variant of hardsigmoid, an activation function that provides a piecewise linear approximation of the standard sigmoid function, mapping inputs to a range between 0 and 1.
462hardsigmoid_backwardMathAlpha5.4aten, KernelGenThe backward pass for hardsigmoid. Computes the gradient of the hardsigmoid activation function: grad_output * (|self| < 3) / 6.
463hardsigmoid_outNeuralNetworkBeta5.0aten, pointwise, nn.functional, Activation, KernelGenA variant of hardsigmoid that supports an output tensor to receive the result.
464hardswishNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenApplies the Hard Swish activation function, commonly used in models like MobileNetV3 to improve accuracy while reducing computational cost compared to traditional Swish.
465hardswish_NeuralNetworkBeta5.0aten, pointwise, KernelGen, ActivationApplies the Hard Swish activation function, commonly used in models like MobileNetV3 to improve accuracy while reducing computational cost compared to traditional Swish. This is an in-place version.
466hardswish_backwardMathAlpha5.4aten, KernelGenThe backward pass for hardswish. Computes the gradient of the hardswish activation function: piecewise derivative 0 for x <= -3, (2x + 3) / 6 for -3 < x < 3, and 1 for x >= 3.
467hardswish_outNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenA variant of hardswish that supports an output tensor to receive the result.
468hardtanhNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenAn activation function that clips input values to lie within a fixed range [min_val, max_val], leaving values inside the range unchanged.
469hardtanh_NeuralNetworkAlpha5.4aten, pointwise, Activation, KernelGenAn activation function that clips input values to lie within a fixed range [min_val, max_val], leaving values inside the range unchanged. This is an in-place version.
470hardtanh_backwardMathAlpha5.4aten, KernelGenThe backward pass for hardtanh activation function. Computes the gradient of hardtanh: grad_output when min_val < x < max_val, 0 otherwise.
471hardtanh_outNeuralNetworkAlpha5.4aten, pointwise, nn.functional, Activation, KernelGenA variant of hardtanh that supports an output tensor to receive the result.
472has_compatible_shallow_copy_typeTensorAlpha5.4atenMetadata-only check that returns True when self can shallow-copy the TensorImpl type of from. Compatibility is decided on the tensors' DispatchKeySets: the sets are equal, or both are dense, both are sparse COO, or both are sparse compressed. It is independent of dtype and shape, and of device within a family, but opaque impls (meta, MKL-DNN, nested, quantized) only match an identical key set even though some of them report a strided layout.
473hc_head_fused_kernelNeuralNetworkStable5.3fused, vLLM, DSAThe head fusion kernel for MHC (Manifold-Constrained Hyper-Connections). This fused implementation computes RMS-normalized hidden states and applies per-head weighted mixing to produce the output activations.
474hc_split_sinkhorn_forwardNeuralNetworkBeta5.3fusedComputes a differentiable approximation of the Wasserstein distance (Optimal Transport) between two probability distributions or point clouds.
475heavisideMathAlpha5.4aten, KernelGenThe heaviside step function. Computes 0 where input < 0, 1 where input > 0, and values where input == 0. Supports tensor values.
476heaviside_MathAlpha5.4aten, KernelGenIn-place version of heaviside. For each element: 0 if x < 0, 1 if x > 0, and the corresponding values element if x == 0.
477histcMathAlpha5.3aten, KernelGenComputes the histogram of a tensor, binning each element into equal-width bins.
478hsplitTensorAlpha5.4aten, KernelGenSplit a tensor into multiple sub-tensors horizontally (column-wise). Equivalent to tensor_split with dim=1 for 2D+ tensors, or dim=0 for 1D tensors.
479hstackTensorStable2.2atenStack tensors in sequence horizontally (column wise). This is equivalent to concatenation along the first axis for 1-D tensors, and along the second axis for all other tensors.
480huber_lossNeuralNetworkAlpha5.4aten, KernelGen, LossComputes the Huber loss between input and target. Quadratic for small errors (|x - y| <= delta) and linear for large errors. Supports reduction none/mean/sum.
481huber_loss_outNeuralNetworkAlpha5.4aten, KernelGen, LossOut-variant of huber_loss. Writes the computed loss into the out tensor.
482hypotMathStable5.3aten, KernelGenGiven the legs of a right triangle, return its hypotenuse. The shapes of both input tensors must be broadcastable.
483hypot_MathAlpha5.4aten, KernelGenIn-place version of hypot. Given the legs of a right triangle, writes its hypotenuse into self. The shape of other must be broadcastable to self.
484hypot_outMathStable5.3aten, KernelGenGiven the legs of a right triangle, return its hypotenuse. The shapes of both input tensors must be broadcastable. This is a variant of hypot that allows the output to be a different tensor.
485i0MathStable5.3aten, KernelGenComputes the modified Bessel function of the first kind of order zero element-wise for a given input tensor.
486i0_MathStable5.3aten, KernelGenThe inplace version of i0.
487i0_outMathStable5.3aten, KernelGenA variant of i0 that assigns the output to the out tensor.
488igammaMathAlpha5.4aten, KernelGenComputes the regularized lower incomplete gamma function element-wise.
489igamma_MathAlpha5.4aten, KernelGenIn-place computation of the regularized lower incomplete gamma function.
490igammacMathAlpha5.4atenComputes the regularized upper incomplete gamma function Q(a, x) element-wise.
491igammac_MathAlpha5.4aten, KernelGen, pointwiseIn-place version of igammac. Computes the regularized upper incomplete gamma function complement.
492igammac_outMathAlpha5.4atenOut-of-place variant of igammac that writes the result to the out tensor.
493ilshiftMathAlpha5.4aten, KernelGenComputes the in-place bitwise left shift of self by other bits.
494im2colMathAlpha5.4aten, KernelGenExtracts sliding local blocks from a batched input tensor.
495indexReductionStable4.2atenExtract, access or modify specific elements, slices, or subsets of data within a tensor. The location of data is specified for each dimension, starting from index 0.
496index_addTensorStable2.2atenAccumulate the elements of alpha times source into the input tensor by adding to the indices in the order given in index.
497index_add_TensorStable4.0atenThe in-place version of index_add().
498index_copyTensorBeta5.3aten, KernelGenCopies the elements from source into input at the positions specified by index along the given dim.
499index_copy_TensorBeta5.3aten, KernelGenThe in-place version of index_copy().
500index_fillTensorAlpha5.4atenFills elements of the input tensor with a scalar or 0-dimensional tensor value at positions selected by index along the given dim.
501index_fill_TensorAlpha5.4atenThe in-place version of index_fill().
502index_putTensorStable2.2atenPuts values from the tensor values into the tensor input using the indices specified in indices (which is a tuple of Tensors).
503index_put_TensorStable3.0atenThe in-place version of index_put().
504index_put_impl_TensorBeta5.3atenAn internal C++ function that handles the heavy lifting for placing values into a tensor at specific indices.
505index_reduce_TensorBeta5.4aten, KernelGenAccumulates the elements of source into input at the indices specified by index along dim, using prod, mean, amax, or amin reduction.
506index_selectTensorStable2.1atenReturns a new tensor which indexes the input tensor along dimension dim using the entries in index.
507index_select_backwardMathAlpha5.4aten, KernelGenBackward of index_select. Accumulates gradients into the output at positions specified by the index.
508indexer_k_quant_and_cacheQuantizationBeta5.3fused, vLLMThis is a fused operator that quantizes K tensors and writes them into the FP8 KV cache.
509inplace_fused_expertsMoEStable5.3fused, Activation, vLLMThis operator writes output directly to hidden_states.
510instance_normNeuralNetworkBeta5.3fusedApply Instance Normalization independently for each channel in every data sample within a batch.
511irshiftMathAlpha5.4aten, KernelGenTriton kernel implementation for irshift.
512is_all_trueTensorBeta5.3aten, pointwise, KernelGenThe low-level implementation for checking if all elements in a tensor are True.
513is_nonzeroMathAlpha5.4aten, KernelGenReturns True if the single-element input tensor is not equal to zero.
514iscloseMathStable2.1aten, pointwiseReturns a new tensor with boolean elements representing if each element of input is "close" to the corresponding element of other. The closeness is defined with rtol and atol.
515isfiniteMathStable2.1aten, pointwiseReturns a new tensor with boolean elements representing if each element is finite or not.
516isinTensorStable2.2atenTests if each element of elements is in test_elements. Returns a boolean tensor of the same shape as elements that is True for elements in test_elements and False otherwise.
517isin_scalar_tensorTensorStable2.2atenA variant of isin().
518isin_tensor_scalarTensorStable2.2atenA variant of isin().
519isinfMathStable2.0aten, pointwiseTests if each element of input is infinite (positive or negative infinity) or not.
520isnanMathStable2.0aten, pointwiseReturns a new tensor with boolean elements representing if each element of input is NaN or not.
521isneginfMathStable5.3aten, KernelGen, pointwiseTests if each element of input is negative infinity or not.
522isneginf_outMathStable5.3aten, KernelGen, pointwiseA variant of isneginf that saves the output to the specified out.
523isposinfMathAlpha5.4aten, KernelGenTests if each element of input is positive infinity or not.
524ixorMathAlpha5.4aten, pointwise, KernelGenComputes the bitwise XOR of input tensors or a tensor and a scalar (inplace).
525jagged_to_padded_dense_forwardTensorAlpha5.4aten, KernelGenConverts a jagged (variable-length) tensor to a padded dense tensor.
526kronLinearAlgStable2.2atenComputes the Kronecker product of input and other.
527kthvalueTensorAlpha5.4aten, KernelGenReturns the kth smallest element of the given input tensor along a given dimension.
528layer_normNeuralNetworkStable1.0atenAn internal IR for applying Layer Normalization for last certain number of dimensions.
529layer_norm_backwardReductionStable3.0atenThe backward case for layer_norm().
530lcmPointwiseAlpha5.4aten, pointwise, KernelGenCompute the element-wise least common multiple of input tensors.
531lcm_PointwiseAlpha5.4aten, pointwise, KernelGenCompute the element-wise least common multiple of input tensors, storing result in-place.
532leMathStable2.0aten, pointwiseComputes that input is less than or equal to other element-wise.
533le_MathAlpha5.4aten, KernelGenThe in-place version of le().
534le_scalarMathStable2.0aten, pointwiseThe scalar version of le().
535le_scalar_MathAlpha5.4aten, KernelGenThe scalar version of le_().
536leaky_reluNeuralNetworkBeta5.3atenApplies the LeakyReLU function element-wise.
537leaky_relu_NeuralNetworkBeta5.3atenThe in-place version of leaky_relu().
538leaky_relu_backwardNeuralNetworkBeta5.3aten, pointwiseA variant of leaky_relu() for backward case.
539leaky_relu_outNeuralNetworkBeta5.3atenA variant of leaky_relu().
540lerp_scalarLinearAlgStable3.0aten, pointwiseThe scalar version of lerp().
541lerp_scalar_LinearAlgStable3.0aten, pointwiseThe in-place, scalar version of lerp().
542lerp_tensorLinearAlgStable3.0aten, pointwisePerforms a linear interpolation of two tensors start (given by input) and end based on a scalar or tensor weight and returns the resulting out tensor.
543lerp_tensor_LinearAlgStable3.0aten, pointwiseThe in-place version of lerp().
544less_MathAlpha5.4aten, KernelGenThe in-place version of less().
545less_equalMathAlpha5.4aten, KernelGenComputes element-wise less-than-or-equal-to comparison.
546less_equal_MathAlpha5.4aten, KernelGenThe in-place version of less_equal().
547less_equal_scalarMathAlpha5.4aten, KernelGenComputes less-than-or-equal-to comparison with a scalar.
548less_equal_scalar_MathAlpha5.4aten, KernelGenThe scalar version of less_equal_().
549less_scalar_MathAlpha5.4aten, KernelGenThe scalar version of less_().
550lgammaMathAlpha5.4aten, KernelGenComputes the logarithm of the gamma function on input.
551lgamma_MathAlpha5.4aten, KernelGenIn-place version of lgamma().
552liftTensorAlpha5.4aten, KernelGenReturns self tensor. Used within compiled graphs to lift tensors out of the graph or modify autograd metadata.
553lift_freshTensorAlpha5.4aten, KernelGenReturns a fresh tensor holding the same values as the input, used to lift constants into a compiled graph.
554lift_fresh_copyTensorStable5.3aten, KernelGenCreates a new, independent copy of a tensor within a compiled graph.
555lift_outTensorAlpha5.4aten, KernelGenOut-of-place variant of lift that copies self to the out tensor.
556linalg_choleskyLinearAlg BLASAlpha5.4aten, KernelGenComputes the Cholesky decomposition of a symmetric positive-definite matrix A or for batches of symmetric positive-definite matrices.
557linalg_crossLinearAlgAlpha5.4atenComputes the cross product of two 3-dimensional vectors along a specified dimension.
558linalg_cross_outLinearAlgAlpha5.4atenA variant of linalg_cross that allows the output to be assigned to out.
559linalg_detLinearAlgAlpha5.4atenComputes the determinant of a square matrix via LU decomposition with partial pivoting.
560linalg_det_outLinearAlgAlpha5.4atenA variant of linalg_det that writes the determinant to the provided out tensor.
561linalg_eigMathAlpha5.4aten, KernelGenComputes the eigenvalue decomposition of a general square matrix, returning the complex eigenvalues and eigenvectors. Also supports batches of matrices.
562linalg_eigvalsMathAlpha5.4aten, KernelGenComputes the eigenvalues of a square matrix.
563linalg_householder_productMathAlpha5.4aten, KernelGenCompute the matrix product of Householder matrices from QR decomposition.
564linalg_ldl_factorLinearAlgAlpha5.4aten, KernelGenComputes the compact LDL factorization of a Hermitian or symmetric matrix.
565linalg_ldl_factor_exMathAlpha5.4aten, KernelGenComputes the LDL factorization with pivot information for a symmetric or Hermitian matrix.
566linalg_ldl_solveLinearAlgAlpha5.4aten, KernelGenTriton kernel implementation for linalg_ldl_solve.
567linalg_lstsqLinearAlg BLASAlpha5.4atenComputes the least-squares solution (gels driver) to A X = B for full-rank over- or underdetermined systems, including batches, via Householder TSQR.
568linalg_luMathAlpha5.4atenComputes the LU factorization with partial pivoting of a matrix and returns the permutation matrix P and the lower/upper triangular factors L and U such that P @ L @ U = A (or L @ U = A for pivot=False).
569linalg_lu_factorMathAlpha5.4atenComputes a compact representation of the LU factorization with partial pivoting of a matrix.
570linalg_lu_factor_exMathAlpha5.4atenComputes a compact representation of the LU factorization with partial pivoting of a matrix and returns an info tensor indicating whether the factorization was successful (info == 0) or the position of the first zero pivot (1-indexed). This is the "expert" version of linalg_lu_factor, equivalent to LAPACK's getrf.
571linalg_lu_factor_ex_outMathAlpha5.4atenOut-of-place version of linalg_lu_factor_ex. Supports calling torch.linalg.lu_factor_ex(A, *, pivot=True, check_errors=False, out=(LU, pivots, info)). The out parameter provides pre-allocated output tensors for in-place writing.
572linalg_lu_factor_outMathatenOut-of-place version of linalg_lu_factor. Supports calling torch.linalg.lu_factor(A, *, pivot=True, out=(LU, pivots)). The out parameter provides pre-allocated output tensors.
573linalg_lu_outMathAlpha5.4atenOut-of-place version of linalg_lu. Supports calling torch.linalg.lu(A, *, pivot=True, out=(P, L, U)). The out parameter provides pre-allocated output tensors for in-place writing.
574linalg_matrix_expLinearAlgBeta5.4atenComputes the matrix exponential of a square matrix via scaling-and-squaring with an optimized Taylor polynomial approximation of degree 18.
575linalg_matrix_exp_outLinearAlgBeta5.4atenA variant of linalg_matrix_exp that writes the matrix exponential to the provided out tensor.
576linalg_matrix_normLinearAlgAlpha5.4atenComputes the matrix norm over the given dimensions. Supports ord values: 1, -1, 2, -2, inf, -inf, 'fro', 'nuc'. For ord=2/-2/nuc, internally uses SVD (singular value decomposition). For ord=1/-1/inf/-inf, uses max/min absolute column/row sums. For ord='fro', uses Frobenius norm via per-row L2 reduction.
577linalg_matrix_powerLinearAlgBeta5.4atenComputes the n-th power of a square matrix using binary exponentiation.
578linalg_matrix_power_outLinearAlgBeta5.4atenOut-of-place version of linalg_matrix_power. Supports calling torch.linalg.matrix_power(A, n, *, out=None). The out parameter provides a pre-allocated output tensor for in-place writing.
579linalg_matrix_rankLinearAlgBeta5.4atenComputes the numerical rank of a matrix or a batch of matrices.
580linalg_matrix_rank_atol_rtol_floatLinearAlgBeta5.4atenComputes numerical matrix rank using optional scalar absolute and relative tolerances.
581linalg_matrix_rank_atol_rtol_float_outLinearAlgBeta5.4atenComputes numerical matrix rank using scalar tolerances and an output tensor.
582linalg_matrix_rank_atol_rtol_tensorLinearAlgBeta5.4atenComputes numerical matrix rank using optional tensor absolute and relative tolerances.
583linalg_matrix_rank_atol_rtol_tensor_outLinearAlgBeta5.4atenComputes numerical matrix rank using tensor tolerances and an output tensor.
584linalg_matrix_rank_outLinearAlgBeta5.4atenComputes numerical matrix rank using a scalar tolerance and an output tensor.
585linalg_matrix_rank_out_tol_tensorLinearAlgBeta5.4atenComputes numerical matrix rank using a tensor tolerance and an output tensor.
586linalg_matrix_rank_tol_tensorLinearAlgBeta5.4atenComputes numerical matrix rank using a positional tensor tolerance.
587linalg_matrix_sqrthLinearAlgAlpha5.4aten, KernelGenComputes the principal square root of a symmetric or Hermitian positive-definite matrix.
588linalg_matrix_sqrth_outLinearAlgAlpha5.4aten, KernelGenWrites the principal square root of a symmetric or Hermitian positive-definite matrix to an output tensor.
589linalg_normLinearAlgBeta5.4atenComputes a vector or matrix norm, mirroring torch.linalg.norm dispatch. The vector branch (ord=None/2/1/0/p/±inf, dim as int or 1-tuple or None) reuses linalg_vector_norm; the matrix branch (ord='fro'/'nuc'/±1/±2/±inf, dim as 2-tuple) reuses linalg_matrix_norm.
590linalg_qrLinearAlgBeta5.4atenComputes the QR decomposition of a matrix (or a batch of matrices) via a blocked Householder algorithm with a TSQR fast path for tall-skinny inputs. Pure-Triton implementation supporting the "reduced", "complete" and "r" modes.
591linalg_qr_outLinearAlgBeta5.4atenThe out= variant of linalg_qr; writes the Q and R factors into the caller-provided output tensors.
592linalg_slogdetLinearAlgAlpha5.4aten, KernelGenComputes the sign and natural logarithm of the absolute value of the determinant of a square matrix using LU decomposition with partial pivoting.
593linalg_solve_triangularLinearAlg BLASAlpha5.4atenSolves a triangular system of linear equations with a unique solution, returning a tensor X such that AX = B (or XA = B when right=True) for a triangular matrix A and matrix of right-hand sides B.
594linalg_solve_triangular_outLinearAlg BLASAlpha5.4atenOut-of-place variant of linalg_solve_triangular: solves a triangular system of linear equations and writes the result into the provided out tensor.
595linalg_svdBLASAlpha5.4aten, KernelGenComputes the singular value decomposition (SVD) of a matrix.
596linalg_svdvalsMathAlpha5.4aten, KernelGenComputes the singular values of a matrix.
597linalg_vecdotMathAlpha5.4atenComputes the dot product of two vectors along a specified dimension.
598linalg_vecdot_outMathAlpha5.4atenComputes the dot product of two vectors along a specified dimension.
599linearBLASAlpha5.4aten, KernelGenApplies a linear transformation to the incoming data: y = x @ W^T + b.
600linear_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for linear_backward.
601linspaceTensorStable2.2atenCreates a one-dimensional tensor of size steps whose values are evenly spaced from start to end, inclusive.
602list_to_tensoraten, KernelGen, IRConstructs a 1-D int32 tensor from a Python list of integers. This is an IR/prim operator; the FlagGems implementation copies the values into the output tensor with a Triton kernel.
603logMathStable2.2aten, pointwiseReturns a new tensor with the natural logarithm of the elements of input.
604log10MathBeta5.3aten, pointwiseReturns a new tensor with the logarithm to the base 10 of the elements of input.
605log10_MathBeta5.3aten, pointwiseThe in-place version of log10().
606log10_outMathBeta5.3aten, pointwiseA variant of log10() that assigns the output to the provided out.
607log1pMathStable5.3aten, KernelGenComputes the natural logarithm of 1+x(y_i=log_e(x_i+1)) for each element in the input tensor.
608log1p_MathStable5.3aten, KernelGenComputes the natural logarithm of 1+x(y_i=log_e(x_i+1)) for each element in the input tensor in-place.
609log2MathAlpha5.4aten, pointwise, KernelGenReturns a new tensor with the logarithm to the base 2 of the elements of input.
610log2_MathAlpha5.4aten, pointwise, KernelGenThe in-place version of log2().
611log_MathBeta5.3aten, KernelGenThe in-place version of log(). Computes the natural logarithm of each element in the input tensor in-place.
612log_normalDistributionBeta5.4aten, skip_precision_checkReturns a new tensor with log-normal samples parameterized by mean and standard deviation.
613log_normal_DistributionAlpha5.1aten, KernelGenFills the tensor with log-normal samples parameterized by mean and standard deviation, in-place.
614log_sigmoidNeuralNetworkStable2.2aten, pointwise, nn.functionalApplies the Logsigmoid function element-wise.
615log_sigmoid_backwardNeuralNetworkAlpha5.4aten, pointwiseComputes the input gradient of the Logsigmoid function.
616log_sigmoid_backward_outNeuralNetworkAlpha5.4aten, pointwiseA variant of log_sigmoid_backward that assigns the output to the grad_input tensor.
617log_sigmoid_forwardMathAlpha5.4aten, KernelGenComputes the output and buffer for the log-sigmoid function, used in autograd backward pass.
618log_softmaxNeuralNetworkStable3.0aten, ReductionAn internal IR for applying a softmax followed by a logarithm.
619log_softmax_backward_dataNeuralNetworkBeta5.3aten, KernelGenComputes the gradient of the input tensor with respect to a log_softmax operation during backpropagation.
620log_softmax_backward_data_outNeuralNetworkBeta5.3aten, KernelGenA variant of _log_softmax_backward_data that assigns the output to the out tensor.
621log_softmax_outNeuralNetworkStable3.0aten, ReductionAn internal IR for applying a softmax followed by a logarithm.
622logaddexpMathStable5.3aten, pointwise, KernelGenComputes the element-wise logarithm of the sum of the exponentials of two input tensors.
623logaddexp2MathBeta5.4aten, pointwiseComputes the element-wise base-2 logarithm of the sum of base-2 exponentiations of two input tensors.
624logaddexp2_outMathBeta5.4aten, pointwiseA variant of logaddexp2 that allows the output to be assigned to an out tensor.
625logaddexp_outMathStable5.3aten, pointwise, KernelGenA variant of logaddexp that allows the output to be assigned to an out tensor.
626logcumsumexpMathAlpha5.4aten, KernelGenTriton kernel implementation for logcumsumexp.
627logcumsumexp_outMathAlpha5.4aten, KernelGenA variant of logcumsumexp that allows the output to be assigned to an out tensor.
628logical_andMathStable2.2aten, pointwiseComputes the element-wise logical AND of the given input tensors. Zeros are treated as False and nonzeros are treated as True.
629logical_and_MathStable5.0aten, pointwiseThe in-place version of logical_and().
630logical_notMathStable2.2aten, pointwiseComputes the element-wise logical NOT of the given input tensor.
631logical_not_MathAlpha5.4aten, KernelGenIn-place version of logical_not. Computes the element-wise logical NOT.
632logical_orMathStable2.2aten, pointwiseComputes the element-wise logical OR of the given input tensors.
633logical_or_MathStable5.0aten, pointwiseThe in-place version of logical_or().
634logical_xorMathStable2.2aten, pointwiseComputes the element-wise logical XOR of the given input tensors.
635logical_xor_MathAlpha5.4aten, KernelGen, pointwiseThe in-place version of logical_xor().
636logitLinearAlgStable5.3aten, pointwise, KernelGenReturns a new tensor with the logit of the elements of input. input is clamped to [eps, 1-eps] when eps is not None. When eps is None and input<0 or input>1, the function will yield NaN.
637logit_LinearAlgStable5.3aten, pointwise, KernelGenThe in-place version of logit().
638logit_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for logit_backward.
639logit_outLinearAlgStable5.3aten, pointwise, KernelGenA variant of logit that allows the output to be assigned to another tensor.
640logspacetensorStable4.0atenCreates a one-dimensional tensor of size steps whose values are evenly spaced from base^start to base^end, inclusive, on a logarithmic scale with base base.
641logsumexpMathAlpha5.3aten, KernelGenComputes the log of the sum of exponentials of elements in the input tensor along given dimensions.
642lshiftMathAlpha5.4aten, KernelGenTriton kernel implementation for lshift.
643lstmNeuralNetworkAlpha5.4aten, KernelGenTriton kernel implementation for lstm.
644ltMathStable2.0aten, pointwiseComputes that input is less than other element-wise.
645lt_MathAlpha5.4aten, KernelGenThe in-place version of lt().
646lt_scalarMathStable2.0aten, pointwiseThe scalar version of lt.
647lt_scalar_MathAlpha5.4aten, KernelGenThe scalar version of lt_().
648lu_unpackMathAlpha5.4aten, KernelGenUnpacks the LU decomposition returned by lu_factor into separate permutation matrix P, lower triangular L, and upper triangular U.
649lu_unpack_outMathAlpha5.4aten, KernelGenOut-of-place variant of lu_unpack that writes results into pre-allocated output tensors.
650make_dep_tokenTensorAlpha5.4aten, KernelGenCreates a scalar tensor as a dependency token for TorchScript tracing and class-based compiler frameworks. The value of the returned tensor is uninitialized and should not be consumed.
651margin_ranking_lossNeuralNetworkBeta5.3aten, nn.functional, KernelGenCompute the margin ranking loss.
652masked_fillTensorStable2.2aten, pointwiseFills elements of given tensor with value where mask is True.
653masked_fill_TensorStable2.2aten, pointwise, skip_precision_checkThe in-place version of masked_fill().
654masked_fill_scalarTensorStable2.2aten, pointwiseFills elements of given tensor with value where mask is True.
655masked_fill_scalar_TensorStable2.2aten, pointwise, skip_precision_checkThe in-place version of masked_fill().
656masked_scaleTensorAlpha5.4aten, KernelGen, pointwiseThe low-level implementation for computing input multiplied by scale with masking.
657masked_scattertensorStable4.2atenCopies elements from source into the given tensor at positions where the mask is True.
658masked_scatter_tensorStable4.2atenThe in-place version of masked_scatter().
659masked_scatter_backwardtensorAlpha5.4atenBackward of masked_scatter with respect to the source tensor. Returns a tensor of shape sizes where the first mask.sum() elements are the gradient values from the positions where mask was True (obtained via stream-compaction / masked_select), and the remaining elements are zero (the tail of source that was never consumed by the forward pass).
660masked_selectTensorStable2.1atenReturns a new 1-D tensor which indexes the input tensor according to the boolean mask mask which is a BoolTensor.
661matmul_bias_activationBLASAlpha5.4KernelGenFused matrix multiplication with bias addition and ReLU activation.
662matmuladdMathAlpha5.4fused, KernelGenComputes matrix multiplication followed by bias addition.
663matrix_exp_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for matrix_exp_backward.
664maxLinearAlgStable2.0aten, ReductionReturns the maximum value of all elements in the input tensor.
665max_dimLinearAlgStable2.0aten, ReductionReturns a namedtuple (values, indices) where values is the maximum value of each row of the input tensor in the given dimension dim. And indices is the index location of each maximum value found (argmax).
666max_pool1dNeuralNetworkAlpha5.4aten, KernelGenTriton kernel implementation for max_pool1d.
667max_pool2d_backwardIRStable4.0atenApplies a 2D max pooling over an input signal composed of several input planes. This is an IR representation rather than a public API and it is for the backward step.
668max_pool2d_with_indicesIRStable4.0atenApplies a 2D max pooling over an input signal composed of several input planes. This is an IR representation rather than a public API.
669max_pool2d_with_indices_backwardNeuralNetworkAlpha5.4aten, KernelGenThe backward version of max_pool2d_with_indices(). This is an IR representation rather than a public API and it is for the backward step.
670max_pool3d_backwardNeuralNetworkBeta5.3aten, nn.functionalThe backward version of max_pool2d_with_indices().
671max_pool3d_with_indicesNeuralNetworkBeta5.3aten, nn.functionalApplies a 3D max pooling over an input signal composed of several input planes.
672max_pool3d_with_indices_backwardNeuralNetworkBeta5.3aten, KernelGenBackward pass for 3D max pooling with indices. Computes the gradient of the input given the gradient of the output and the indices from the forward pass.
673max_unpool2dNeuralNetworkAlpha5.4aten, KernelGenComputes a partial inverse of MaxPool2d.
674max_unpool3dMathAlpha5.4aten, KernelGenTriton kernel implementation for max_unpool3d.
675maximumMathStable2.1aten, pointwiseComputes the element-wise maximum of input and other.
676meanLinearAlgStable1.0aten, ReductionReturns the mean value of all elements in the input tensor. Input must be floating point or complex.
677mean_dimReductionStable2.0atenReturns the mean value of each row of the input tensor in the given dimension dim. If dim is a list of dimensions, reduce over all of them.
678medianLinearAlgBeta5.4aten, KernelGen, ReductionReturns the median of the values in input.
679median_dimLinearAlgBeta5.4aten, KernelGen, ReductionReturns a namedtuple (values, indices) where values contains the median of each row of input in the dimension dim, and indices contains the index location of each median value found.
680median_dim_valuesLinearAlgBeta5.4aten, KernelGen, ReductionA variant of median_dim() with the out argument.
681median_outLinearAlgBeta5.4aten, KernelGen, ReductionA variant of median() with the out argument.
682mhc_bwdNeuralNetworkBeta5.3fused, vLLM, DSAThe backward case for MHC (Manifold-Constrained Hyper-Connections). This is the Triton implmentation for Sinkhorn implicit CG differentiation. It computes the gradient of the Sinkhorn normalization using implicit differentiation via the conjugate gradient method.
683mhc_postNeuralNetworkBeta5.3fused, vLLM, DSATriton implementation of mHC Post operator (optimized v3).
684mhc_preNeuralNetworkBeta5.3fused, vLLM, DSATriton implementation of mHC Pre operator (optimized v2).
685minTensorStable2.0aten, ReductionReturns the minimum value of all elements in the input tensor.
686min_dimLinearAlgStable2.0aten, ReductionReturns a namedtuple (values, indices) where values is the minimum value of each row of the input tensor in the given dimension dim. And indices is the index location of each minimum value found (argmin).
687minimumMathStable2.1aten, pointwiseComputes the element-wise minimum of input and other.
688miopen_batch_norm_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for miopen_batch_norm_backward.
689mishNeuralNetworkAlpha5.4aten, KernelGen, pointwise, ActivationApplies the Mish activation function element-wise.
690mish_NeuralNetworkAlpha5.4aten, KernelGen, pointwise, ActivationThe in-place version of mish().
691mish_backwardNeuralNetworkAlpha5.4aten, KernelGen, pointwiseComputes the gradient of Mish activation with respect to the input.
692mkldnn_rnn_layerMathAlpha5.4aten, KernelGen, NoCPUApplies a single-layer unidirectional LSTM (oneDNN mkldnn_rnn_layer, mode=2) to the input sequence, returning the output sequence and the final hidden and cell states.
693mmBLASStable1.0atenPerforms a matrix multiplication of the two input matrices.
694mm_outBLASStable3.0atenA variant of mm() with out specified.
695mm_w8a8_fp8BLASAlpha5.4NoCPU, QuantizationDynamically quantizes BF16 activations and weights to FP8 E4M3, computes their matrix product with FP8 Tensor Cores, and applies FP32 scales.
696modeLinearAlgBeta5.3aten, ReductionReturns a namedtuple (values, indices) where values is the mode value of each row of the input tensor in the given dimension dim, i.e. a value which appears most often in that row, and indices is the index location of each mode value found.
697moe_align_block_size_tritonMoEStable4.2fused, Reduction, vLLMAligns the token distribution across experts to be compatible with block size for matrix multiplication.
698moe_sumMoEStable4.2fused, Reduction, vLLMAn implementation of Mixture of Experts (MoE) with sum-based aggregation instead of the more common weighted average.
699mse_lossNeuralNetworkStable2.2aten, pointwise, nn.functionalCompute the element-wise mean squared error, with optional weighting.
700mse_loss_backwardNeuralNetworkAlpha5.4aten, NeuralNetwork, KernelGenCompute the gradient of mean squared error loss with respect to the input.
701mulMathStable1.0aten, pointwiseMultiplies input by other.
702mul_MathStable2.2aten, pointwiseThe in-place version of mul().
703multinomialDistributionStable2.1aten, skip_precision_checkReturns a tensor where each row contains num_samples indices sampled from the multinomial probability distribution located in the corresponding row of tensor input.
704multiplyMathAlpha5.4aten, KernelGenMultiplies input by other element-wise (out-of-place alias for mul).
705multiply_MathAlpha5.4aten, KernelGenIn-place multiplication using Triton kernel (alias for mul_). Supports tensor-tensor and tensor-scalar operations.
706mvBLASStable2.0atenPerforms a matrix-vector product of the matrix input and the vector vec.
707mvlgammaMathAlpha5.4aten, KernelGenComputes the multivariate log-gamma function element-wise (out-of-place).
708mvlgamma_MathAlpha5.4aten, KernelGenComputes the multivariate log-gamma function in-place for each element of the input tensor.
709nan_to_numMathStable3.0aten, pointwiseReplaces NaN, positive infinity, and negative infinity values in input with the values specified by nan, posinf, and neginf, respectively.
710nan_to_num_MathAlpha5.4aten, KernelGen, pointwiseReplaces NaN, positive infinity, and negative infinity values in input in-place.
711nanmedianLinearAlgBeta5.4aten, ReductionReturns the median of the values in input, ignoring NaN values.
712nanmedian_dimLinearAlgBeta5.4aten, ReductionReturns a namedtuple (values, indices) where values contains the median of each row of input in the dimension dim, ignoring NaN values, and indices contains the index location of each median value found.
713nanmedian_dim_valuesLinearAlgBeta5.4aten, ReductionA variant of nanmedian_dim() with the out argument.
714nanmedian_outLinearAlgBeta5.4aten, ReductionA variant of nanmedian() with the out argument.
715nansumLinearAlgBeta5.4aten, ReductionReturns the sum of all elements in the input tensor, treating NaN values as zero. Supports global reduction and reduction along specified dimensions.
716nansum_outLinearAlgBeta5.4aten, ReductionA variant of nansum() with the out argument.
717narrowTensorAlpha5.4aten, KernelGen, skip_precision_checkPure layout operation (narrow); returns a zero-copy view sharing storage with the input.
718narrow_copyMathAlpha5.4aten, KernelGenTriton kernel implementation for narrow_copy.
719native_batch_normNeuralNetworkAlpha5.4aten, KernelGenApply native batch normalization using the existing implementation.
720native_batch_norm_legitNeuralNetworkAlpha5.4aten, KernelGenBatch normalization with in-place running statistics updates, returning the normalized output along with the saved mean and inverse standard deviation.
721native_batch_norm_legit_functionalNeuralNetworkAlpha5.4aten, KernelGenFunctional version of batch normalization that returns the saved mean and inverse standard deviation, along with updated running estimates.
722native_batch_norm_legit_no_statsNeuralNetworkAlpha5.4aten, KernelGenBatch normalization without persistent running statistics.
723native_batch_norm_legit_no_stats_outNeuralNetworkAlpha5.4aten, KernelGenOut variant of batch normalization without running statistics.
724native_batch_norm_legit_no_trainingNeuralNetworkAlpha5.4aten, KernelGenInference-only batch normalization using precomputed running_mean and running_var without updating them; returns the normalized output plus two empty save_mean/save_var tensors matching the aten schema.
725native_batch_norm_legit_outNeuralNetworkAlpha5.4aten, KernelGenOut variant of batch normalization with running statistics.
726native_dropout_backwardNeuralNetworkStable3.0aten, nn.functionalThe backward case of dropout().
727native_group_normNeuralNetworkAlpha5.4aten, KernelGenApply native group normalization using the existing implementation.
728native_layer_normNeuralNetworkAlpha5.4aten, KernelGenApply native layer normalization using the existing implementation.
729neMathStable2.0aten, pointwiseComputes that input is not equal to other element-wise.
730ne_MathAlpha5.4aten, KernelGen, pointwiseComputes element-wise not-equal comparison in-place.
731ne_scalarMathStable2.0aten, pointwiseThe scalar version of ne().
732negMathStable2.0aten, pointwiseReturns a new tensor with the negative of the elements of input.
733neg_MathStable2.2aten, pointwiseThe in-place version of neg().
734negativeMathAlpha5.4aten, KernelGennegative operator
735negative_MathAlpha5.4aten, KernelGenComputes the negative of input in-place.
736nested_from_padded_tensorTensorAlpha5.4aten, KernelGenConstructs a nested tensor from a padded tensor and offsets.
737nested_sum_backwardMathAlpha5.4aten, KernelGenTriton kernel implementation for _nested_sum_backward, the backward of a (nested) tensor sum over a single dimension. It broadcasts the gradient back to the shape of the original input along the reduced dim.
738nested_tensor_from_mask_left_alignedTensorAlpha5.4aten, KernelGenChecks whether a 3D tensor t (N * L * D) combined with a 2D boolean padding mask (N * L) is left-aligned, i.e. the valid (True) elements of every row appear at the beginning with no gaps. Returns a bool.
739nested_view_from_buffer_copyTensorAlpha5.4aten, KernelGen, MetaxCreates a nested tensor from a flat buffer using component sizes, strides, and offsets, with a Metax backend specialization.
740nested_view_from_jaggedTensorAlpha5.4aten, KernelGenConstructs a jagged-layout nested tensor from a flat values buffer and an offsets tensor. This is a pure view operation that wraps the values buffer without copying any data.
741new_fullTensorBeta5.3aten, pointwiseReturns a Tensor of size size filled with fill_value. By default, the returned Tensor has the same torch.dtype and torch.device as this tensor.
742new_onesTensorAlpha5.4aten, KernelGenReturns a Tensor of size size filled with 1, with the same torch.dtype and torch.device as the input tensor.
743nextafterMathAlpha5.4aten, KernelGenReturn the next floating-point value after input towards other, elementwise.
744nextafter_MathAlpha5.4aten, pointwiseReturns the next representable floating-point value from input toward other, elementwise. In-place version of nextafter.
745nll_loss2dNeuralNetworkAlpha5.4aten, KernelGenNegative log likelihood loss for a 4D input (N, C, H, W) of log-probabilities and a 3D integer target (N, H, W). Returns only the loss tensor, applying the requested reduction (none/mean/sum) with optional per-class weights and an ignore_index.
746nll_loss2d_backwardNeuralNetworkStable2.2aten, IRAn internal IR for supporting torch.nn.NLLLoss2d, which has been deprecated and is now integrated into the standard torch.nn.NLLLoss. This is the backward case.
747nll_loss2d_forwardNeuralNetworkStable2.2aten, IRAn internal IR for supporting torch.nn.NLLLoss2d, which has been deprecated and is now integrated into the standard torch.nn.NLLLoss. This is the forward case.
748nll_loss_backwardNeuralNetworkStable2.2aten, IRCompute the negative log likelihood loss. This is the backward case.
749nll_loss_forwardNeuralNetworkStable2.2aten, IRCompute the negative log likelihood loss. This is the forward case.
750nll_loss_nd_backwardNeuralNetworkStable5.0atenMeasures the performance of a classification model by penalizing low probabilities for correct classe.s This computes the gradients of this loss with respect to model parameters using automatic differentiation.
751nll_loss_nd_forwardNeuralNetworkStable5.0atenMeasures the performance of a classification model by calculating the negative log probability of the true class. This defines the computation flow, transforming input data through layers to produce output predictions.
752nonzeroTensorStable2.1atenReturns a 2-D tensor where each row is the index for a nonzero value. When as_tuple is explicitly set to True, this returns a tuple of 1-D index tensors, allowing for advanced indexing of all nonzero values.
753nonzero_numpyTensorAlpha5.3aten, KernelGenReturns a tuple of 1-D tensors, one for each dimension, containing the indices of the nonzero elements in the input tensor (NumPy-style).
754nonzero_staticTensorAlpha5.4atenReturns a fixed-size 2-D int64 tensor containing indices of nonzero elements. Rows beyond the number of nonzero elements are filled with fill_value.
755nonzero_static_outTensorAlpha5.4atenA variant of nonzero_static that writes the fixed-size result to out.
756normReductionAlpha5.4aten, KernelGen, reductionComputes the norm of vectors, matrices, and tensors.
757norm_scalarReductionAlpha5.4aten, KernelGen, reductionA variant of norm that computes the norm over all elements.
758norm_scalaropt_dimReductionAlpha5.4aten, KernelGen, reductionA variant of norm that computes the norm along specified dimensions.
759normal_TensorAlpha5.3skip_precision_checkRandom sampling operator (normal_).
760normal_float_float_DistributionStable5.0aten, pointwise, skip_precision_checkReturns a tensor of random numbers drawn from separate normal distributions whose mean and standard deviation are given. This is one of the variants that takes a float mean and a float std.
761normal_float_tensorDistributionStable2.1aten, pointwiseReturns a tensor of random numbers drawn from separate normal distributions whose mean and standard deviation are given. This is one of the variants that takes a float mean and a tensor std.
762normal_tensor_floatDistributionStable2.1aten, pointwiseReturns a tensor of random numbers drawn from separate normal distributions whose mean and standard deviation are given. This is one of the variants that takes a tensor mean and a float std.
763normal_tensor_tensorDistributionStable2.1aten, pointwiseReturns a tensor of random numbers drawn from separate normal distributions whose mean and standard deviation are given. This is one of the variants that takes a tensor mean and a tensor std.
764normed_cumsumReductionStable2.1atenGet the normalized cumulative sum where each step is divided by the total sum of the dataset, resulting in values ranging from 0 to 1. Internally used by the multinomial operator.
765not_equalLogicAlpha5.4aten, KernelGen, pointwiseComputes the element-wise logical NOT EQUAL of the given input tensors.
766not_equal_LogicAlpha5.4aten, KernelGen, pointwiseComputes element-wise not-equal comparison in-place.
767not_equal_scalarLogicAlpha5.4aten, KernelGen, pointwiseComputes the element-wise logical NOT EQUAL comparing a tensor to a scalar.
768nuclear_normMathAlpha5.4aten, KernelGenComputes the nuclear norm (sum of singular values) of a matrix.
769one_hotNeuralNetworkStable5.0aten, nn.functional, KernelGenTakes LongTensor with index values of shape (*) and returns a tensor of shape (*, num_classes) that have zeros everywhere except where the index of last dimension matches the corresponding value of the input tensor, in which case it will be 1.
770onesTensorStable2.1aten, skip_precision_checkReturns a tensor filled with the scalar value 1, with the shape defined by the variable argument size.
771ones_likeTensorStable2.1atenReturns a tensor filled with the scalar value 1, with the same size as input.
772ormqrLinearAlgAlpha5.4aten, KernelGenComputes the matrix-matrix multiplication of a product of Householder matrices with a general matrix.
773outerBLASStable2.0fusedComputes outer product of self and the input vector. If the self tensor is a vector of size n and the input tensor is a vector of size m, the out tensor (if specified) must be a matrix of size n * m.
774outplace_fused_expertsMoEStable5.3fused, Activation, vLLMThis operator allocates and returns a new output tensor.
775pack_seq_tritonNeuralNetworkBeta5.3fused, vLLM, DeepSeekV4Pack variable-length token sequences into a padded batched tensor.
776padNeuralNetworkStable2.1aten, pointwise, nn.functionalThis pads a tenor using the specified mode.
777pairwise_distanceMathAlpha5.4aten, nn.functionalComputes the pairwise distance between input vectors, or between columns of input matrices. Distances are computed using p-norm, with constant eps added to avoid division by zero if p is negative.
778pdistMathAlpha5.4aten, KernelGenComputes the p-norm distance between every pair of row vectors in the input. Input must be a 2-D tensor of shape (N, M). Returns a tensor of shape (N*(N-1)//2,) containing pairwise distances.
779pdist_backwardMathAlpha5.4aten, KernelGenComputes the gradient of pairwise distance computation.
780pdist_forwardMathAlpha5.4aten, KernelGenComputes the pairwise distance between rows of a matrix.
781per_token_group_quant_fp8QuantizationBeta5.3NoCPU, vLLMFunction to perform per-token-group quantization on an input tensor x. It converts the tensor values into signed float8 values and returns the quantized tensor along with the scaling factor used for quantization.
782permute_copyTensorAlpha5.4aten, KernelGenCreates and returns a copy of x with permuted dimensions.
783pixel_shuffleNeuralNetworkStable5.3aten, nn.functionalRearranges elements in a tensor to a new tensor of different shape.
784pixel_unshuffleNeuralNetworkStable5.3aten, KernelGenRearranges elements from a low-resolution feature map with many channels into a higher-resolution feature map with fewer channels.
785pixel_unshuffle_outNeuralNetworkBeta5.0aten, KernelGenA variant of pixel_unshuffle that assigns the output to the out tensor.
786poissonMathBeta5.3aten, KernelGenReturns a tensor of the same size as input with each element sampled from a Poisson distribution with rate given by the corresponding element in input.
787polarMathStable3.0aten, pointwiseConstructs a complex tensor whose elements are Cartesian coordinates corresponding to the polar coordinates with absolute value abs and angle angle.
788polygammaMathAlpha5.4atenComputes the n-th derivative of the digamma function of the input (the polygamma function), dispatching to dedicated digamma (n=0), trigamma (n=1), or Hurwitz-zeta (n>=2) Triton kernels.
789polygamma_MathAlpha5.4atenThe in-place version of polygamma().
790polygamma_outMathAlpha5.4atenA variant of polygamma that allows the output to be assigned to the out tensor.
791post_layer_norm_residualNeuralNetworkBeta5.4fusedFuses LayerNorm followed by a residual addition.
792pow_scalarMathStable1.0atenTakes the power of each element in input with exponent and returns a tensor with the result. The input is a single float, while the exponent is a tensor.
793pow_tensor_scalarMathStable1.0aten, pointwiseTakes the power of each element in input with exponent and returns a tensor with the result. The input is a tensor, while the exponent is a float.
794pow_tensor_scalar_MathStable2.2aten, pointwiseThis is the in-place version of pow_tensor_scalar().
795pow_tensor_tensorMathStable1.0aten, pointwiseTakes the power of each element in input with exponent and returns a tensor with the result. The input is a tensor, while the exponent is also a tensor.
796pow_tensor_tensor_MathStable2.2aten, pointwiseThis is the in-place version of pow_tensor_tensor().
797preluNeuralNetworkStable5.3aten, Activation, pointwise, nn.functional, KernelGenAn activation function used in neural networks that improves upon ReLU (Rectified Linear Unit) by allowing the network to learn the slope of negative inputs. It performs an element-wise operation that keeps positive values and scales negative values by a learnable parameter.
798prelu_kernelNeuralNetworkAlpha5.4aten, Activation, pointwise, KernelGenApplies the element-wise Parametric ReLU (PReLU) activation: f(x)=max(0,x)+weight*min(0,x), where weight is a learnable parameter with the same shape as the input.
799prelu_kernel_backwardMathAlpha5.1aten, KernelGenComputes backward gradients for Parametric ReLU (PReLU) forward pass. PReLU scales negative inputs by a learnable per-channel weight. This backward kernel computes gradients with respect to both input and weight.
800prodLinearAlgStable2.0aten, ReductionReturns the product of all elements in the input tensor.
801prod_dim_intReductionStable2.0atenReturns the product of each row of the input tensor in the given dimension dim.
802quantileTensorStable2.2atenComputes the q-th quantiles of each row of the input tensor along the dimension dim.
803quantized_lstmNeuralNetworkAlpha5.4aten, KernelGenApplies a multi-layer quantized LSTM over an input sequence. Takes the input sequence, initial hidden and cell states, and per-layer quantized weights, and returns the output sequence together with the final hidden and cell states.
804rad2degMathAlpha5.3aten, KernelGen, pointwiseConverts each element from angles in radians to degrees.
805rad2deg_MathAlpha5.3aten, KernelGen, pointwiseIn-place version of rad2deg.
806randDistributionStable2.1atenReturns a tensor filled with random numbers from a uniform distribution on the interval [0,1).
807rand_likeDistributionStable2.1atenReturns a tensor with the same size as input that is filled with random numbers from a uniform distribution on the interval [0,1).
808randintDistributionStable5.1aten, KernelGenReturns a tensor filled with random integers generated uniformly between low (inclusive) and high (exclusive).
809randint_likeDistributionAlpha5.4aten, KernelGenReturns a tensor with the same size as input that is filled with random integers generated uniformly between 0 (inclusive) and high (exclusive).
810randnDistributionStable2.1atenReturns a tensor filled with random numbers from a normal distribution with mean 0 and variance 1 (also called the standard normal distribution).
811randn_likeDistributionStable2.1atenReturns a tensor with the same size as input that is filled with random numbers from a normal distribution with mean 0 and variance 1.
812random_TensorAlpha5.1skip_precision_checkRandom sampling operator (random_).
813randpermDistributionStable2.2aten, skip_precision_checkReturns a random permutation of integers from 0 to n - 1.
814rangeTensorAlpha5.4aten, KernelGenReturns a 1-D tensor of size floor(end - start) + 1 with values from start to end with step 1.
815reciprocalMathStable1.0aten, pointwiseReturns a new tensor with the reciprocal of the elements of input.
816reciprocal_MathStable2.2aten, pointwiseThis is the in-place version of reciprocal().
817reflection_pad1dNeuralNetworkStable5.3aten, pointwise, KernelGenPads the input 3D or 2D tensor (typically representing signals or sequences) by reflecting the boundary values at the edges.
818reflection_pad1d_backwardMathAlpha5.3aten, KernelGenComputes the gradient of reflection_pad1d with respect to the input tensor.
819reflection_pad1d_outNeuralNetworkStable5.3aten, pointwise, KernelGenA variant of reflection_pad1d that assigns the output to out tensor.
820reflection_pad2dNeuralNetworkStable5.3aten, pointwise, KernelGenPads the input 3D or 2D tensor (typically representing signals or sequences) by reflecting the boundary values at the both edges.
821reflection_pad2d_backwardMathAlpha5.4aten, KernelGenComputes the gradient of reflection_pad2d with respect to the input tensor.
822reflection_pad2d_outNeuralNetworkStable5.3aten, pointwise, KernelGenA variant of reflection_pad2d that assigns the output to out tensor.
823reflection_pad3dNeuralNetworkAlpha5.4aten, pointwise, KernelGenPads the input 5D tensor by reflecting the boundary values at both edges along the last three spatial dimensions (depth, height, width).
824reflection_pad3d_backwardMathAlpha5.1aten, KernelGenComputes the gradient of reflection_pad3d with respect to the input tensor.
825regluNeuralNetworkBeta5.3fused, TransformerRectified Gated Linear Unit is a variant of GLU that uses ReLU instead of the sigmoid function for gating.
826reluNeuralNetworkStable1.0aten, Activation, pointwise, nn.functionalApply the RELU (Rectified Linear Unit) activation function element-wise.
827relu6NeuralNetworkStable5.3aten, pointwise, Activation, KernelGenApplies the element-wise function f(x)=min(max(0,x),6). This is a variation of the standard ReLU activation function that "caps" its output at a maximum value of 6.
828relu_NeuralNetworkStable2.2aten, pointwise, ActivationThis is the in-place version of relu().
829remainderMathAlpha5.4aten, KernelGenComputes Python's modulus operation entrywise. The result has the same sign as the divisor other and its absolute value is less than that of other.
830remainder_scalarMathStable2.2atenComputes Python's modulus operation entrywise. The result has the same sign as the divisor other and its absolute value is less than that of other.
831remainder_scalar_MathStable2.2atenThis is the in-place version of remainder().
832remainder_scalar_tensorMathStable2.2atenComputes Python's modulus operation entrywise. The result has the same sign as the divisor other and its absolute value is less than that of other.
833remainder_tensorMathStable2.2atenComputes Python's modulus operation entrywise. The result has the same sign as the divisor other and its absolute value is less than that of other.
834remainder_tensor_MathStable2.2atenThis is the in-place version of remainder().
835renormMathAlpha5.4aten, KernelGenReturns a tensor where each sub-tensor along the given dimension is normalized such that the p-norm is lower than a max norm value.
836renorm_MathAlpha5.4aten, KernelGenThe in-place version of renorm().
837repeatTensorStable2.1atenRepeats this tensor along the specified dimensions.
838repeat_interleave_self_intTensorStable2.2aten, pointwiseRepeats elements of a tensor. The number of repetitions is specified as an integer repeats.
839repeat_interleave_self_tensorTensorStable2.2aten, pointwiseRepeats elements of a tensor. The number of repetitions is specified as a tensor repeats. repeats is broadcasted to fit the shape of the given axis.
840repeat_interleave_tensorTensorStable2.2aten, pointwiseRepeats 0 repeats[0] times, 1 repeats[1] times, 2 repeats[2] times, etc.
841replication_pad1dTensorStable5.4aten, KernelGenPads the edge of a 1D input tensor by repeating the boundary values.
842replication_pad1d_outTensorStable5.4aten, KernelGenA variant of replication_pad1d that assigns the output to the out tensor.
843replication_pad2dTensorAlpha5.4aten, KernelGenPads the edge of a 2D input tensor by repeating the boundary values.
844replication_pad2d_backwardNeuralNetworkAlpha5.4atenComputes the gradient for replication_pad2d. Gradients from padded output are redistributed back to the original input boundaries via replication (edge-value duplication) semantics.
845replication_pad2d_backward_grad_inputNeuralNetworkAlpha5.4atenA variant of replication_pad2d_backward that writes gradients into a pre-allocated grad_input tensor instead of allocating a new one.
846replication_pad2d_outTensorAlpha5.4aten, KernelGenA variant of replication_pad2d that assigns the output to the out tensor.
847replication_pad3dNeuralNetworkBeta5.3atenPads the edge of a 3D input tensor by repeating the boundary values.
848replication_pad3d_backwardNeuralNetworkAlpha5.4aten, KernelGenAccumulates output gradients into the clamped input positions used by three-dimensional replication padding.
849reshape_and_cacheAttentionStable3.0fused, vLLMStore the key/value token states into the pre-allcated kv_cache buffers of paged attention.
850reshape_and_cache_flashAttentionStable3.0fusedStore the key/value token states into the pre-allcated kv_cache buffers of paged attention.
851resizeTensorAlpha5.4aten, KernelGenResizes the input tensor to the specified size.
852resize_TensorAlpha5.4aten, KernelGenThe in-place version of resize().
853resize_asTensorAlpha5.4aten, KernelGenTriton kernel implementation for resize_as.
854resize_as_TensorAlpha5.4aten, KernelGenThe in-place version of resize_as().
855resize_outputMathAlpha5.3aten, KernelGenResizes tensor output storage and copies overlapping elements.
856resize_output_MathAlpha5.4aten, KernelGenResizes the output tensor in-place to the given size.
857resolve_conjScienceStable2.1atenReturns a new tensor with materialized conjugation if input's conjugate bit is set to True, else returns input. The output tensor will always have its conjugate bit set to False.
858resolve_negScienceStable2.1atenReturns a new tensor with materialized negation if input's negative bit is set to True, else returns input. The output tensor will always have its negative bit set to False.
859rms_normNeuralNetworkStable2.0aten, nn.functional, ReductionApply Root Mean Square Layer Normalization over a mini-batch of inputs.
860rms_norm_w8a16_fp8NeuralNetworkAlpha5.4NoCPU, Normalization, QuantizationApplies Root Mean Square Layer Normalization to BF16 activations using group-wise FP8 E4M3 weights and per-group scales.
861rnn_reluMathAlpha5.4aten, KernelGen, NoCPUTriton kernel implementation for rnn_relu.
862rollBLASBeta5.3aten, KernelGenRoll the tensor input along the given dimension(s). Elements that are shifted beyond the last position are re-introduced at the first position.
863rot90MathAlpha5.4aten, KernelGenTriton kernel implementation for rot90.
864roundMathBeta5.3aten, pointwiseRounds elements of input to the nearest integer.
865round_MathBeta5.3aten, pointwiseThe inplace version of round.
866round_outMathBeta5.3aten, pointwiseA variant of round that assigns the output to the specifiec out.
867rrelu_with_noise_backwardNeuralNetworkBeta5.0aten, KernelGenComputes the gradient of the Randomized Leaky ReLU (RReLU) activation function with respect to its input during backpropagation. It uses the noise tensor generated in the forward pass to correctly apply the slope to negative input values.
868rrelu_with_noise_functionalMathAlpha5.4aten, KernelGenTriton kernel implementation for rrelu_with_noise_functional.
869rshiftMathAlpha5.4aten, pointwise, KernelGenApplies a bitwise right shift to an integer tensor with a tensor or scalar shift value.
870rshift_outMathAlpha5.4aten, pointwise, KernelGenA variant of rshift() that assigns the output to the given out parameter.
871rsqrtMathStable1.0aten, pointwiseReturns a new tensor with the reciprocal of the square-root of each of the elements of input.
872rsqrt_MathStable2.2aten, pointwiseThe in-place version of rsqrt().
873rsub_scalarMathAlpha5.3aten, KernelGenSubstracts other, scaled by alpha, from input. This is the scalar version.
874rsub_tensorMathAlpha5.3aten, KernelGenSubstracts other, scaled by alpha, from input. This is the tensor version.
875rwkv_ka_fusionRWKVStable4.1fusedMerges, aligns, and enhances features from different data sources or spatial directions using the efficient, linear-time RWKV framework.
876rwkv_mm_sparsityRWKVStable4.1fusedOptimized, lossless sparse matrix multiplication in RWKV-7 models.
877safe_softmaxNeuralNetworkAlpha5.3aten, IR, KernelGenApply a softmax function. Note this version may not be functional.
878scalar_tensorTensorAlpha5.4atenCreates a 0-dimensional (scalar) tensor from a Python numeric value. The tensor's dtype can be specified, otherwise inferred from the input value.
879scaled_dot_product_attentionNeuralNetworkStable2.2nn.functional, AttentionComputes scaled dot product attention on query, key and value tensors, using an optional attention mask if passed and applying dropout if a probability greater than 0.0 is specified. The optional scale argument can only be specified as a keyword argument.
880scaled_dot_product_attention_backwardNeuralNetworkStable2.2nn.functional, AttentionThe backward case for scaled_dot_product_attention.
881scaled_dot_product_attention_forwardNeuralNetworkStable2.2nn.functional, AttentionThe forward case for scaled_dot_product_attention.
882scaled_dot_product_attention_mathNeuralNetworkAlpha5.4aten, KernelGenComputes scaled dot-product attention using the math-based implementation.
883scaled_dot_product_cudnn_attentionNeuralNetworkAlpha5.4aten, NoCPU, KernelGenComputes scaled dot product attention using the cuDNN attention API interface, returning output, log-sum-exp, and auxiliary tensors for backward computation.
884scaled_dot_product_cudnn_attention_backwardNeuralNetworkBeta5.4aten, NoCPUBackward kernel for FlashAttention, computing gradients of queries, keys, values, and attention outputs efficiently.
885scaled_dot_product_efficient_attentionNeuralNetworkAlpha5.4KernelGen, nn.functional, Attention, atenComputes the efficient memory attention for scaled_dot_product_attention. This is a composite operator that wraps scaled_dot_product_attention_forward and additionally returns log-sumexp and dropout RNG state.
886scaled_dot_product_efficient_attention_backwardNeuralNetworkBeta5.4aten, NoCPUBackward kernel for FlashAttention, computing gradients of queries, keys, values, and attention outputs efficiently.
887scaled_dot_product_flash_attentionNeuralNetworkAlpha5.4aten, NoCPU, KernelGenRun the scaled dot product FlashAttention forward operation.
888scaled_dot_product_flash_attention_backwardNeuralNetworkBeta5.4aten, NoCPUBackward kernel for FlashAttention, computing gradients of queries, keys, values, and attention outputs efficiently.
889scaled_dot_product_fused_attention_overrideableNeuralNetworkAlpha5.4aten, Attention, KernelGenComputes scaled dot product attention on query, key and value tensors, using an optional attention mask if passed. Returns extra values including logsumexp, cumulative sequence lengths, max sequence lengths, philox RNG state, and debug attention mask.
890scaled_grouped_mmNeuralNetworkBeta5.4MoEPerforms grouped matrix multiplication with row-wise and column-wise scaling, commonly used by quantized Mixture-of-Experts (MoE) workloads.
891scaled_mmBLASBeta5.4atenPerforms a scaled matrix multiplication. The result of self @ mat2 is multiplied by scale_a and scale_b, then an optional bias is added.
892scaled_mm_outBLASBeta5.4atenA variant of _scaled_mm that writes the result into out.
893scaled_softmax_backwardReductionStable4.2atenThe backward pass for a scaled softmax function, commonly used in Scaled Dot-Product Attention (SDPA) within Transformer models, computes the gradient of the loss with respect to the input logits, incorporating a scaling factor to stabilize training.
894scaled_softmax_forwardReductionStable4.2atenThe backward pass for a scaled softmax function, commonly used in Scaled Dot-Product Attention (SDPA) within Transformer models, computes the gradient of the loss with respect to the input logits, incorporating a scaling factor to stabilize training.
895scatter_addTensorAlpha5.4aten, KernelGenOut-of-place version of scatter_add_: adds all values from the tensor src into inp at the indices specified in the index tensor along dimension dim and returns the result without modifying the input.
896scatter_add_TensorStable4.2atenAdds all values from the tensor src into self at the indices specified in the index tensor in a similar fashion as scatter_(). For each value in src, it is added to an index in self which is specified by its index in src for dimension != dim and by the corresponding value in index for dimension = dim.
897scatter_reduceTensorStable2.2atenWrites all values from the tensor src into provided tensor at the indices specified in the index tensor. For each value in src, its output index is specified by its index in src for dimension != dim and by the corresponding value in index for dimension = dim. The optional reduce argument allows specification of an optional reduction operation, which is applied to all values in the tensor src into the tensor at the indices specified in the index.
898scatter_reduce_TensorStable3.0aten, KernelGenThis is the in-place version of scatter_reduce().
899scatter_reduce_two_ReductionAlpha5.3aten, KernelGenA specific low-level ATen operator primarily encountered during model compilation or when using advanced backends like TensorRT or MPS.
900scatter_srcTensorStable2.2atenWrites all values from the tensor src into provided tensor at the indices specified in the index tensor. For each value in src, its output index is specified by its index in src for dimension != dim and by the corresponding value in index for dimension = dim. The optional reduce argument allows specification of an optional reduction operation, which is applied to all values in the tensor src into the tensor at the indices specified in the index.
901scatter_src_TensorStable3.0atenThis is the in-place version of scatter_src().
902searchsortedTensorBeta5.4aten, KernelGenFinds insertion indices for tensor values in sorted innermost-dimension boundaries.
903searchsorted_outTensorBeta5.4aten, KernelGenA variant of searchsorted.Tensor that assigns the result to out.
904searchsorted_scalarTensorBeta5.4aten, KernelGenFinds insertion indices for a scalar value in one-dimensional sorted boundaries.
905searchsorted_scalar_outTensorBeta5.4aten, KernelGenA variant of searchsorted.Scalar that assigns the result to out.
906segment_reduceReductionBeta5.4atenReduces contiguous segments of a tensor along the specified axis. Segments are described by lengths or offsets, and supported reductions include sum, mean, max, min, and prod.
907segment_reduce_backwardReductionBeta5.4atenComputes gradients for segment_reduce with lengths or offsets based segment definitions.
908segment_reduce_backward_outReductionBeta5.4atenA variant of _segment_reduce_backward that assigns the gradient input to the out tensor.
909segment_reduce_outReductionBeta5.4atenA variant of segment_reduce that assigns the reduced segments to the out tensor.
910select_backwardNeuralNetworkBeta5.3atenCalculate the gradient during the backward pass in the neural network.
911select_scatterTensorStable2.2atenEmbeds the values of the src tensor into input at the given index. This function returns a tensor with fresh storage; it does not create a view.
912seluNeuralNetworkStable5.3aten, pointwise, nn.functional, Activation, KernelGenApplies an element-wise activation function that induces self-normalizing properties in neural networks. It scales the Exponential Linear Unit (ELU) to ensure activations remain close to zero mean and unit variance.
913selu_NeuralNetworkStable5.3aten, pointwise, Activation, KernelGenThis is the in-place version of selu.
914sgnMathAlpha5.4aten, KernelGenComputes the sign of each element in the input tensor, element-wise. This function extends sign() to complex tensors.
915sgn_MathStable5.3aten, KernelGenComputes the sign of each element in the self tensor, element-wise. This function is an extension of sign() designed to handle complex tensors in addition to real-valued ones.
916sgn_outMathAlpha5.4aten, KernelGenWrites the element-wise sign of the input tensor to the out tensor. This function extends sign() to complex tensors.
917sigmoidNeuralNetworkStable2.0aten, pointwiseComputes the expit (also known as the logistic sigmoid function) of the elements of input.
918sigmoid_NeuralNetworkStable2.2aten, pointwiseThe in-place version of sigmoid().
919sigmoid_backwardNeuralNetworkStable3.0aten, pointwiseThe backward version of sigmoid().
920signMathAlpha5.4aten, KernelGen, pointwiseReturns a new tensor with the sign of the elements of input. For real numbers: 1 if x > 0, -1 if x < 0, 0 if x == 0, 0 if x is NaN. Complex dtypes are not supported (use torch.sgn instead).
921sign_outMathAlpha5.4aten, KernelGen, pointwiseOut-variant of sign. Writes the element-wise sign to the out tensor.
922signbitTensorBeta5.3aten, pointwiseTests if each element of input has its sign bit set or not.
923signbit_outTensorBeta5.3aten, pointwiseA variant of signbit that assigns the output to out.
924siluNeuralNetworkStable1.0aten, pointwise, nn.functionalSiLU (Sigmoid Linear Unit), a simple approximation of ReLU but without any discontinuity of the first derivative.
925silu_NeuralNetworkStable2.2aten, nn.functional, pointwiseThe in-place version of silu().
926silu_and_mulActivationStable2.0fused, pointwise, vLLMA custom operator in vLLM as activation function for SwiGLU.
927silu_and_mul_outActivationStable2.0fused, pointwise, vLLMA variant of silu_and_mul with an extra out argument.
928silu_and_mul_with_clampActivationStable5.3fused, pointwise, vLLMA custom operator in vLLM as activation function for SwiGLU.
929silu_and_mul_with_clamp_outActivationStable5.3fused, pointwise, vLLMA variant of silu_and_mul_with_clamp with an extra out argument.
930silu_backwardNeuralNetworkStable3.0aten, pointwiseA variant of silu() for backward case.
931sinMathStable2.0aten, pointwiseReturns a new tensor with the sine of the elements in the input tensor, where each value in this input tensor is in radians.
932sin_MathStable2.2aten, pointwiseThe in-place version of sin().
933sincMathAlpha5.4aten, KernelGenComputes the normalized sinc function sin(π·x) / (π·x) for each element of the input.
934sinc_MathAlpha5.4aten, KernelGenThe in-place version of sinc().
935sinhMathAlpha5.4aten, KernelGenComputes the hyperbolic sine (e^x - e^{-x}) / 2 element-wise.
936sinh_MathStable5.3aten, KernelGenComputes the hyperbolic sine (e^x-e^{-x})/2 of each element in a tensor. This is an in-place version.
937skip_layer_normNeuralNetworkStable2.0fused, TransformerAn optimized operation used in Transformer models to improve performance by combining residual connection (skip connection) addition and Layer Normalization (LayerNorm) into a single kernel.
938sliceTensorAlpha5.4aten, KernelGenSlices the input tensor along the specified dimension. Returns a view of the original tensor with the given dimension sliced from start to end.
939slice_backwardNeuralNetworkStable5.0atenAn automatic differentiation (autograd) function that computes the gradient of a tensor slicing operation (tensor[start:end]) during backpropagation.
940slice_scatterTensorStable2.2atenEmbeds the values of the src tensor into input at the given dimension. This function returns a tensor with fresh storage; it does not create a view.
941smooth_l1_lossNeuralNetworkAlpha5.3aten, pointwise, nn.functionalCompute the smooth L1 loss between input and target tensors.
942smooth_l1_loss_backwardNeuralNetworkAlpha5.3aten, pointwise, nn.functionalCompute the gradient of smooth L1 loss with respect to the input tensor.
943soft_margin_lossNeuralNetworkBeta5.3nn.functional, KernelGenCompute the soft margin loss.
944soft_margin_loss_backwardMathAlpha5.4aten, KernelGenBackward pass for soft margin loss, a two-class classification logistic loss between input and target tensors.
945softmaxNeuralNetworkStable1.0aten, nn.functionalApply a softmax function.
946softmax_backwardReductionStable3.0aten, nn.functionalThe backward version of softmax().
947softmax_backward_outReductionStable3.0aten, nn.functionalA variant of softmax_backward().
948softmax_outNeuralNetworkStable1.0aten, nn.functionalApply a softmax function, with given out.
949softplusNeuralNetworkStable4.0aten, nn.functional, pointwiseApplies element-wise, the function Softplus.
950softplus_backwardNeuralNetworkAlpha5.4aten, Activation, pointwiseThe backward version of softplus().
951softshrinkNeuralNetworkStable5.3aten, nn.functional, Activation, KernelGenApplies the soft shrinkage function element-wise to an input tensor. It is an activation function often used in signal processing and sparse representation, such as image denoising.
952softshrink_outNeuralNetworkStable5.3aten, nn.functional, Activation, KernelGenThis is a variant of softshrink that supports an output tensor.
953sortTensorStable2.2aten, skip_precision_checkSorts the elements of the input tensor along a given dimension in ascending order by value.
954sort_stableTensorStable3.0aten, skip_precision_checkSorts the elements of the input tensor along a given dimension in ascending order by value. This is a variant of sort() where stable is set to True to preserve the order of equivalent elements.
955sparse_attn_tritonNeuralNetworkBeta5.3fused, NoCPUSparse attention with attention-sink.
956sparse_mla_fwd_interfaceDSAStable5.3fusedA generic interface for sparse MLA (Multi-head Latent Attention) for DeepSeek v3/v3.2. It is currently not exposed as a standalone operator for use.
957sparse_sampled_addmmBLASAlpha5.4atenComputes alpha * (mat1 @ mat2) * spy(input) + beta * input, where input is a sparse CSR tensor and spy(input) keeps only its sparsity pattern. The result shares the CSR pattern with input.
958sparse_sampled_addmm_outBLASAlpha5.4atenA variant of sparse_sampled_addmm that assigns the result to the provided sparse CSR out tensor.
959sparse_semi_structured_addmmBLASAlpha5.4aten, KernelGenPerforms a matrix multiplication of the sparse semi-structured (2:4 sparsity) matrices mat1 and mat2 with input added.
960sparse_semi_structured_mmBLASaten, KernelGenPerforms a sparse semi-structured matrix multiplication of a dense matrix with a 2:4 sparsity pattern against a dense matrix.
961special_airy_aiMathAlpha5.4aten, pointwise, KernelGenComputes the Airy function Ai(x) for each element of the input tensor.
962special_airy_ai_outMathAlpha5.4aten, pointwise, KernelGenA variant of special_airy_ai with output saved to provided out.
963special_bessel_j0MathAlpha5.4aten, pointwise, KernelGenComputes the Bessel function of the first kind of order 0 for each element of input.
964special_bessel_j1MathAlpha5.4aten, KernelGen, pointwiseComputes the Bessel function of the first kind of order 1 for each element of input.
965special_bessel_y0MathAlpha5.4aten, KernelGen, pointwiseComputes the Bessel function of the second kind of order 0 for each element of input.
966special_bessel_y1MathAlpha5.4aten, KernelGenComputes the Bessel function of the second kind of order 1 (Y_1(x)) for each element in the input tensor.
967special_chebyshev_polynomial_uMathAlpha5.4aten, pointwise, KernelGenComputes the Chebyshev polynomial of the second kind U_n(x) for each element of the input tensor.
968special_chebyshev_polynomial_vMathAlpha5.4aten, KernelGenTriton kernel implementation for special_chebyshev_polynomial_v.
969special_chebyshev_polynomial_wMathAlpha5.4aten, KernelGenComputes the Chebyshev polynomial of the fourth kind :math:W_{n}(\\text{input}).
970special_chebyshev_polynomial_w_outMathAlpha5.4aten, KernelGenA variant of special_chebyshev_polynomial_w() with output saved to provided out.
971special_digammaMathAlpha5.4aten, pointwise, KernelGenComputes the digamma function (logarithmic derivative of the gamma function). Alias for digamma; delegates to the digamma kernel.
972special_erfMathAlpha5.4aten, pointwise, KernelGenComputes the error function of input element-wise. Alias for erf; delegates to the erf kernel.
973special_erfcMathAlpha5.4aten, pointwise, KernelGenComputes the complementary error function.
974special_erfcxMathAlpha5.4aten, pointwise, KernelGenComputes the scaled complementary error function for each element of input.
975special_erfinvMathAlpha5.4aten, KernelGenComputes the inverse error function element-wise. Maps values in (-1, 1) to the real line.
976special_erfinv_MathAlpha5.4aten, KernelGenThe in-place version of special_erfinv().
977special_erfinv_outMathAlpha5.4aten, KernelGenA variant of special_erfinv() that assigns the output to the out tensor.
978special_exp2MathAlpha5.4aten, pointwise, KernelGenComputes the base two exponential function of input element-wise. Alias for exp2; delegates to the exp2 kernel.
979special_expitMathAlpha5.4aten, pointwise, KernelGenComputes the expit (sigmoid) function element-wise. Returns 1 / (1 + exp(-x)).
980special_gammaincMathAlpha5.4aten, KernelGenComputes the regularized lower incomplete gamma function P(a, x) element-wise.
981special_gammainccMathAlpha5.4aten, KernelGen, pointwiseComputes the regularized upper incomplete gamma function Q(a, x) element-wise. Out-of-place counterpart of igammac_; delegates to the igammac kernel.
982special_gammalnMathAlpha5.4aten, KernelGenComputes the logarithm of the absolute value of the Gamma function on input.
983special_gammaln_outMathAlpha5.4aten, KernelGenOut-of-place variant of special_gammaln that writes the result to the out tensor.
984special_hermite_polynomial_hMathAlpha5.1aten, KernelGenComputes the physicist's Hermite polynomial H_n(x) of degree n at input.
985special_i0eMathStable5.3aten, pointwise, KernelGenComputes the exponentially scaled zeroth order modified Bessel function of the first kind for each element of input.
986special_i0e_outMathStable5.3aten, pointwise, KernelGenA variant of special_i0e() with output saved to provided out..
987special_i1MathStable5.3aten, pointwise, KernelGenComputes the modified Bessel function of the first kind of order 1 (I_1(x)) for each element in the input tensor, designed for special mathematical functions.
988special_i1_outMathStable5.3aten, pointwise, KernelGenA variant of special_i1 that allows the output to be assigned to another tensor.
989special_i1eMathAlpha5.4aten, pointwise, KernelGenComputes the exponentially scaled modified Bessel function of the first kind of order 1, i1e(x) = i1(x) * exp(-|x|), for each element in the input tensor.
990special_i1e_outMathAlpha5.4aten, pointwise, KernelGenA variant of special_i1e() that saves the output to the provided out tensor.
991special_legendre_polynomial_pMathAlpha5.4aten, KernelGenComputes the Legendre polynomial P_n(x) of degree n for each element in the input tensor, designed for special mathematical functions.
992special_log1pMathAlpha5.4aten, KernelGenComputes log(1 + x) for each element of the input tensor.
993special_log1p_outMathAlpha5.4aten, nn.functional, KernelGenA variant of special.log1p that allows the output to be assigned to another tensor.
994special_log_ndtrMathAlpha5.4aten, pointwise, KernelGenComputes the log of the cumulative distribution function of the standard normal distribution (log_ndtr(x)) for each element in the input tensor.
995special_log_softmaxMathAlpha5.4aten, nn.functional, KernelGenApplies the log of the softmax function along a given dimension.
996special_logitMathAlpha5.4aten, KernelGenReturns a new tensor with the logit of the elements of input. Input is clamped to [eps, 1 - eps] when eps is not None.
997special_logit_outMathAlpha5.4aten, KernelGenA variant of special_logit that allows the output to be assigned to another tensor.
998special_logsumexpReductionAlpha5.4aten, KernelGenComputes the natural logarithm of the sum of exponentials of each row of the input tensor in the given dimension dim.
999special_modified_bessel_i0MathAlpha5.4aten, pointwise, KernelGenComputes the modified Bessel function of the first kind of order 0 (I_0(x)) for each element in the input tensor, designed for special mathematical functions.
1000special_modified_bessel_i0_outMathAlpha5.4aten, pointwise, KernelGenA variant of special_modified_bessel_i0 that writes the modified Bessel function of the first kind of order 0 (I_0(x)) into a preallocated output tensor.
1001special_modified_bessel_k0MathAlpha5.4aten, KernelGenTriton kernel implementation for special_modified_bessel_k0.
1002special_modified_bessel_k0_outMathAlpha5.4aten, KernelGenTriton kernel implementation for special_modified_bessel_k0.out.
1003special_modified_bessel_k1MathAlpha5.4aten, KernelGenModified Bessel function of the second kind of order 1.
1004special_modified_bessel_k1_outMathAlpha5.4aten, KernelGenA variant of special_modified_bessel_k1 that allows the output to be assigned to another tensor.
1005special_multigammalnMathAlpha5.4aten, KernelGenComputes the multivariate log-gamma function for each element of the input. Out-of-place counterpart of mvlgamma_; delegates to the mvlgamma kernel, which supports orders p up to 12.
1006special_ndtrMathAlpha5.4aten, KernelGenComputes the standard normal cumulative distribution function (CDF).
1007special_ndtriMathAlpha5.4aten, pointwise, KernelGenComputes the argument x for which the area under the standard Gaussian probability density function, integrated from minus infinity to x, equals input. In other words it is the inverse of the standard normal CDF.
1008special_roundMathAlpha5.4aten, KernelGenRounds a tensor to the nearest integer, with ties rounding to the nearest even value (banker's rounding).
1009special_round_outMathAlpha5.4aten, KernelGenA variant of special_round() with output saved to provided out.
1010special_scaled_modified_bessel_k1MathAlpha5.4aten, pointwiseComputes the scaled modified Bessel function of the first kind of order 1 (scaled K_1(x) = exp(x)*K_1(x)) for each element in the input tensor.
1011special_shifted_chebyshev_polynomial_tMathAlpha5.4aten, KernelGenComputes the shifted Chebyshev polynomial of the first kind T_n^*(x) for each pair of elements in x and n.
1012special_shifted_chebyshev_polynomial_uMathAlpha5.4aten, KernelGenComputes the shifted Chebyshev polynomial of the second kind U_n^*(x) for each pair of elements in x and n.
1013special_shifted_chebyshev_polynomial_u_Mathaten, KernelGenInplace variant of special_shifted_chebyshev_polynomial_u.
1014special_shifted_chebyshev_polynomial_vMathAlpha5.4aten, pointwise, KernelGenComputes the shifted Chebyshev polynomial of the third kind V_n^*(x).
1015special_shifted_chebyshev_polynomial_wMathAlpha5.4aten, KernelGenComputes the shifted Chebyshev polynomial of the second kind W_n(x).
1016special_sincMathAlpha5.4aten, KernelGenComputes sin(pi * x) / (pi * x) for each element of input.
1017special_softmaxMathAlpha5.4aten, reduction, KernelGenComputes the softmax of input along dimension dim; alias of torch.nn.functional.softmax exposed under torch.special.
1018special_xlog1pyMathAlpha5.4aten, KernelGenComputes x * log1p(y) for each element pair in the input tensors x and y.
1019special_xlogyMathAlpha5.4aten, KernelGenComputes x * log(y) for each element pair in the input tensors x and y. Returns 0 if x == 0.
1020special_xlogy_MathAlpha5.4aten, KernelGenThe in-place version of special_xlogy().
1021split_with_sizes_copyMathAlpha5.4aten, KernelGenSplits a tensor into sub-tensors along a given dimension, where each split has a specified size. Always returns copies.
1022sqrtMathStable4.0aten, pointwiseReturns a new tensor with the square-root of the elements of input.
1023sqrt_MathStable4.0aten, pointwiseThis is the in-place version of sqrt().
1024squareMathBeta5.3aten, pointwiseReturns a new tensor with the square of the elements of input.
1025square_MathBeta5.3aten, pointwiseThe inplace version of square.
1026square_outMathBeta5.3aten, pointwiseA variant of square that assigns the output to the provided out.
1027squeeze_copyMathAlpha5.4aten, KernelGenTriton kernel implementation for squeeze_copy.
1028stackTensorStable2.2atenConcatenates a sequence of tensors along a new dimension.
1029stage_deepseek_v4_mega_moe_inputsNeuralNetworkAlpha5.4fused, DeepSeekV4, MoEStages DeepSeekV4 Mega-MoE inputs by quantizing hidden states to FP8 with UE8M0 scales and copying top-k indices and weights to output buffers.
1030stdReductionStable4.0atenCalculates the standard deviation over the dimensions specified by dim. dim can be a single dimension, list of dimensions, or None to reduce over all dimensions.
1031subMathStable1.0aten, pointwiseSubtracts other, scaled by alpha, from the input tensor.
1032sub_MathStable2.2aten, pointwiseSubtracts other, scaled by alpha, from the input tensor. This is the in-place version.
1033subtractMathAlpha5.4aten, KernelGenSubtracts other, scaled by alpha, from input.
1034subtract_MathAlpha5.4aten, KernelGenIn-place version of subtract. Subtracts other, scaled by alpha, from input.
1035sumLinearAlgStable2.0aten, ReductionReturns the sum of all elements in the input tensor.
1036sum_dimLinearAlgStable2.0aten, ReductionReturns the sum of each row of the input tensor in the given dimension dim. dim is a list of dimensions, reduce over all of them.
1037sum_dim_outLinearAlgStable3.0aten, ReductionA variant of sum_dim() with the out argument.
1038sum_outLinearAlgStable3.0aten, ReductionA variant of sum() with the out argument.
1039sum_to_sizeMathAlpha5.4aten, KernelGenTriton kernel implementation for sum_to_size.
1040swigluNeuralNetworkStable5.0fused, TransformerSwish-Gated Linear Unit, a variant of GLU with the Swish activation function.
1041sym_constrain_rangeTensorAlpha5.4aten, KernelGenConstrains the range of a symbolic integer value. Used for torch.compile symbolic tracing to provide bounds information.
1042sym_storage_offsetTensorAlpha5.4aten, KernelGenReturns the storage offset of the input tensor as an integer. Used for torch.compile symbolic tracing.
1043sym_strideTensorAlpha5.4aten, KernelGenReturns the stride of the input tensor as a tuple of integers. Used for torch.compile symbolic tracing.
1044t_copyTensorStable5.3aten, KernelGenTranspose a 2D tensor into a new tensor with contiguous memory layout.
1045t_copy_outTensorStable5.3aten, KernelGenA variant of t_copy() that allows the output to be assigned to the out tensor.
1046takeTensorAlpha5.4aten, KernelGenReturns a new tensor with the elements of input at the given flat index. The input is treated as a flattened 1-D tensor and the result takes the shape of index. Negative indices are supported.
1047take_outTensorAlpha5.4aten, KernelGenOut-variant of take. Writes the gathered elements into the out tensor, resizing it to the shape of index if needed.
1048tanNeuralNetworkStable4.1aten, pointwiseReturns a new tensor with the tangent of the elements in the input tensor, where each value in this input tensor is in radians.
1049tan_NeuralNetworkStable4.1aten, pointwise, KernelGenThis is the in-place version of tan().
1050tanhMathStable2.0aten, pointwiseReturns a new tensor with the hyperbolic tangent of the elements of input.
1051tanh_MathStable2.2aten, pointwiseThis is the in-place version of tanh().
1052tanh_backwardMathStable3.0aten, pointwiseThis is the backward case for tanh().
1053te_rmsnorm_bwdNeuralNetworkAlpha5.4fused, NormalizationRMSNorm backward pass aligned with TransformerEngine's rmsnorm_bwd signature. Computes gradients dx (w.r.t. input) and dgamma (w.r.t. weight).
1054te_rmsnorm_fwdNeuralNetworkAlpha5.4fused, NormalizationRMSNorm forward pass aligned with TransformerEngine's rmsnorm_fwd signature. Supports zero_centered_gamma, pre-allocated output tensor, and output dtype conversion. Returns (output, None, rsigma).
1055tensor_splitMathAlpha5.3aten, KernelGenSplits a tensor into multiple sub-tensors along a given dimension, using either a number of sections or a list of split indices.
1056thnn_differentiable_gru_cell_backwardNeuralNetworkAlpha5.4aten, KernelGenComputes differentiable gradients for a fused GRU cell.
1057thnn_fused_lstm_cellNeuralNetworkAlpha5.4aten, KernelGenFused LSTM cell computation with optional input and hidden biases.
1058thnn_fused_lstm_cell_backward_implMathAlpha5.4aten, KernelGenComputes gradients for the fused LSTM cell operation (_thnn_fused_lstm_cell).
1059thresholdNeuralNetworkStable3.0aten, nn.functional, pointwiseApply a threshold to each element of the input Tensor.
1060threshold_MathAlpha5.4aten, KernelGenIn-place version of threshold(). Thresholds each element of the input Tensor.
1061threshold_backwardNeuralNetworkStable3.0aten, nn.functional, pointwiseThis is the backward version for threshold.
1062tileTensorStable2.1atenConstructs a tensor by repeating the elements of input. The dims argument specifies the number of repetitions in each dimension.
1063to_copyTensorStable5.3aten, pointwise, skip_precision_check
1064top_k_per_row_decodeNeuralNetworkBeta5.3fused, vLLM, DeepSeekV4, KernelGenTriton top-K per row for DeepSeek V4 decode-phase token selection. Radix-select based approach with three dispatch tiers for different vocab sizes.
1065top_k_per_row_prefillNeuralNetworkBeta5.3fused, KernelGenTriton top-K per row for DeepSeek V4 sparse attention prefill phase. Replaces vLLM persistent_topk CUDA kernel with in-place masking + adaptive topk selection.
1066topkTensorStable2.1aten, skip_precision_checkReturns the k largest elements of the given input tensor along a given dimension. If dim is not given, the last dimension of the input is chosen. If largest is False then the k smallest elements are returned.
1067topk_softmaxMoEStable4.0fused, vLLMSelects the k most likely next-token candicates, sets all others to zero, and renormalize the prbabilities of these top candidates.
1068topk_softplus_sqrtMoEBeta5.3fused, KernelGen, vLLMFused softplus + sqrt + top-k selection and optional renormalization for MoE gating in models like DeepSeek-V3/V4.
1069topk_w8a16_fp8TensorAlpha5.4NoCPU, QuantizationReturns the k largest (or smallest) elements of a group-wise FP8 E5M2 tensor along the last dimension, dequantizing with per-group scales.
1070traceReductionStable4.0atenReturns the sum of the elements of the diagonal of the input 2-D matrix.
1071transposeTensorAlpha5.1aten, KernelGenReturns a view of the original tensor with its two dimensions swapped. This is a zero-copy view operation that swaps the sizes and strides of the two specified dimensions.
1072trilBLASStable5.3aten, KernelGenReturns the lower triangular part of an input matrix (or a batch of matrices) and sets all other elements to zero.
1073tril_BLASBeta5.3atenThe in-place version of tril().
1074tril_outBLASBeta5.3atenA variant of tril() that explicitly assigns the output to the out parameter.
1075triton_lighting_indexer_k_tiled_interfaceNeuralNetworkAlpha5.3fused, DSAPart of FP8 MQA framework. It is currently not exposed as an operator for use.
1076triuBLASStable1.0atenReturns the upper triangular part of a matrix (2-D tensor) or batch of matrices input, the other elements of the result tensor out are set to 0.
1077triu_NeuralNetworkStable5.0atenThe in-place version of triu().
1078true_divideMathAlpha5.4aten, KernelGen, pointwisePerforms element-wise true division between two tensors.
1079true_divide_MathAlpha5.4aten, KernelGen, pointwisePerforms element-wise true division in-place between two tensors.
1080truncMathAlpha5.4aten, KernelGenRounds each element of the input tensor to the nearest integer towards zero.
1081trunc_MathAlpha5.4aten, KernelGenThe in-place version of trunc().
1082trunc_divideMathStable2.1atenThe div function with rounding_mode set to trunc.
1083trunc_divide_MathStable2.1atenThe in-place version of trunc_divide.
1084unbindTensorAlpha5.4aten, KernelGen, skip_precision_checkPure layout operation (unbind); returns zero-copy views sharing storage with the input.
1085unbind_copyTensorAlpha5.4aten, KernelGenRemoves a tensor dimension and returns a tuple of all slices along a given dimension as copies.
1086unflattenMathAlpha5.4aten, KernelGenExpand a single dimension into multiple dimensions (inverse of flatten).
1087unfoldMathAlpha5.4aten, KernelGenReturn a view of the tensor with a sliding window along the given dimension.
1088unfold_backwardNeuralNetworkStable5.0aten, nn.functionalAn operator for calculating the gradient of the unfold operation during backpropagation. It takes the gradient of the unfolded output and accumulates it back into the original input shape, reversing sliding local block extraction and resolving overlaps.
1089unfold_copyTensorAlpha5.4aten, KernelGenReturns a view of the original tensor which contains all slices of size from self tensor in the specified dimension with given step between two slices.
1090uniform_DistributionStable2.1aten, skip_precision_checkFills self tensor with numbers sampled from the continuous uniform distribution.
1091unique2TensorStable2.1atenReturns the unique elements of the input tensor. This is an internal PyTorch function.
1092unique_consecutiveDistributionBeta5.3aten, KernelGenEliminates all but the first element from every consecutive group of equivalent elements.
1093unique_dimTensorBeta5.4atenReturns the unique slices of the input tensor along a dimension.
1094unpack_seq_tritonNeuralNetworkBeta5.3fused, vLLM, DeepSeekV4Unpack a packed sequence tensor back to its original variable-length form.
1095unsafe_chunkMathAlpha5.4aten, KernelGenSplit a tensor into a specified number of chunks along a given dimension.
1096unsafe_indexReductionBeta5.4atenIndexes input along each dimension with the given indices, matching aten._unsafe_index. Unlike the safe index operator, it performs no bounds checking and rejects bool / int8 masks.
1097unsafe_masked_indexMathAlpha5.4aten, KernelGenGathers elements from self at given indices where mask is True, filling unmasked positions with a scalar value.
1098unsafe_masked_index_put_accumulateMathAlpha5.4aten, KernelGenReads values and indices tensors, performs masked scatter accumulation on input tensor at specified indices where mask is True. The accumulate variant uses atomic_add instead of replacement.
1099unsafe_split_with_sizesTensorAlpha5.4aten, KernelGenSplits a tensor into sub-tensors along a given dimension, where each split has a specified size. Returns views into the original tensor without safety checks.
1100unsafe_viewTensorAlpha5.4aten, KernelGenCreates a new view of an existing tensor with a different shape without performing safety checks, such as verifying if the tensor is contiguous or if the reshape is valid.
1101unsqueezeTensorAlpha5.4aten, KernelGenReturns a new tensor with a dimension of size one inserted at the specified position.
1102unsqueeze_TensorAlpha5.4aten, KernelGenIn-place version of unsqueeze().
1103upsample_bicubic2dNeuralNetworkStable5.0aten, ReductionA variant of upsample() that has mode set to bicubic.
1104upsample_bicubic2d_aaNeuralNetworkStable2.2aten, ReductionA variant of upsample() that has mode set to bicubic.
1105upsample_bicubic2d_aa_backwardNeuralNetworkStable5.0aten, ReductionA backward case for _upsample_bicubic2d_aa().
1106upsample_bilinear2dNeuralNetworkAlpha5.4aten, KernelGenUpsamples the input, using bilinear mode. The input has to be 4 dimensional, and the output_size is a tuple of 2 ints.
1107upsample_bilinear2d_aaNeuralNetworkAlpha5.1aten, KernelGenUpsamples the input, using bilinear mode with anti-aliasing.
1108upsample_bilinear2d_aa_backwardNeuralNetworkAlpha5.4aten, KernelGenA backward case for _upsample_bilinear2d_aa().
1109upsample_lanczos2d_aaNeuralNetworkAlpha5.4aten, KernelGen, ReductionResizes a 4D image tensor with antialiased Lanczos-3 interpolation.
1110upsample_lanczos2d_aa_outNeuralNetworkAlpha5.4aten, KernelGen, ReductionAn out variant of _upsample_lanczos2d_aa().
1111upsample_lanczos2d_aa_vecNeuralNetworkAlpha5.4aten, KernelGen, ReductionA vector-parameter variant of _upsample_lanczos2d_aa().
1112upsample_linear1dNeuralNetworkBeta5.3atenUpsamples the input, using linear mode. The input has to be 3 dimensional, and the output_size is an optional tuple of ints.
1113upsample_linear1d_backwardNeuralNetworkAlpha5.4aten, ReductionA backward case for upsample_linear1d().
1114upsample_nearest1dNeuralNetworkStable5.0atenUpsamples the input, using nearest neighbours' pixel values. The input has to be 3 dimensional, and the output_size is an optional tuple of ints.
1115upsample_nearest2dNeuralNetworkStable2.2atenUpsamples the input, using nearest neighbours' pixel values. The input has to be 4 dimensional. The scales can be provided with scales_h and scales_w.
1116upsample_nearest3dNeuralNetworkStable5.0atenPerforms 3D nearest-neighbor interpolation to increase the spatial size of volumetric data, such as 5D tensors. It scales up inputs by copying values from the nearest pixel/voxel, without calculating new values through linear interpolation.
1117upsample_nearest_exact1dNeuralNetworkBeta5.0aten, ReductionIncreases the length of a 1D tensor using nearest-neighbor interpolation, ensuring the output aligns with library-standard algorithms like PIL.
1118upsample_nearest_exact1d_backwardNeuralNetworkAlpha5.4aten, KernelGen, ReductionA backward case for _upsample_nearest_exact1d().
1119upsample_nearest_exact1d_backward_grad_inputNeuralNetworkAlpha5.4aten, KernelGen, ReductionAn out variant of _upsample_nearest_exact1d_backward().
1120upsample_nearest_exact2dMathStable5.3aten, KernelGenTriton kernel implementation for _upsample_nearest_exact2d.
1121upsample_nearest_exact2d_backwardNeuralNetworkAlpha5.4aten, KernelGenA backward case for _upsample_nearest_exact2d().
1122upsample_nearest_exact3dNeuralNetworkAlpha5.1aten, NeuralNetwork, KernelGenIncreases the depth, height, and width of a 3D tensor using nearest-neighbor interpolation, ensuring the output aligns with library-standard algorithms like PIL.
1123upsample_trilinear3dMathAlpha5.4aten, KernelGenUpsamples the input (NCDHW) using trilinear interpolation.
1124value_selecting_reduction_backwardReductionAlpha5.4aten, KernelGenComputes the gradient of a value-selecting reduction (such as max.dim or min.dim) by scattering the upstream gradient to the positions given by the indices of the selected values in the original input.
1125varTensorBeta5.3aten, KernelGenCalculates the variance over all dimensions.
1126var_correctionTensorBeta5.3atenA variant of the var() operator, with an optional correction for specifying difference between the sample size and sample degrees of freedom.
1127var_dimTensorBeta5.3atenCalculates the variance over the dimensions specified by dim.
1128var_meanLinearAlgStable2.0aten, ReductionCalculates the variance and mean over the dimensions specified by dim. dim can be a single dimension, list of dimensions, or None to reduce over all dimensions.
1129vdotBLASStable2.2atenComputes the dot product of two 1D vectors along a dimension.
1130vector_normLinearAlg NeuralNetworkStable2.0aten, ReductionComputes a vector norm.
1131view_as_complexTensorAlpha5.4atenReturns a view of the real input tensor as a complex tensor (zero-copy view operation).
1132view_copyTensorAlpha5.4aten, KernelGenReturns a copy of the tensor with the specified shape. Unlike view(), this always returns a new tensor rather than an alias.
1133vstackTensorStable2.2atenStack tensors in sequence vertically (row wise).
1134w8a8_block_fp8_matmulBLASAlpha5.3vLLMPerforms matrix multiplication with block-wise quantization.
1135weight_int4pack_mm_with_scales_and_zerosBLASAlpha5.4aten, KernelGenInt4 quantized matrix multiplication with separate scales and zeros. Computes C = matmul(A, W_deq) where W is a packed uint8 int4 weight matrix and A is the activation matrix, using per-group scale and zero-point quantization.
1136weight_int8pack_mmBLASAlpha5.4aten, KernelGenPerforms int8 weight-only quantized matrix multiplication with per-channel scales.
1137weight_normNeuralNetworkStable3.0fusedReparameterizes a module's weight tensor by decoupling its magnitude (g) from its direction (v). It is a hook that compute the actual weight before each forward pass.
1138weight_norm_interfaceNeuralNetworkStable2.2aten, fusedApply weight normalization to neural network layers, decoupling the magnitued of a weight tensor from its direction. It is used to stabilize training, particularly for models with small batch sizes.
1139weight_norm_interface_backwardNeuralNetworkStable3.0aten, fusedComputes the gradients for weight normalization during the backward pass. It calculates the necessary derivatives for updating both the magnitude (g) and direction (v) parameters of a weight-normalized layer, based on gradients received from the previous operation.
1140where_selfTensorStable2.1aten, pointwiseReturns a LongTensor. This operation is identical to torch.nonzero(condition, as_tuple=True).
1141where_self_outTensorStable2.2aten, pointwiseThis is a variant of where_self() with an argument out.
1142xlogyMathBeta5.4aten, pointwiseComputes input * log(other) element-wise, returning zero where input is zero (following PyTorch's xlogy semantics).
1143xlogy_MathAlpha5.4aten, KernelGen, pointwiseComputes x * log(y) element-wise in-place on x.
1144xlogy_outMathBeta5.4aten, pointwiseA variant of xlogy that allows the output to be assigned to an out tensor.
1145xlogy_scalar_tensorMathBeta5.4aten, pointwiseA variant of xlogy where input is a scalar.
1146xlogy_scalar_tensor_outMathBeta5.4aten, pointwiseA variant of xlogy where input is a scalar and the output is assigned to an out tensor.
1147xlogy_tensor_scalarMathBeta5.4aten, pointwiseA variant of xlogy where other is a scalar.
1148xlogy_tensor_scalar_outMathBeta5.4aten, pointwiseA variant of xlogy where other is a scalar and the output is assigned to an out tensor.
1149xorMathAlpha5.4aten, pointwise, KernelGenComputes the bitwise XOR of input tensors or a tensor and a scalar (non-inplace).
1150zeroTensorStable5.3aten, KernelGenFills tensor with zeros.
1151zero_TensorStable5.0atenFills self tensor with zeros.
1152zero_outTensorStable5.3aten, KernelGenFills tensor with zeros but assign the output to the out tensor.
1153zerosTensorStable2.1aten, skip_precision_checkReturns a tensor filled with the scalar value 0, with the shape defined by the variable argument size.
1154zeros_likeTensorStable2.1atenReturns a tensor filled with the scalar value 0, with the same size as input.