1#pragma once
2// @generated by torchgen/gen.py from DispatchKeyFunctions_inl.h
3
4// NB: The implementing C++ file is RegisterDispatchKey.cpp
5
6// The only #includes we need are for custom classes that have defaults in the C++ API
7#include <c10/core/MemoryFormat.h>
8#include <c10/core/Scalar.h>
9#include <ATen/core/Reduction.h>
10
11#if defined(AT_PER_OPERATOR_HEADERS) && defined(TORCH_ASSERT_ONLY_METHOD_OPERATORS)
12#error This change adds a dependency on all pytorch operators, meaning the \
13 file will need to be re-compiled every time an operator is changed or added. \
14 Consider including a specific operator from \
15 <ATen/ops/{my_operator}_meta_dispatch.h>. \
16 See NOTE [TORCH_ASSERT_ONLY_METHOD_OPERATORS].
17#endif
18
19#include <ATen/ops/_add_relu_meta_dispatch.h>
20#include <ATen/ops/_addmm_activation_meta_dispatch.h>
21#include <ATen/ops/_amp_update_scale_meta_dispatch.h>
22#include <ATen/ops/_coalesced_meta_dispatch.h>
23#include <ATen/ops/_convert_indices_from_coo_to_csr_meta_dispatch.h>
24#include <ATen/ops/_convert_indices_from_csr_to_coo_meta_dispatch.h>
25#include <ATen/ops/_efficientzerotensor_meta_dispatch.h>
26#include <ATen/ops/_fused_sdp_choice_meta_dispatch.h>
27#include <ATen/ops/_index_put_impl_meta_dispatch.h>
28#include <ATen/ops/_linalg_det_meta_dispatch.h>
29#include <ATen/ops/_linalg_eigh_meta_dispatch.h>
30#include <ATen/ops/_linalg_slogdet_meta_dispatch.h>
31#include <ATen/ops/_linalg_solve_ex_meta_dispatch.h>
32#include <ATen/ops/_linalg_svd_meta_dispatch.h>
33#include <ATen/ops/_log_softmax_meta_dispatch.h>
34#include <ATen/ops/_log_softmax_backward_data_meta_dispatch.h>
35#include <ATen/ops/_mkldnn_transpose_meta_dispatch.h>
36#include <ATen/ops/_reshape_alias_meta_dispatch.h>
37#include <ATen/ops/_resize_output_meta_dispatch.h>
38#include <ATen/ops/_softmax_meta_dispatch.h>
39#include <ATen/ops/_softmax_backward_data_meta_dispatch.h>
40#include <ATen/ops/_sparse_coo_tensor_with_dims_meta_dispatch.h>
41#include <ATen/ops/_sparse_coo_tensor_with_dims_and_tensors_meta_dispatch.h>
42#include <ATen/ops/_upsample_bicubic2d_aa_meta_dispatch.h>
43#include <ATen/ops/_upsample_bicubic2d_aa_backward_meta_dispatch.h>
44#include <ATen/ops/_upsample_bilinear2d_aa_meta_dispatch.h>
45#include <ATen/ops/_upsample_bilinear2d_aa_backward_meta_dispatch.h>
46#include <ATen/ops/_upsample_nearest_exact1d_meta_dispatch.h>
47#include <ATen/ops/_upsample_nearest_exact1d_backward_meta_dispatch.h>
48#include <ATen/ops/_upsample_nearest_exact2d_meta_dispatch.h>
49#include <ATen/ops/_upsample_nearest_exact2d_backward_meta_dispatch.h>
50#include <ATen/ops/_upsample_nearest_exact3d_meta_dispatch.h>
51#include <ATen/ops/_upsample_nearest_exact3d_backward_meta_dispatch.h>
52#include <ATen/ops/acos_meta_dispatch.h>
53#include <ATen/ops/acosh_meta_dispatch.h>
54#include <ATen/ops/adaptive_max_pool2d_meta_dispatch.h>
55#include <ATen/ops/adaptive_max_pool2d_backward_meta_dispatch.h>
56#include <ATen/ops/adaptive_max_pool3d_meta_dispatch.h>
57#include <ATen/ops/adaptive_max_pool3d_backward_meta_dispatch.h>
58#include <ATen/ops/add_meta_dispatch.h>
59#include <ATen/ops/addbmm_meta_dispatch.h>
60#include <ATen/ops/addcdiv_meta_dispatch.h>
61#include <ATen/ops/addcmul_meta_dispatch.h>
62#include <ATen/ops/addmm_meta_dispatch.h>
63#include <ATen/ops/addmv_meta_dispatch.h>
64#include <ATen/ops/all_meta_dispatch.h>
65#include <ATen/ops/amax_meta_dispatch.h>
66#include <ATen/ops/amin_meta_dispatch.h>
67#include <ATen/ops/aminmax_meta_dispatch.h>
68#include <ATen/ops/any_meta_dispatch.h>
69#include <ATen/ops/arange_meta_dispatch.h>
70#include <ATen/ops/argmax_meta_dispatch.h>
71#include <ATen/ops/argmin_meta_dispatch.h>
72#include <ATen/ops/as_strided_meta_dispatch.h>
73#include <ATen/ops/asin_meta_dispatch.h>
74#include <ATen/ops/asinh_meta_dispatch.h>
75#include <ATen/ops/atan_meta_dispatch.h>
76#include <ATen/ops/atan2_meta_dispatch.h>
77#include <ATen/ops/atanh_meta_dispatch.h>
78#include <ATen/ops/avg_pool2d_meta_dispatch.h>
79#include <ATen/ops/avg_pool2d_backward_meta_dispatch.h>
80#include <ATen/ops/avg_pool3d_meta_dispatch.h>
81#include <ATen/ops/avg_pool3d_backward_meta_dispatch.h>
82#include <ATen/ops/baddbmm_meta_dispatch.h>
83#include <ATen/ops/bernoulli_meta_dispatch.h>
84#include <ATen/ops/bitwise_and_meta_dispatch.h>
85#include <ATen/ops/bitwise_left_shift_meta_dispatch.h>
86#include <ATen/ops/bitwise_not_meta_dispatch.h>
87#include <ATen/ops/bitwise_or_meta_dispatch.h>
88#include <ATen/ops/bitwise_right_shift_meta_dispatch.h>
89#include <ATen/ops/bitwise_xor_meta_dispatch.h>
90#include <ATen/ops/bmm_meta_dispatch.h>
91#include <ATen/ops/cat_meta_dispatch.h>
92#include <ATen/ops/cauchy_meta_dispatch.h>
93#include <ATen/ops/ceil_meta_dispatch.h>
94#include <ATen/ops/clamp_meta_dispatch.h>
95#include <ATen/ops/clamp_max_meta_dispatch.h>
96#include <ATen/ops/clamp_min_meta_dispatch.h>
97#include <ATen/ops/copy_sparse_to_sparse_meta_dispatch.h>
98#include <ATen/ops/copysign_meta_dispatch.h>
99#include <ATen/ops/cos_meta_dispatch.h>
100#include <ATen/ops/cosh_meta_dispatch.h>
101#include <ATen/ops/cumprod_meta_dispatch.h>
102#include <ATen/ops/cumsum_meta_dispatch.h>
103#include <ATen/ops/digamma_meta_dispatch.h>
104#include <ATen/ops/div_meta_dispatch.h>
105#include <ATen/ops/elu_meta_dispatch.h>
106#include <ATen/ops/elu_backward_meta_dispatch.h>
107#include <ATen/ops/embedding_renorm_meta_dispatch.h>
108#include <ATen/ops/empty_meta_dispatch.h>
109#include <ATen/ops/empty_strided_meta_dispatch.h>
110#include <ATen/ops/eq_meta_dispatch.h>
111#include <ATen/ops/erf_meta_dispatch.h>
112#include <ATen/ops/erfc_meta_dispatch.h>
113#include <ATen/ops/erfinv_meta_dispatch.h>
114#include <ATen/ops/exp_meta_dispatch.h>
115#include <ATen/ops/exp2_meta_dispatch.h>
116#include <ATen/ops/expm1_meta_dispatch.h>
117#include <ATen/ops/exponential_meta_dispatch.h>
118#include <ATen/ops/eye_meta_dispatch.h>
119#include <ATen/ops/fill_meta_dispatch.h>
120#include <ATen/ops/floor_meta_dispatch.h>
121#include <ATen/ops/floor_divide_meta_dispatch.h>
122#include <ATen/ops/fmax_meta_dispatch.h>
123#include <ATen/ops/fmin_meta_dispatch.h>
124#include <ATen/ops/fmod_meta_dispatch.h>
125#include <ATen/ops/frac_meta_dispatch.h>
126#include <ATen/ops/fractional_max_pool2d_meta_dispatch.h>
127#include <ATen/ops/fractional_max_pool2d_backward_meta_dispatch.h>
128#include <ATen/ops/fractional_max_pool3d_meta_dispatch.h>
129#include <ATen/ops/gather_meta_dispatch.h>
130#include <ATen/ops/gcd_meta_dispatch.h>
131#include <ATen/ops/ge_meta_dispatch.h>
132#include <ATen/ops/gelu_meta_dispatch.h>
133#include <ATen/ops/gelu_backward_meta_dispatch.h>
134#include <ATen/ops/geometric_meta_dispatch.h>
135#include <ATen/ops/glu_meta_dispatch.h>
136#include <ATen/ops/gt_meta_dispatch.h>
137#include <ATen/ops/hardshrink_meta_dispatch.h>
138#include <ATen/ops/hardshrink_backward_meta_dispatch.h>
139#include <ATen/ops/hardsigmoid_meta_dispatch.h>
140#include <ATen/ops/hardsigmoid_backward_meta_dispatch.h>
141#include <ATen/ops/hardswish_meta_dispatch.h>
142#include <ATen/ops/hardtanh_meta_dispatch.h>
143#include <ATen/ops/heaviside_meta_dispatch.h>
144#include <ATen/ops/hypot_meta_dispatch.h>
145#include <ATen/ops/i0_meta_dispatch.h>
146#include <ATen/ops/igamma_meta_dispatch.h>
147#include <ATen/ops/igammac_meta_dispatch.h>
148#include <ATen/ops/index_meta_dispatch.h>
149#include <ATen/ops/index_add_meta_dispatch.h>
150#include <ATen/ops/index_copy_meta_dispatch.h>
151#include <ATen/ops/index_fill_meta_dispatch.h>
152#include <ATen/ops/index_reduce_meta_dispatch.h>
153#include <ATen/ops/isin_meta_dispatch.h>
154#include <ATen/ops/isneginf_meta_dispatch.h>
155#include <ATen/ops/isposinf_meta_dispatch.h>
156#include <ATen/ops/lcm_meta_dispatch.h>
157#include <ATen/ops/le_meta_dispatch.h>
158#include <ATen/ops/leaky_relu_meta_dispatch.h>
159#include <ATen/ops/leaky_relu_backward_meta_dispatch.h>
160#include <ATen/ops/lerp_meta_dispatch.h>
161#include <ATen/ops/lgamma_meta_dispatch.h>
162#include <ATen/ops/linalg_cholesky_ex_meta_dispatch.h>
163#include <ATen/ops/linalg_cross_meta_dispatch.h>
164#include <ATen/ops/linalg_inv_ex_meta_dispatch.h>
165#include <ATen/ops/linalg_ldl_factor_ex_meta_dispatch.h>
166#include <ATen/ops/linalg_ldl_solve_meta_dispatch.h>
167#include <ATen/ops/linalg_lu_meta_dispatch.h>
168#include <ATen/ops/linalg_lu_factor_ex_meta_dispatch.h>
169#include <ATen/ops/linalg_lu_solve_meta_dispatch.h>
170#include <ATen/ops/linalg_qr_meta_dispatch.h>
171#include <ATen/ops/linalg_vector_norm_meta_dispatch.h>
172#include <ATen/ops/linspace_meta_dispatch.h>
173#include <ATen/ops/log_meta_dispatch.h>
174#include <ATen/ops/log10_meta_dispatch.h>
175#include <ATen/ops/log1p_meta_dispatch.h>
176#include <ATen/ops/log2_meta_dispatch.h>
177#include <ATen/ops/log_normal_meta_dispatch.h>
178#include <ATen/ops/logaddexp_meta_dispatch.h>
179#include <ATen/ops/logaddexp2_meta_dispatch.h>
180#include <ATen/ops/logit_meta_dispatch.h>
181#include <ATen/ops/logit_backward_meta_dispatch.h>
182#include <ATen/ops/logspace_meta_dispatch.h>
183#include <ATen/ops/lshift_meta_dispatch.h>
184#include <ATen/ops/lt_meta_dispatch.h>
185#include <ATen/ops/lu_unpack_meta_dispatch.h>
186#include <ATen/ops/masked_fill_meta_dispatch.h>
187#include <ATen/ops/masked_scatter_meta_dispatch.h>
188#include <ATen/ops/max_meta_dispatch.h>
189#include <ATen/ops/max_pool2d_with_indices_meta_dispatch.h>
190#include <ATen/ops/max_pool2d_with_indices_backward_meta_dispatch.h>
191#include <ATen/ops/maximum_meta_dispatch.h>
192#include <ATen/ops/mean_meta_dispatch.h>
193#include <ATen/ops/min_meta_dispatch.h>
194#include <ATen/ops/minimum_meta_dispatch.h>
195#include <ATen/ops/mish_meta_dispatch.h>
196#include <ATen/ops/mm_meta_dispatch.h>
197#include <ATen/ops/mse_loss_meta_dispatch.h>
198#include <ATen/ops/mul_meta_dispatch.h>
199#include <ATen/ops/ne_meta_dispatch.h>
200#include <ATen/ops/neg_meta_dispatch.h>
201#include <ATen/ops/nextafter_meta_dispatch.h>
202#include <ATen/ops/nll_loss_backward_meta_dispatch.h>
203#include <ATen/ops/nll_loss_forward_meta_dispatch.h>
204#include <ATen/ops/norm_meta_dispatch.h>
205#include <ATen/ops/normal_meta_dispatch.h>
206#include <ATen/ops/polygamma_meta_dispatch.h>
207#include <ATen/ops/pow_meta_dispatch.h>
208#include <ATen/ops/prod_meta_dispatch.h>
209#include <ATen/ops/put_meta_dispatch.h>
210#include <ATen/ops/random_meta_dispatch.h>
211#include <ATen/ops/range_meta_dispatch.h>
212#include <ATen/ops/reciprocal_meta_dispatch.h>
213#include <ATen/ops/reflection_pad1d_meta_dispatch.h>
214#include <ATen/ops/reflection_pad1d_backward_meta_dispatch.h>
215#include <ATen/ops/reflection_pad3d_meta_dispatch.h>
216#include <ATen/ops/reflection_pad3d_backward_meta_dispatch.h>
217#include <ATen/ops/relu_meta_dispatch.h>
218#include <ATen/ops/remainder_meta_dispatch.h>
219#include <ATen/ops/renorm_meta_dispatch.h>
220#include <ATen/ops/replication_pad1d_meta_dispatch.h>
221#include <ATen/ops/replication_pad1d_backward_meta_dispatch.h>
222#include <ATen/ops/replication_pad2d_meta_dispatch.h>
223#include <ATen/ops/replication_pad3d_meta_dispatch.h>
224#include <ATen/ops/resize_meta_dispatch.h>
225#include <ATen/ops/resize_as_sparse_meta_dispatch.h>
226#include <ATen/ops/round_meta_dispatch.h>
227#include <ATen/ops/rrelu_with_noise_meta_dispatch.h>
228#include <ATen/ops/rshift_meta_dispatch.h>
229#include <ATen/ops/rsqrt_meta_dispatch.h>
230#include <ATen/ops/scatter_meta_dispatch.h>
231#include <ATen/ops/scatter_add_meta_dispatch.h>
232#include <ATen/ops/scatter_reduce_meta_dispatch.h>
233#include <ATen/ops/set_meta_dispatch.h>
234#include <ATen/ops/sgn_meta_dispatch.h>
235#include <ATen/ops/sigmoid_meta_dispatch.h>
236#include <ATen/ops/sigmoid_backward_meta_dispatch.h>
237#include <ATen/ops/sign_meta_dispatch.h>
238#include <ATen/ops/signbit_meta_dispatch.h>
239#include <ATen/ops/silu_meta_dispatch.h>
240#include <ATen/ops/silu_backward_meta_dispatch.h>
241#include <ATen/ops/sin_meta_dispatch.h>
242#include <ATen/ops/sinc_meta_dispatch.h>
243#include <ATen/ops/sinh_meta_dispatch.h>
244#include <ATen/ops/slow_conv_transpose2d_meta_dispatch.h>
245#include <ATen/ops/smooth_l1_loss_meta_dispatch.h>
246#include <ATen/ops/softplus_meta_dispatch.h>
247#include <ATen/ops/softplus_backward_meta_dispatch.h>
248#include <ATen/ops/softshrink_meta_dispatch.h>
249#include <ATen/ops/softshrink_backward_meta_dispatch.h>
250#include <ATen/ops/sort_meta_dispatch.h>
251#include <ATen/ops/sparse_resize_meta_dispatch.h>
252#include <ATen/ops/sparse_resize_and_clear_meta_dispatch.h>
253#include <ATen/ops/special_airy_ai_meta_dispatch.h>
254#include <ATen/ops/special_bessel_j0_meta_dispatch.h>
255#include <ATen/ops/special_bessel_j1_meta_dispatch.h>
256#include <ATen/ops/special_bessel_y0_meta_dispatch.h>
257#include <ATen/ops/special_bessel_y1_meta_dispatch.h>
258#include <ATen/ops/special_chebyshev_polynomial_t_meta_dispatch.h>
259#include <ATen/ops/special_chebyshev_polynomial_u_meta_dispatch.h>
260#include <ATen/ops/special_chebyshev_polynomial_v_meta_dispatch.h>
261#include <ATen/ops/special_chebyshev_polynomial_w_meta_dispatch.h>
262#include <ATen/ops/special_entr_meta_dispatch.h>
263#include <ATen/ops/special_erfcx_meta_dispatch.h>
264#include <ATen/ops/special_hermite_polynomial_h_meta_dispatch.h>
265#include <ATen/ops/special_hermite_polynomial_he_meta_dispatch.h>
266#include <ATen/ops/special_i0e_meta_dispatch.h>
267#include <ATen/ops/special_i1_meta_dispatch.h>
268#include <ATen/ops/special_i1e_meta_dispatch.h>
269#include <ATen/ops/special_laguerre_polynomial_l_meta_dispatch.h>
270#include <ATen/ops/special_legendre_polynomial_p_meta_dispatch.h>
271#include <ATen/ops/special_log_ndtr_meta_dispatch.h>
272#include <ATen/ops/special_modified_bessel_i0_meta_dispatch.h>
273#include <ATen/ops/special_modified_bessel_i1_meta_dispatch.h>
274#include <ATen/ops/special_modified_bessel_k0_meta_dispatch.h>
275#include <ATen/ops/special_modified_bessel_k1_meta_dispatch.h>
276#include <ATen/ops/special_ndtri_meta_dispatch.h>
277#include <ATen/ops/special_scaled_modified_bessel_k0_meta_dispatch.h>
278#include <ATen/ops/special_scaled_modified_bessel_k1_meta_dispatch.h>
279#include <ATen/ops/special_shifted_chebyshev_polynomial_t_meta_dispatch.h>
280#include <ATen/ops/special_shifted_chebyshev_polynomial_u_meta_dispatch.h>
281#include <ATen/ops/special_shifted_chebyshev_polynomial_v_meta_dispatch.h>
282#include <ATen/ops/special_shifted_chebyshev_polynomial_w_meta_dispatch.h>
283#include <ATen/ops/special_spherical_bessel_j0_meta_dispatch.h>
284#include <ATen/ops/special_xlog1py_meta_dispatch.h>
285#include <ATen/ops/special_zeta_meta_dispatch.h>
286#include <ATen/ops/sqrt_meta_dispatch.h>
287#include <ATen/ops/sub_meta_dispatch.h>
288#include <ATen/ops/sum_meta_dispatch.h>
289#include <ATen/ops/tan_meta_dispatch.h>
290#include <ATen/ops/tanh_meta_dispatch.h>
291#include <ATen/ops/tanh_backward_meta_dispatch.h>
292#include <ATen/ops/threshold_meta_dispatch.h>
293#include <ATen/ops/threshold_backward_meta_dispatch.h>
294#include <ATen/ops/topk_meta_dispatch.h>
295#include <ATen/ops/triangular_solve_meta_dispatch.h>
296#include <ATen/ops/tril_meta_dispatch.h>
297#include <ATen/ops/triu_meta_dispatch.h>
298#include <ATen/ops/trunc_meta_dispatch.h>
299#include <ATen/ops/unfold_meta_dispatch.h>
300#include <ATen/ops/uniform_meta_dispatch.h>
301#include <ATen/ops/upsample_bicubic2d_meta_dispatch.h>
302#include <ATen/ops/upsample_bicubic2d_backward_meta_dispatch.h>
303#include <ATen/ops/upsample_bilinear2d_meta_dispatch.h>
304#include <ATen/ops/upsample_bilinear2d_backward_meta_dispatch.h>
305#include <ATen/ops/upsample_linear1d_meta_dispatch.h>
306#include <ATen/ops/upsample_linear1d_backward_meta_dispatch.h>
307#include <ATen/ops/upsample_nearest1d_meta_dispatch.h>
308#include <ATen/ops/upsample_nearest1d_backward_meta_dispatch.h>
309#include <ATen/ops/upsample_nearest2d_meta_dispatch.h>
310#include <ATen/ops/upsample_nearest2d_backward_meta_dispatch.h>
311#include <ATen/ops/upsample_nearest3d_meta_dispatch.h>
312#include <ATen/ops/upsample_nearest3d_backward_meta_dispatch.h>
313#include <ATen/ops/upsample_trilinear3d_meta_dispatch.h>
314#include <ATen/ops/upsample_trilinear3d_backward_meta_dispatch.h>
315#include <ATen/ops/view_meta_dispatch.h>
316#include <ATen/ops/view_as_complex_meta_dispatch.h>
317#include <ATen/ops/view_as_real_meta_dispatch.h>
318#include <ATen/ops/xlogy_meta_dispatch.h>
319#include <ATen/ops/zero_meta_dispatch.h>
320
321
322
323