1#pragma once
2
3// @generated by torchgen/gen.py from MethodOperators.h
4
5#ifdef TORCH_ASSERT_NO_OPERATORS
6#error This change adds a dependency on native_functions.yaml, \
7 meaning the file will need to be re-compiled every time an operator \
8 is changed or added. Consider if your change would be better placed in \
9 another file, or if a more specific header might achieve the same goal. \
10 See NOTE: [Tensor vs. TensorBase]
11#endif
12
13// Forward declarations of any types needed in the operator signatures.
14// We can't directly include these classes because it will cause circular include dependencies.
15// This file is included by TensorBody.h, which defines the Tensor class.
16#include <ATen/core/ATen_fwd.h>
17
18#include <ATen/ops/_addmm_activation_ops.h>
19#include <ATen/ops/_autocast_to_full_precision_ops.h>
20#include <ATen/ops/_autocast_to_reduced_precision_ops.h>
21#include <ATen/ops/_backward_ops.h>
22#include <ATen/ops/_coalesced_ops.h>
23#include <ATen/ops/_conj_ops.h>
24#include <ATen/ops/_conj_physical_ops.h>
25#include <ATen/ops/_dimI_ops.h>
26#include <ATen/ops/_dimV_ops.h>
27#include <ATen/ops/_fw_primal_ops.h>
28#include <ATen/ops/_indices_ops.h>
29#include <ATen/ops/_is_all_true_ops.h>
30#include <ATen/ops/_is_any_true_ops.h>
31#include <ATen/ops/_is_zerotensor_ops.h>
32#include <ATen/ops/_neg_view_ops.h>
33#include <ATen/ops/_nested_tensor_offsets_ops.h>
34#include <ATen/ops/_nested_tensor_size_ops.h>
35#include <ATen/ops/_nested_tensor_strides_ops.h>
36#include <ATen/ops/_nnz_ops.h>
37#include <ATen/ops/_reshape_alias_ops.h>
38#include <ATen/ops/_to_dense_ops.h>
39#include <ATen/ops/_values_ops.h>
40#include <ATen/ops/_version_ops.h>
41#include <ATen/ops/abs_ops.h>
42#include <ATen/ops/absolute_ops.h>
43#include <ATen/ops/acos_ops.h>
44#include <ATen/ops/acosh_ops.h>
45#include <ATen/ops/add_ops.h>
46#include <ATen/ops/addbmm_ops.h>
47#include <ATen/ops/addcdiv_ops.h>
48#include <ATen/ops/addcmul_ops.h>
49#include <ATen/ops/addmm_ops.h>
50#include <ATen/ops/addmv_ops.h>
51#include <ATen/ops/addr_ops.h>
52#include <ATen/ops/adjoint_ops.h>
53#include <ATen/ops/alias_ops.h>
54#include <ATen/ops/align_as_ops.h>
55#include <ATen/ops/align_to_ops.h>
56#include <ATen/ops/all_ops.h>
57#include <ATen/ops/allclose_ops.h>
58#include <ATen/ops/amax_ops.h>
59#include <ATen/ops/amin_ops.h>
60#include <ATen/ops/aminmax_ops.h>
61#include <ATen/ops/and_ops.h>
62#include <ATen/ops/angle_ops.h>
63#include <ATen/ops/any_ops.h>
64#include <ATen/ops/arccos_ops.h>
65#include <ATen/ops/arccosh_ops.h>
66#include <ATen/ops/arcsin_ops.h>
67#include <ATen/ops/arcsinh_ops.h>
68#include <ATen/ops/arctan2_ops.h>
69#include <ATen/ops/arctan_ops.h>
70#include <ATen/ops/arctanh_ops.h>
71#include <ATen/ops/argmax_ops.h>
72#include <ATen/ops/argmin_ops.h>
73#include <ATen/ops/argsort_ops.h>
74#include <ATen/ops/argwhere_ops.h>
75#include <ATen/ops/as_strided_ops.h>
76#include <ATen/ops/as_strided_scatter_ops.h>
77#include <ATen/ops/asin_ops.h>
78#include <ATen/ops/asinh_ops.h>
79#include <ATen/ops/atan2_ops.h>
80#include <ATen/ops/atan_ops.h>
81#include <ATen/ops/atanh_ops.h>
82#include <ATen/ops/baddbmm_ops.h>
83#include <ATen/ops/bernoulli_ops.h>
84#include <ATen/ops/bincount_ops.h>
85#include <ATen/ops/bitwise_and_ops.h>
86#include <ATen/ops/bitwise_left_shift_ops.h>
87#include <ATen/ops/bitwise_not_ops.h>
88#include <ATen/ops/bitwise_or_ops.h>
89#include <ATen/ops/bitwise_right_shift_ops.h>
90#include <ATen/ops/bitwise_xor_ops.h>
91#include <ATen/ops/bmm_ops.h>
92#include <ATen/ops/broadcast_to_ops.h>
93#include <ATen/ops/cauchy_ops.h>
94#include <ATen/ops/ccol_indices_ops.h>
95#include <ATen/ops/ceil_ops.h>
96#include <ATen/ops/chalf_ops.h>
97#include <ATen/ops/cholesky_inverse_ops.h>
98#include <ATen/ops/cholesky_ops.h>
99#include <ATen/ops/cholesky_solve_ops.h>
100#include <ATen/ops/chunk_ops.h>
101#include <ATen/ops/clamp_max_ops.h>
102#include <ATen/ops/clamp_min_ops.h>
103#include <ATen/ops/clamp_ops.h>
104#include <ATen/ops/clip_ops.h>
105#include <ATen/ops/clone_ops.h>
106#include <ATen/ops/coalesce_ops.h>
107#include <ATen/ops/col_indices_ops.h>
108#include <ATen/ops/conj_ops.h>
109#include <ATen/ops/conj_physical_ops.h>
110#include <ATen/ops/contiguous_ops.h>
111#include <ATen/ops/copy_ops.h>
112#include <ATen/ops/copysign_ops.h>
113#include <ATen/ops/corrcoef_ops.h>
114#include <ATen/ops/cos_ops.h>
115#include <ATen/ops/cosh_ops.h>
116#include <ATen/ops/count_nonzero_ops.h>
117#include <ATen/ops/cov_ops.h>
118#include <ATen/ops/cross_ops.h>
119#include <ATen/ops/crow_indices_ops.h>
120#include <ATen/ops/cummax_ops.h>
121#include <ATen/ops/cummin_ops.h>
122#include <ATen/ops/cumprod_ops.h>
123#include <ATen/ops/cumsum_ops.h>
124#include <ATen/ops/data_ops.h>
125#include <ATen/ops/deg2rad_ops.h>
126#include <ATen/ops/dense_dim_ops.h>
127#include <ATen/ops/dequantize_ops.h>
128#include <ATen/ops/det_ops.h>
129#include <ATen/ops/detach_ops.h>
130#include <ATen/ops/diag_embed_ops.h>
131#include <ATen/ops/diag_ops.h>
132#include <ATen/ops/diagflat_ops.h>
133#include <ATen/ops/diagonal_ops.h>
134#include <ATen/ops/diagonal_scatter_ops.h>
135#include <ATen/ops/diff_ops.h>
136#include <ATen/ops/digamma_ops.h>
137#include <ATen/ops/dist_ops.h>
138#include <ATen/ops/div_ops.h>
139#include <ATen/ops/divide_ops.h>
140#include <ATen/ops/dot_ops.h>
141#include <ATen/ops/dsplit_ops.h>
142#include <ATen/ops/eq_ops.h>
143#include <ATen/ops/equal_ops.h>
144#include <ATen/ops/erf_ops.h>
145#include <ATen/ops/erfc_ops.h>
146#include <ATen/ops/erfinv_ops.h>
147#include <ATen/ops/exp2_ops.h>
148#include <ATen/ops/exp_ops.h>
149#include <ATen/ops/expand_as_ops.h>
150#include <ATen/ops/expand_ops.h>
151#include <ATen/ops/expm1_ops.h>
152#include <ATen/ops/exponential_ops.h>
153#include <ATen/ops/fill_diagonal_ops.h>
154#include <ATen/ops/fill_ops.h>
155#include <ATen/ops/fix_ops.h>
156#include <ATen/ops/flatten_ops.h>
157#include <ATen/ops/flip_ops.h>
158#include <ATen/ops/fliplr_ops.h>
159#include <ATen/ops/flipud_ops.h>
160#include <ATen/ops/float_power_ops.h>
161#include <ATen/ops/floor_divide_ops.h>
162#include <ATen/ops/floor_ops.h>
163#include <ATen/ops/fmax_ops.h>
164#include <ATen/ops/fmin_ops.h>
165#include <ATen/ops/fmod_ops.h>
166#include <ATen/ops/frac_ops.h>
167#include <ATen/ops/frexp_ops.h>
168#include <ATen/ops/gather_ops.h>
169#include <ATen/ops/gcd_ops.h>
170#include <ATen/ops/ge_ops.h>
171#include <ATen/ops/geometric_ops.h>
172#include <ATen/ops/geqrf_ops.h>
173#include <ATen/ops/ger_ops.h>
174#include <ATen/ops/greater_equal_ops.h>
175#include <ATen/ops/greater_ops.h>
176#include <ATen/ops/gt_ops.h>
177#include <ATen/ops/hardshrink_backward_ops.h>
178#include <ATen/ops/hardshrink_ops.h>
179#include <ATen/ops/heaviside_ops.h>
180#include <ATen/ops/histc_ops.h>
181#include <ATen/ops/histogram_ops.h>
182#include <ATen/ops/hsplit_ops.h>
183#include <ATen/ops/hypot_ops.h>
184#include <ATen/ops/i0_ops.h>
185#include <ATen/ops/igamma_ops.h>
186#include <ATen/ops/igammac_ops.h>
187#include <ATen/ops/index_add_ops.h>
188#include <ATen/ops/index_copy_ops.h>
189#include <ATen/ops/index_fill_ops.h>
190#include <ATen/ops/index_ops.h>
191#include <ATen/ops/index_put_ops.h>
192#include <ATen/ops/index_reduce_ops.h>
193#include <ATen/ops/index_select_ops.h>
194#include <ATen/ops/indices_ops.h>
195#include <ATen/ops/inner_ops.h>
196#include <ATen/ops/int_repr_ops.h>
197#include <ATen/ops/inverse_ops.h>
198#include <ATen/ops/is_coalesced_ops.h>
199#include <ATen/ops/is_complex_ops.h>
200#include <ATen/ops/is_conj_ops.h>
201#include <ATen/ops/is_distributed_ops.h>
202#include <ATen/ops/is_floating_point_ops.h>
203#include <ATen/ops/is_inference_ops.h>
204#include <ATen/ops/is_leaf_ops.h>
205#include <ATen/ops/is_neg_ops.h>
206#include <ATen/ops/is_nonzero_ops.h>
207#include <ATen/ops/is_pinned_ops.h>
208#include <ATen/ops/is_same_size_ops.h>
209#include <ATen/ops/is_set_to_ops.h>
210#include <ATen/ops/is_signed_ops.h>
211#include <ATen/ops/isclose_ops.h>
212#include <ATen/ops/isfinite_ops.h>
213#include <ATen/ops/isinf_ops.h>
214#include <ATen/ops/isnan_ops.h>
215#include <ATen/ops/isneginf_ops.h>
216#include <ATen/ops/isposinf_ops.h>
217#include <ATen/ops/isreal_ops.h>
218#include <ATen/ops/istft_ops.h>
219#include <ATen/ops/item_ops.h>
220#include <ATen/ops/kron_ops.h>
221#include <ATen/ops/kthvalue_ops.h>
222#include <ATen/ops/lcm_ops.h>
223#include <ATen/ops/ldexp_ops.h>
224#include <ATen/ops/le_ops.h>
225#include <ATen/ops/lerp_ops.h>
226#include <ATen/ops/less_equal_ops.h>
227#include <ATen/ops/less_ops.h>
228#include <ATen/ops/lgamma_ops.h>
229#include <ATen/ops/log10_ops.h>
230#include <ATen/ops/log1p_ops.h>
231#include <ATen/ops/log2_ops.h>
232#include <ATen/ops/log_normal_ops.h>
233#include <ATen/ops/log_ops.h>
234#include <ATen/ops/log_softmax_ops.h>
235#include <ATen/ops/logaddexp2_ops.h>
236#include <ATen/ops/logaddexp_ops.h>
237#include <ATen/ops/logcumsumexp_ops.h>
238#include <ATen/ops/logdet_ops.h>
239#include <ATen/ops/logical_and_ops.h>
240#include <ATen/ops/logical_not_ops.h>
241#include <ATen/ops/logical_or_ops.h>
242#include <ATen/ops/logical_xor_ops.h>
243#include <ATen/ops/logit_ops.h>
244#include <ATen/ops/logsumexp_ops.h>
245#include <ATen/ops/lshift_ops.h>
246#include <ATen/ops/lt_ops.h>
247#include <ATen/ops/lu_solve_ops.h>
248#include <ATen/ops/mH_ops.h>
249#include <ATen/ops/mT_ops.h>
250#include <ATen/ops/masked_fill_ops.h>
251#include <ATen/ops/masked_scatter_ops.h>
252#include <ATen/ops/masked_select_ops.h>
253#include <ATen/ops/matmul_ops.h>
254#include <ATen/ops/matrix_H_ops.h>
255#include <ATen/ops/matrix_exp_ops.h>
256#include <ATen/ops/matrix_power_ops.h>
257#include <ATen/ops/max_ops.h>
258#include <ATen/ops/maximum_ops.h>
259#include <ATen/ops/mean_ops.h>
260#include <ATen/ops/median_ops.h>
261#include <ATen/ops/min_ops.h>
262#include <ATen/ops/minimum_ops.h>
263#include <ATen/ops/mm_ops.h>
264#include <ATen/ops/mode_ops.h>
265#include <ATen/ops/moveaxis_ops.h>
266#include <ATen/ops/movedim_ops.h>
267#include <ATen/ops/msort_ops.h>
268#include <ATen/ops/mul_ops.h>
269#include <ATen/ops/multinomial_ops.h>
270#include <ATen/ops/multiply_ops.h>
271#include <ATen/ops/mv_ops.h>
272#include <ATen/ops/mvlgamma_ops.h>
273#include <ATen/ops/nan_to_num_ops.h>
274#include <ATen/ops/nanmean_ops.h>
275#include <ATen/ops/nanmedian_ops.h>
276#include <ATen/ops/nanquantile_ops.h>
277#include <ATen/ops/nansum_ops.h>
278#include <ATen/ops/narrow_copy_ops.h>
279#include <ATen/ops/narrow_ops.h>
280#include <ATen/ops/ne_ops.h>
281#include <ATen/ops/neg_ops.h>
282#include <ATen/ops/negative_ops.h>
283#include <ATen/ops/new_empty_ops.h>
284#include <ATen/ops/new_empty_strided_ops.h>
285#include <ATen/ops/new_full_ops.h>
286#include <ATen/ops/new_ones_ops.h>
287#include <ATen/ops/new_zeros_ops.h>
288#include <ATen/ops/nextafter_ops.h>
289#include <ATen/ops/nonzero_numpy_ops.h>
290#include <ATen/ops/nonzero_ops.h>
291#include <ATen/ops/norm_ops.h>
292#include <ATen/ops/normal_ops.h>
293#include <ATen/ops/not_equal_ops.h>
294#include <ATen/ops/numpy_T_ops.h>
295#include <ATen/ops/or_ops.h>
296#include <ATen/ops/orgqr_ops.h>
297#include <ATen/ops/ormqr_ops.h>
298#include <ATen/ops/outer_ops.h>
299#include <ATen/ops/output_nr_ops.h>
300#include <ATen/ops/permute_ops.h>
301#include <ATen/ops/pin_memory_ops.h>
302#include <ATen/ops/pinverse_ops.h>
303#include <ATen/ops/polygamma_ops.h>
304#include <ATen/ops/positive_ops.h>
305#include <ATen/ops/pow_ops.h>
306#include <ATen/ops/prelu_ops.h>
307#include <ATen/ops/prod_ops.h>
308#include <ATen/ops/put_ops.h>
309#include <ATen/ops/q_per_channel_axis_ops.h>
310#include <ATen/ops/q_per_channel_scales_ops.h>
311#include <ATen/ops/q_per_channel_zero_points_ops.h>
312#include <ATen/ops/q_scale_ops.h>
313#include <ATen/ops/q_zero_point_ops.h>
314#include <ATen/ops/qr_ops.h>
315#include <ATen/ops/qscheme_ops.h>
316#include <ATen/ops/quantile_ops.h>
317#include <ATen/ops/rad2deg_ops.h>
318#include <ATen/ops/random_ops.h>
319#include <ATen/ops/ravel_ops.h>
320#include <ATen/ops/reciprocal_ops.h>
321#include <ATen/ops/record_stream_ops.h>
322#include <ATen/ops/refine_names_ops.h>
323#include <ATen/ops/relu_ops.h>
324#include <ATen/ops/remainder_ops.h>
325#include <ATen/ops/rename_ops.h>
326#include <ATen/ops/renorm_ops.h>
327#include <ATen/ops/repeat_interleave_ops.h>
328#include <ATen/ops/repeat_ops.h>
329#include <ATen/ops/requires_grad_ops.h>
330#include <ATen/ops/reshape_as_ops.h>
331#include <ATen/ops/reshape_ops.h>
332#include <ATen/ops/resize_as_ops.h>
333#include <ATen/ops/resize_as_sparse_ops.h>
334#include <ATen/ops/resize_ops.h>
335#include <ATen/ops/resolve_conj_ops.h>
336#include <ATen/ops/resolve_neg_ops.h>
337#include <ATen/ops/retain_grad_ops.h>
338#include <ATen/ops/retains_grad_ops.h>
339#include <ATen/ops/roll_ops.h>
340#include <ATen/ops/rot90_ops.h>
341#include <ATen/ops/round_ops.h>
342#include <ATen/ops/row_indices_ops.h>
343#include <ATen/ops/rshift_ops.h>
344#include <ATen/ops/rsqrt_ops.h>
345#include <ATen/ops/scatter_add_ops.h>
346#include <ATen/ops/scatter_ops.h>
347#include <ATen/ops/scatter_reduce_ops.h>
348#include <ATen/ops/select_ops.h>
349#include <ATen/ops/select_scatter_ops.h>
350#include <ATen/ops/set_data_ops.h>
351#include <ATen/ops/set_ops.h>
352#include <ATen/ops/sgn_ops.h>
353#include <ATen/ops/sigmoid_ops.h>
354#include <ATen/ops/sign_ops.h>
355#include <ATen/ops/signbit_ops.h>
356#include <ATen/ops/sin_ops.h>
357#include <ATen/ops/sinc_ops.h>
358#include <ATen/ops/sinh_ops.h>
359#include <ATen/ops/size_ops.h>
360#include <ATen/ops/slice_ops.h>
361#include <ATen/ops/slice_scatter_ops.h>
362#include <ATen/ops/slogdet_ops.h>
363#include <ATen/ops/smm_ops.h>
364#include <ATen/ops/softmax_ops.h>
365#include <ATen/ops/sort_ops.h>
366#include <ATen/ops/sparse_dim_ops.h>
367#include <ATen/ops/sparse_mask_ops.h>
368#include <ATen/ops/sparse_resize_and_clear_ops.h>
369#include <ATen/ops/sparse_resize_ops.h>
370#include <ATen/ops/split_ops.h>
371#include <ATen/ops/split_with_sizes_ops.h>
372#include <ATen/ops/sqrt_ops.h>
373#include <ATen/ops/square_ops.h>
374#include <ATen/ops/squeeze_ops.h>
375#include <ATen/ops/sspaddmm_ops.h>
376#include <ATen/ops/std_ops.h>
377#include <ATen/ops/stft_ops.h>
378#include <ATen/ops/stride_ops.h>
379#include <ATen/ops/sub_ops.h>
380#include <ATen/ops/subtract_ops.h>
381#include <ATen/ops/sum_ops.h>
382#include <ATen/ops/sum_to_size_ops.h>
383#include <ATen/ops/svd_ops.h>
384#include <ATen/ops/swapaxes_ops.h>
385#include <ATen/ops/swapdims_ops.h>
386#include <ATen/ops/t_ops.h>
387#include <ATen/ops/take_along_dim_ops.h>
388#include <ATen/ops/take_ops.h>
389#include <ATen/ops/tan_ops.h>
390#include <ATen/ops/tanh_ops.h>
391#include <ATen/ops/tensor_split_ops.h>
392#include <ATen/ops/tile_ops.h>
393#include <ATen/ops/to_dense_ops.h>
394#include <ATen/ops/to_mkldnn_ops.h>
395#include <ATen/ops/to_ops.h>
396#include <ATen/ops/to_padded_tensor_ops.h>
397#include <ATen/ops/to_sparse_bsc_ops.h>
398#include <ATen/ops/to_sparse_bsr_ops.h>
399#include <ATen/ops/to_sparse_csc_ops.h>
400#include <ATen/ops/to_sparse_csr_ops.h>
401#include <ATen/ops/to_sparse_ops.h>
402#include <ATen/ops/topk_ops.h>
403#include <ATen/ops/trace_ops.h>
404#include <ATen/ops/transpose_ops.h>
405#include <ATen/ops/triangular_solve_ops.h>
406#include <ATen/ops/tril_ops.h>
407#include <ATen/ops/triu_ops.h>
408#include <ATen/ops/true_divide_ops.h>
409#include <ATen/ops/trunc_ops.h>
410#include <ATen/ops/type_as_ops.h>
411#include <ATen/ops/unbind_ops.h>
412#include <ATen/ops/unflatten_ops.h>
413#include <ATen/ops/unfold_ops.h>
414#include <ATen/ops/uniform_ops.h>
415#include <ATen/ops/unsafe_chunk_ops.h>
416#include <ATen/ops/unsafe_split_ops.h>
417#include <ATen/ops/unsafe_split_with_sizes_ops.h>
418#include <ATen/ops/unsqueeze_ops.h>
419#include <ATen/ops/values_ops.h>
420#include <ATen/ops/var_ops.h>
421#include <ATen/ops/vdot_ops.h>
422#include <ATen/ops/view_as_ops.h>
423#include <ATen/ops/view_ops.h>
424#include <ATen/ops/vsplit_ops.h>
425#include <ATen/ops/where_ops.h>
426#include <ATen/ops/xlogy_ops.h>
427#include <ATen/ops/xor_ops.h>
428#include <ATen/ops/zero_ops.h>
429
430namespace at {
431namespace _ops {
432
433} // namespace _ops
434} // namespace at
435