xref: /aosp_15_r20/external/pytorch/functorch/op_analysis/annotated_ops (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1data, misc
2rename, named
3align_to, named
4align_as, named
5align_tensors, named
6refine_names, named
7dropout, composite pointwise
8feature_dropout, composite pointwise
9alpha_dropout, reduction
10feature_alpha_dropout, reduction
11abs, primitive pointwise
12absolute, alias
13angle, composite pointwise
14view_as_real, complex
15view_as_complex, complex
16sgn, primitive pointwise
17real, complex
18imag, complex
19conj, complex
20conj_physical, complex
21resolve_conj, complex
22acos, primitive pointwise
23arccos, alias
24add, primitive pointwise
25addmv, composite matmul
26addr, composite pointwise
27affine_grid_generator, factory
28all, reduction
29any, reduction
30arange, factory
31argmax, reduction
32argmin, reduction
33acosh, primitive pointwise
34arccosh, alias
35asinh, primitive pointwise
36arcsinh, alias
37atanh, primitive pointwise
38arctanh, alias
39as_strided, view/reshape
40asin, primitive pointwise
41arcsin, alias
42atan, primitive pointwise
43arctan, alias
44atleast_1d, view/reshape
45atleast_2d, view/reshape
46atleast_3d, view/reshape
47baddbmm, composite matmul
48bartlett_window, factory
49bernoulli, factory
50bilinear, composite matmul
51binary_cross_entropy, reduction
52binary_cross_entropy_with_logits, reduction
53bincount, misc
54bitwise_not, primitive pointwise
55copysign, composite pointwise
56logical_not, composite pointwise
57logical_xor, composite pointwise
58logical_and, composite pointwise
59logical_or, composite pointwise
60blackman_window, factory
61bmm, composite matmul
62broadcast_tensors, view/reshape
63broadcast_to, view/reshape
64cat, view/reshape
65block_diag, view/reshape
66ceil, primitive pointwise
67chain_matmul, alias
68unsafe_chunk, view/reshape
69chunk, view/reshape
70tensor_split, view/reshape
71clamp, composite pointwise
72clamp_max, composite pointwise
73clamp_min, composite pointwise
74clip, alias
75complex, complex
76polar, composite pointwise
77constant_pad_nd, view/reshape
78contiguous, view/reshape
79cos, primitive pointwise
80cosh, primitive pointwise
81cosine_embedding_loss, reduction
82count_nonzero, reduction
83cummax, reduction
84cummin, reduction
85cumprod, reduction
86cumsum, reduction
87ctc_loss, misc
88diag_embed, view/reshape
89diagflat, view/reshape
90diagonal, view/reshape
91diff, reduction
92gradient, misc
93div, primitive pointwise
94divide, alias
95true_divide, primitive pointwise
96dot, reduction
97vdot, reduction
98einsum, composite matmul
99embedding, misc
100row_stack, alias
101embedding_bag, reduction
102empty, factory
103new_empty, factory
104new_empty_strided, factory
105new_full, factory
106new_zeros, factory
107new_ones, factory
108empty_quantized, factory
109empty_like, factory
110empty_strided, factory
111erf, primitive pointwise
112erfc, composite pointwise
113exp, primitive pointwise
114exp2, composite pointwise
115expm1, composite pointwise
116expand, view/reshape
117expand_as, view/reshape
118eye, factory
119flatten, view/reshape
120unflatten, view/reshape
121floor, primitive pointwise
122floor_divide, composite pointwise
123frac, primitive pointwise
124full, factory
125full_like, factory
126from_file, misc
127gcd, primitive pointwise
128lcm, composite pointwise
129grid_sampler, misc
130grid_sampler_2d, misc
131grid_sampler_3d, misc
132hann_window, factory
133hamming_window, factory
134kaiser_window, factory
135hinge_embedding_loss, reduction
136group_norm, reduction
137index, scatter/gather
138index_copy, scatter/gather
139index_put, scatter/gather
140instance_norm, reduction
141inverse, linalg
142isclose, composite pointwise
143isnan, composite pointwise
144isreal, composite pointwise
145kl_div, reduction
146kron, composite pointwise
147kthvalue, reduction
148layer_norm, reduction
149nan_to_num, composite pointwise
150linear, composite matmul
151fbgemm_linear_int8_weight_fp32_activation, fbgemm
152fbgemm_linear_int8_weight, fbgemm
153fbgemm_linear_quantize_weight, fbgemm
154fbgemm_pack_gemm_matrix_fp16, fbgemm
155fbgemm_linear_fp16_weight_fp32_activation, fbgemm
156fbgemm_linear_fp16_weight, fbgemm
157fbgemm_pack_quantized_matrix, fbgemm
158ldexp, composite pointwise
159linspace, factory
160log, primitive pointwise
161log10, composite pointwise
162log1p, composite pointwise
163log2, composite pointwise
164logaddexp, reduction
165logaddexp2, reduction
166xlogy, composite pointwise
167logdet, linalg
168logspace, factory
169log_softmax, reduction
170logcumsumexp, reduction
171logsumexp, reduction
172margin_ranking_loss, reduction
173matmul, composite matmul
174matrix_rank, linalg
175matrix_power, alias
176matrix_exp, linalg
177max, reduction
178amax, reduction
179mean, reduction
180median, reduction
181nanmedian, reduction
182min, reduction
183amin, reduction
184mm, composite matmul
185mode, misc
186mul, reduction
187multiply, alias
188mv, composite matmul
189mvlgamma, primitive pointwise
190narrow_copy, view/reshape
191narrow, view/reshape
192ones, factory
193ones_like, factory
194pairwise_distance, reduction
195cdist, reduction
196pdist, reduction
197cosine_similarity, reduction
198permute, scatter/gather
199movedim, view/reshape
200moveaxis, alias
201numpy_T, view/reshape
202pixel_shuffle, view/reshape
203pixel_unshuffle, view/reshape
204channel_shuffle, view/reshape
205pin_memory, misc
206pinverse, linalg
207poisson_nll_loss, reduction
208rad2deg, composite pointwise
209deg2rad, composite pointwise
210scalar_tensor, factory
211rand, factory
212rand_like, factory
213randint, factory
214randint_like, factory
215randn, factory
216randn_like, factory
217randperm, factory
218range, factory
219ravel, view/reshape
220reciprocal, composite pointwise
221neg, composite pointwise
222negative, alias
223repeat, view/reshape
224repeat_interleave, view/reshape
225reshape, view/reshape
226reshape_as, view/reshape
227round, primitive pointwise
228rrelu, composite pointwise
229relu, composite pointwise
230relu6, composite pointwise
231prelu, composite pointwise
232gelu, composite pointwise
233hardshrink, composite pointwise
234rsqrt, composite pointwise
235select, view/reshape
236selu, composite pointwise
237celu, composite pointwise
238silu, composite pointwise
239mish, composite pointwise
240sigmoid, composite pointwise
241logit, composite pointwise
242sin, primitive pointwise
243sinc, composite pointwise
244sinh, primitive pointwise
245detach, misc
246slice, view/reshape
247slogdet, linalg
248smm, sparse
249softmax, reduction
250unsafe_split, view/reshape
251split, view/reshape
252unsafe_split_with_sizes, view/reshape
253split_with_sizes, view/reshape
254hsplit, view/reshape
255vsplit, view/reshape
256dsplit, view/reshape
257squeeze, view/reshape
258sspaddmm, sparse
259stack, view/reshape
260hstack, view/reshape
261vstack, view/reshape
262dstack, view/reshape
263stft, fft
264istft, fft
265sum, reduction
266nansum, reduction
267sum_to_size, reduction
268sqrt, primitive pointwise
269square, composite pointwise
270std, reduction
271std_mean, reduction
272prod, reduction
273t, view/reshape
274tan, primitive pointwise
275tanh, primitive pointwise
276tensordot, composite matmul
277threshold, composite pointwise
278tile, view/reshape
279transpose, view/reshape
280one_hot, scatter/gather
281flip, view/reshape
282fliplr, view/reshape
283flipud, view/reshape
284roll, view/reshape
285rot90, view/reshape
286trapz, composite pointwise
287triplet_margin_loss, reduction
288trunc, composite pointwise
289fix, alias
290type_as, misc
291unique_dim, misc
292unique_consecutive, misc
293unique_dim_consecutive, misc
294unsqueeze, view/reshape
295vander, factory
296var, reduction
297var_mean, reduction
298view_as, view/reshape
299where, misc
300norm_except_dim, reduction
301zeros, factory
302zeros_like, factory
303poisson, factory
304binomial, factory
305norm, reduction
306frexp, composite pointwise
307frobenius_norm, reduction
308nuclear_norm, reduction
309clone, view/reshape
310positive, composite pointwise
311sub, primitive pointwise
312subtract, alias
313rsub, primitive pointwise
314heaviside, composite pointwise
315addmm, composite matmul
316sparse_csr_tensor, sparse
317sparse_coo_tensor, sparse
318sparse_mask, sparse
319to_dense, sparse
320coalesce, sparse
321indices, sparse
322values, sparse
323crow_indices, sparse
324col_indices, sparse
325hspmm, sparse
326unbind, view/reshape
327to_sparse, sparse
328quantize_per_tensor, quantize
329quantize_per_channel, quantize
330dequantize, quantize
331q_per_channel_scales, quantize
332q_per_channel_zero_points, quantize
333int_repr, quantize
334fake_quantize_per_tensor_affine, quantize
335fake_quantize_per_tensor_affine_cachemask, quantize
336fake_quantize_per_channel_affine, quantize
337fake_quantize_per_channel_affine_cachemask, quantize
338choose_qparams_optimized, quantize
339to, misc
340meshgrid, view/reshape
341cartesian_prod, misc
342combinations, misc
343lstm, rnn
344gru, rnn
345rnn_tanh, rnn
346rnn_relu, rnn
347lstm_cell, rnn
348gru_cell, rnn
349rnn_tanh_cell, rnn
350rnn_relu_cell, rnn
351quantized_lstm_cell, rnn
352quantized_gru_cell, rnn
353quantized_rnn_relu_cell, rnn
354quantized_rnn_tanh_cell, rnn
355masked_fill, misc
356masked_scatter, misc
357view, view/reshape
358put, scatter/gather
359index_add, scatter/gather
360index_fill, scatter/gather
361scatter, scatter/gather
362scatter_add, scatter/gather
363bitwise_and, primitive pointwise
364bitwise_or, primitive pointwise
365bitwise_xor, primitive pointwise
366addbmm, composite matmul
367diag, view/reshape
368cross, misc
369triu, view/reshape
370tril, view/reshape
371tril_indices, factory
372triu_indices, factory
373trace, reduction
374ne, composite pointwise
375not_equal, alias
376eq, primitive pointwise
377ge, primitive pointwise
378greater_equal, alias
379le, primitive pointwise
380less_equal, alias
381gt, primitive pointwise
382greater, alias
383lt, primitive pointwise
384less, alias
385take, scatter/gather
386take_along_dim, scatter/gather
387index_select, scatter/gather
388masked_select, scatter/gather
389nonzero, misc
390nonzero_numpy, misc
391gather, scatter/gather
392addcmul, composite pointwise
393addcdiv, composite pointwise
394cross_entropy_loss, reduction
395lstsq, alias
396triangular_solve, linalg
397symeig, linalg
398eig, linalg
399svd, linalg
400swapaxes, alias
401swapdims, alias
402cholesky, linalg
403cholesky_solve, linalg
404solve, linalg
405cholesky_inverse, linalg
406qr, linalg
407geqrf, linalg
408orgqr, alias
409ormqr, linalg
410lu_solve, linalg
411lu_unpack, linalg
412multinomial, misc
413lgamma, primitive pointwise
414digamma, primitive pointwise
415polygamma, primitive pointwise
416erfinv, primitive pointwise
417i0, primitive pointwise
418sign, composite pointwise
419signbit, composite pointwise
420dist, reduction
421atan2, primitive pointwise
422lerp, composite pointwise
423histc, misc
424fmod, primitive pointwise
425hypot, composite pointwise
426igamma, primitive pointwise
427igammac, primitive pointwise
428nextafter, composite pointwise
429remainder, composite pointwise
430fmin, reduction
431fmax, reduction
432maximum, composite pointwise
433minimum, composite pointwise
434quantile, misc
435nanquantile, misc
436sort, misc
437msort, misc
438argsort, misc
439topk, misc
440renorm, reduction
441unfold, misc
442pow, primitive pointwise
443float_power, composite pointwise
444normal, factory
445alias, misc
446bucketize, misc
447searchsorted, misc
448mse_loss, reduction
449l1_loss, reduction
450multi_margin_loss, reduction
451multilabel_margin_loss, reduction
452multilabel_margin_loss_forward, reduction
453nll_loss, reduction
454nll_loss_nd, reduction
455nll_loss_forward, reduction
456nll_loss2d, reduction
457nll_loss2d_forward, reduction
458smooth_l1_loss, reduction
459huber_loss, reduction
460soft_margin_loss, reduction
461elu, composite pointwise
462glu, composite pointwise
463hardsigmoid, composite pointwise
464hardtanh, composite pointwise
465hardswish, composite pointwise
466leaky_relu, composite pointwise
467log_sigmoid, composite pointwise
468log_sigmoid_forward, composite pointwise
469rrelu_with_noise, composite pointwise
470softplus, composite pointwise
471softshrink, composite pointwise
472reflection_pad1d, misc
473reflection_pad2d, misc
474replication_pad1d, misc
475replication_pad2d, misc
476replication_pad3d, misc
477upsample_linear1d, misc
478upsample_bilinear2d, misc
479upsample_trilinear3d, misc
480upsample_bicubic2d, misc
481upsample_nearest1d, misc
482upsample_nearest2d, misc
483upsample_nearest3d, misc
484col2im, misc
485column_stack, view/reshape
486im2col, view/reshape
487isfinite, composite pointwise
488isinf, composite pointwise
489isposinf, composite pointwise
490isneginf, composite pointwise
491special_entr, primitive pointwise
492special_expm1, alias
493special_exp2, alias
494special_gammaln, alias
495special_erf, alias
496special_erfc, alias
497special_erfinv, alias
498special_ndtr, primitive pointwise
499special_xlog1py, composite pointwise
500special_i0, alias
501special_i0e, composite pointwise
502special_i1, primitive pointwise
503special_i1e, composite pointwise
504special_logit, composite pointwise
505special_expit, composite pointwise
506fft_fft, fft
507fft_ifft, fft
508fft_rfft, fft
509fft_irfft, fft
510fft_hfft, fft
511fft_ihfft, fft
512fft_fft2, fft
513fft_ifft2, fft
514fft_rfft2, fft
515fft_irfft2, fft
516fft_fftn, fft
517fft_ifftn, fft
518fft_rfftn, fft
519fft_irfftn, fft
520fft_fftfreq, fft
521fft_rfftfreq, fft
522fft_fftshift, fft
523fft_ifftshift, fft
524linalg_cholesky_ex, alias
525linalg_cholesky, alias
526linalg_det, linalg
527det, alias
528linalg_lstsq, linalg
529linalg_slogdet, alias
530linalg_eig, alias
531linalg_eigvals, alias
532linalg_eigh, linalg
533linalg_eigvalsh, linalg
534linalg_householder_product, linalg
535linalg_inv_ex, alias
536linalg_inv, alias
537inner, reduction
538outer, composite pointwise
539ger, alias
540linalg_norm, reduction
541linalg_vector_norm, reduction
542linalg_matrix_norm, reduction
543linalg_svd, alias
544linalg_svdvals, linalg
545linalg_cond, linalg
546linalg_pinv, alias
547linalg_solve, alias
548linalg_tensorinv, linalg
549linalg_tensorsolve, linalg
550linalg_qr, alias
551linalg_matrix_power, linalg
552linalg_matrix_rank, alias
553linalg_multi_dot, linalg
554segment_reduce, misc
555pad_sequence, misc
556flatten_dense_tensors, misc
557unflatten_dense_tensors, misc
558bitwise_left_shift, primitive pointwise
559bitwise_right_shift, primitive pointwise
560trapezoid, reduction
561special_ndtri, primitive pointwise
562special_psi, primitive pointwise
563special_digamma, primitive pointwise
564special_erfcx, primitive pointwise
565special_xlogy, primitive pointwise
566special_zeta, primitive pointwise
567special_sinc, primitive pointwise
568special_round, primitive pointwise
569special_log1p, primitive pointwise
570isin, reduction
571