A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y | ||
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
1 | Op Name | Op Category | CompositeAD | Has Batching Rule | ||||||||||||||||||||||
2 | Ignore Composites | _cast_Byte | private | TRUE | ||||||||||||||||||||||
3 | implemented | total | _cast_Char | private | TRUE | |||||||||||||||||||||
4 | backend | 3 | 10 | _cast_Double | private | TRUE | ||||||||||||||||||||
5 | backward | 38 | 127 | _cast_Float | private | TRUE | ||||||||||||||||||||
6 | batch_norm | 1 | 10 | _cast_Int | private | TRUE | ||||||||||||||||||||
7 | conv | 7 | 76 | _cast_Long | private | TRUE | ||||||||||||||||||||
8 | core alias | 38 | 65 | _cast_Short | private | TRUE | ||||||||||||||||||||
9 | core complex | 6 | 8 | _cast_Half | private | TRUE | ||||||||||||||||||||
10 | core composite matmul | 11 | 12 | _backward | backward | TRUE | ||||||||||||||||||||
11 | core composite pointwise | 92 | 102 | set_data | out | TRUE | ||||||||||||||||||||
12 | core reduction | 71 | 143 | data | core misc | TRUE | ||||||||||||||||||||
13 | core factory | 10 | 72 | is_leaf | non_tensor | TRUE | ||||||||||||||||||||
14 | core fbgemm | 0 | 8 | output_nr | non_tensor | TRUE | ||||||||||||||||||||
15 | core fft | 7 | 20 | _version | private | TRUE | ||||||||||||||||||||
16 | core linalg | 16 | 32 | requires_grad_ | inplace | TRUE | ||||||||||||||||||||
17 | core misc | 37 | 89 | retain_grad | out | TRUE | ||||||||||||||||||||
18 | core named | 0 | 6 | retains_grad | non_tensor | TRUE | ||||||||||||||||||||
19 | core primitive pointwise | 87 | 88 | _fw_primal | private | FALSE | ||||||||||||||||||||
20 | core quantize | 0 | 15 | _make_dual | private | FALSE | ||||||||||||||||||||
21 | core rnn | 0 | 16 | _unpack_dual | private | TRUE | ||||||||||||||||||||
22 | core scatter/gather | 12 | 29 | _new_zeros_with_same_feature_meta | private | FALSE | ||||||||||||||||||||
23 | core sparse | 0 | 17 | rename_ | inplace | TRUE | ||||||||||||||||||||
24 | core unknown | 21 | 38 | rename | core named | TRUE | ||||||||||||||||||||
25 | core view/reshape | 55 | 90 | align_to | core named | TRUE | ||||||||||||||||||||
26 | non_tensor | 10 | 36 | align_to.ellipsis_idx | core named | TRUE | ||||||||||||||||||||
27 | pool | 20 | 41 | align_as | core named | TRUE | ||||||||||||||||||||
28 | private | 14 | 197 | align_tensors | core named | TRUE | ||||||||||||||||||||
29 | inplace | 98 | 313 | _assert_async | private | FALSE | ||||||||||||||||||||
30 | out | 0 | 576 | refine_names | core named | TRUE | ||||||||||||||||||||
31 | Total - inplace/out | 556 | 1347 | _use_cudnn_ctc_loss | private | FALSE | ||||||||||||||||||||
32 | Total | 654 | 2236 | _cudnn_ctc_loss | private | FALSE | ||||||||||||||||||||
33 | _use_cudnn_rnn_flatten_weight | private | TRUE | |||||||||||||||||||||||
34 | _cudnn_rnn_flatten_weight | private | FALSE | |||||||||||||||||||||||
35 | _cudnn_rnn | private | FALSE | |||||||||||||||||||||||
36 | _cudnn_rnn_backward | backward | FALSE | |||||||||||||||||||||||
37 | _cudnn_init_dropout_state | private | FALSE | |||||||||||||||||||||||
38 | _debug_has_internal_overlap | private | TRUE | |||||||||||||||||||||||
39 | _fused_dropout | private | FALSE | |||||||||||||||||||||||
40 | _masked_scale | private | FALSE | |||||||||||||||||||||||
41 | native_dropout | backend | FALSE | |||||||||||||||||||||||
42 | native_dropout_backward | backward | FALSE | |||||||||||||||||||||||
43 | _sobol_engine_draw | private | TRUE | |||||||||||||||||||||||
44 | _sobol_engine_ff_ | inplace | TRUE | |||||||||||||||||||||||
45 | _sobol_engine_scramble_ | inplace | TRUE | |||||||||||||||||||||||
46 | _sobol_engine_initialize_state_ | inplace | TRUE | |||||||||||||||||||||||
47 | _reshape_from_tensor | private | TRUE | |||||||||||||||||||||||
48 | _shape_as_tensor | private | TRUE | |||||||||||||||||||||||
49 | dropout | core composite pointwise | TRUE | |||||||||||||||||||||||
50 | dropout_ | inplace | TRUE | |||||||||||||||||||||||
51 | feature_dropout | core composite pointwise | TRUE | |||||||||||||||||||||||
52 | feature_dropout_ | inplace | TRUE | |||||||||||||||||||||||
53 | alpha_dropout | core reduction | TRUE | |||||||||||||||||||||||
54 | alpha_dropout_ | inplace | TRUE | |||||||||||||||||||||||
55 | feature_alpha_dropout | core reduction | TRUE | |||||||||||||||||||||||
56 | feature_alpha_dropout_ | inplace | TRUE | |||||||||||||||||||||||
57 | abs | core primitive pointwise | FALSE | |||||||||||||||||||||||
58 | abs_ | inplace | FALSE | |||||||||||||||||||||||
59 | abs.out | out | FALSE | |||||||||||||||||||||||
60 | absolute | core alias | TRUE | |||||||||||||||||||||||
61 | absolute_ | inplace | TRUE | |||||||||||||||||||||||
62 | absolute.out | out | TRUE | |||||||||||||||||||||||
63 | angle | core composite pointwise | FALSE | |||||||||||||||||||||||
64 | angle.out | out | FALSE | |||||||||||||||||||||||
65 | view_as_real | core complex | FALSE | |||||||||||||||||||||||
66 | view_as_complex | core complex | FALSE | |||||||||||||||||||||||
67 | sgn | core primitive pointwise | FALSE | |||||||||||||||||||||||
68 | sgn_ | inplace | FALSE | |||||||||||||||||||||||
69 | sgn.out | out | FALSE | |||||||||||||||||||||||
70 | real | core complex | TRUE | |||||||||||||||||||||||
71 | imag | core complex | TRUE | |||||||||||||||||||||||
72 | _conj | private | FALSE | |||||||||||||||||||||||
73 | conj | core complex | TRUE | |||||||||||||||||||||||
74 | _conj_physical | private | FALSE | |||||||||||||||||||||||
75 | conj_physical | core complex | TRUE | |||||||||||||||||||||||
76 | conj_physical.out | out | FALSE | |||||||||||||||||||||||
77 | conj_physical_ | inplace | FALSE | |||||||||||||||||||||||
78 | resolve_conj | core complex | TRUE | |||||||||||||||||||||||
79 | resolve_neg | core unknown | TRUE | |||||||||||||||||||||||
80 | _neg_view | private | FALSE | |||||||||||||||||||||||
81 | acos | core primitive pointwise | FALSE | |||||||||||||||||||||||
82 | acos_ | inplace | FALSE | |||||||||||||||||||||||
83 | acos.out | out | FALSE | |||||||||||||||||||||||
84 | arccos | core alias | TRUE | |||||||||||||||||||||||
85 | arccos_ | inplace | TRUE | |||||||||||||||||||||||
86 | arccos.out | out | TRUE | |||||||||||||||||||||||
87 | avg_pool1d | pool | TRUE | |||||||||||||||||||||||
88 | adaptive_avg_pool1d | pool | TRUE | |||||||||||||||||||||||
89 | adaptive_max_pool1d | pool | TRUE | |||||||||||||||||||||||
90 | add.Tensor | core primitive pointwise | FALSE | |||||||||||||||||||||||
91 | add_.Tensor | inplace | FALSE | |||||||||||||||||||||||
92 | add.out | out | FALSE | |||||||||||||||||||||||
93 | _add_relu.Tensor | private | FALSE | |||||||||||||||||||||||
94 | _add_relu_.Tensor | inplace | FALSE | |||||||||||||||||||||||
95 | _add_relu.out | out | FALSE | |||||||||||||||||||||||
96 | _add_relu.Scalar | private | FALSE | |||||||||||||||||||||||
97 | _add_relu_.Scalar | inplace | FALSE | |||||||||||||||||||||||
98 | add.Scalar | core primitive pointwise | FALSE | |||||||||||||||||||||||
99 | add_.Scalar | inplace | FALSE | |||||||||||||||||||||||
100 | addmv | core composite matmul | FALSE |