CompositeImplicitAutogradFunctions_inl.h 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508
  1. #pragma once
  2. // @generated by torchgen/gen.py from DispatchKeyFunctions_inl.h
  3. // NB: The implementing C++ file is RegisterDispatchKey.cpp
  4. // The only #includes we need are for custom classes that have defaults in the C++ API
  5. #include <c10/core/MemoryFormat.h>
  6. #include <c10/core/Scalar.h>
  7. #include <ATen/core/Reduction.h>
  8. #if defined(AT_PER_OPERATOR_HEADERS) && defined(TORCH_ASSERT_ONLY_METHOD_OPERATORS)
  9. #error This change adds a dependency on all pytorch operators, meaning the \
  10. file will need to be re-compiled every time an operator is changed or added. \
  11. Consider including a specific operator from \
  12. <ATen/ops/{my_operator}_compositeimplicitautograd_dispatch.h>. \
  13. See NOTE [TORCH_ASSERT_ONLY_METHOD_OPERATORS].
  14. #endif
  15. #include <ATen/ops/_add_batch_dim_compositeimplicitautograd_dispatch.h>
  16. #include <ATen/ops/_assert_tensor_metadata_compositeimplicitautograd_dispatch.h>
  17. #include <ATen/ops/_autocast_to_full_precision_compositeimplicitautograd_dispatch.h>
  18. #include <ATen/ops/_autocast_to_reduced_precision_compositeimplicitautograd_dispatch.h>
  19. #include <ATen/ops/_backward_compositeimplicitautograd_dispatch.h>
  20. #include <ATen/ops/_batch_norm_impl_index_compositeimplicitautograd_dispatch.h>
  21. #include <ATen/ops/_batch_norm_impl_index_backward_compositeimplicitautograd_dispatch.h>
  22. #include <ATen/ops/_cast_Byte_compositeimplicitautograd_dispatch.h>
  23. #include <ATen/ops/_cast_Char_compositeimplicitautograd_dispatch.h>
  24. #include <ATen/ops/_cast_Double_compositeimplicitautograd_dispatch.h>
  25. #include <ATen/ops/_cast_Float_compositeimplicitautograd_dispatch.h>
  26. #include <ATen/ops/_cast_Half_compositeimplicitautograd_dispatch.h>
  27. #include <ATen/ops/_cast_Int_compositeimplicitautograd_dispatch.h>
  28. #include <ATen/ops/_cast_Long_compositeimplicitautograd_dispatch.h>
  29. #include <ATen/ops/_cast_Short_compositeimplicitautograd_dispatch.h>
  30. #include <ATen/ops/_choose_qparams_per_tensor_compositeimplicitautograd_dispatch.h>
  31. #include <ATen/ops/_convolution_compositeimplicitautograd_dispatch.h>
  32. #include <ATen/ops/_convolution_double_backward_compositeimplicitautograd_dispatch.h>
  33. #include <ATen/ops/_convolution_mode_compositeimplicitautograd_dispatch.h>
  34. #include <ATen/ops/_cufft_clear_plan_cache_compositeimplicitautograd_dispatch.h>
  35. #include <ATen/ops/_cufft_get_plan_cache_max_size_compositeimplicitautograd_dispatch.h>
  36. #include <ATen/ops/_cufft_get_plan_cache_size_compositeimplicitautograd_dispatch.h>
  37. #include <ATen/ops/_cufft_set_plan_cache_max_size_compositeimplicitautograd_dispatch.h>
  38. #include <ATen/ops/_debug_has_internal_overlap_compositeimplicitautograd_dispatch.h>
  39. #include <ATen/ops/_dim_arange_compositeimplicitautograd_dispatch.h>
  40. #include <ATen/ops/_embedding_bag_backward_compositeimplicitautograd_dispatch.h>
  41. #include <ATen/ops/_embedding_bag_sparse_backward_compositeimplicitautograd_dispatch.h>
  42. #include <ATen/ops/_gather_sparse_backward_compositeimplicitautograd_dispatch.h>
  43. #include <ATen/ops/_grid_sampler_2d_cpu_fallback_backward_compositeimplicitautograd_dispatch.h>
  44. #include <ATen/ops/_has_compatible_shallow_copy_type_compositeimplicitautograd_dispatch.h>
  45. #include <ATen/ops/_is_zerotensor_compositeimplicitautograd_dispatch.h>
  46. #include <ATen/ops/_lu_with_info_compositeimplicitautograd_dispatch.h>
  47. #include <ATen/ops/_nnpack_available_compositeimplicitautograd_dispatch.h>
  48. #include <ATen/ops/_pack_padded_sequence_backward_compositeimplicitautograd_dispatch.h>
  49. #include <ATen/ops/_pad_circular_compositeimplicitautograd_dispatch.h>
  50. #include <ATen/ops/_pad_enum_compositeimplicitautograd_dispatch.h>
  51. #include <ATen/ops/_pad_packed_sequence_compositeimplicitautograd_dispatch.h>
  52. #include <ATen/ops/_remove_batch_dim_compositeimplicitautograd_dispatch.h>
  53. #include <ATen/ops/_reshape_from_tensor_compositeimplicitautograd_dispatch.h>
  54. #include <ATen/ops/_rowwise_prune_compositeimplicitautograd_dispatch.h>
  55. #include <ATen/ops/_saturate_weight_to_fp16_compositeimplicitautograd_dispatch.h>
  56. #include <ATen/ops/_scaled_dot_product_attention_compositeimplicitautograd_dispatch.h>
  57. #include <ATen/ops/_scaled_dot_product_attention_math_compositeimplicitautograd_dispatch.h>
  58. #include <ATen/ops/_shape_as_tensor_compositeimplicitautograd_dispatch.h>
  59. #include <ATen/ops/_sobol_engine_draw_compositeimplicitautograd_dispatch.h>
  60. #include <ATen/ops/_sobol_engine_ff_compositeimplicitautograd_dispatch.h>
  61. #include <ATen/ops/_sobol_engine_initialize_state_compositeimplicitautograd_dispatch.h>
  62. #include <ATen/ops/_sobol_engine_scramble_compositeimplicitautograd_dispatch.h>
  63. #include <ATen/ops/_sparse_bsc_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  64. #include <ATen/ops/_sparse_bsr_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  65. #include <ATen/ops/_sparse_compressed_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  66. #include <ATen/ops/_sparse_coo_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  67. #include <ATen/ops/_sparse_csc_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  68. #include <ATen/ops/_sparse_csr_tensor_unsafe_compositeimplicitautograd_dispatch.h>
  69. #include <ATen/ops/_sparse_log_softmax_compositeimplicitautograd_dispatch.h>
  70. #include <ATen/ops/_sparse_mm_compositeimplicitautograd_dispatch.h>
  71. #include <ATen/ops/_sparse_softmax_compositeimplicitautograd_dispatch.h>
  72. #include <ATen/ops/_sparse_sum_compositeimplicitautograd_dispatch.h>
  73. #include <ATen/ops/_test_ambiguous_defaults_compositeimplicitautograd_dispatch.h>
  74. #include <ATen/ops/_test_autograd_multiple_dispatch_compositeimplicitautograd_dispatch.h>
  75. #include <ATen/ops/_test_check_tensor_compositeimplicitautograd_dispatch.h>
  76. #include <ATen/ops/_test_serialization_subcmul_compositeimplicitautograd_dispatch.h>
  77. #include <ATen/ops/_test_string_default_compositeimplicitautograd_dispatch.h>
  78. #include <ATen/ops/_thnn_differentiable_gru_cell_backward_compositeimplicitautograd_dispatch.h>
  79. #include <ATen/ops/_thnn_differentiable_lstm_cell_backward_compositeimplicitautograd_dispatch.h>
  80. #include <ATen/ops/_thnn_fused_lstm_cell_backward_compositeimplicitautograd_dispatch.h>
  81. #include <ATen/ops/_to_cpu_compositeimplicitautograd_dispatch.h>
  82. #include <ATen/ops/_unpack_dual_compositeimplicitautograd_dispatch.h>
  83. #include <ATen/ops/_upsample_bicubic2d_aa_compositeimplicitautograd_dispatch.h>
  84. #include <ATen/ops/_upsample_bilinear2d_aa_compositeimplicitautograd_dispatch.h>
  85. #include <ATen/ops/_upsample_nearest_exact1d_compositeimplicitautograd_dispatch.h>
  86. #include <ATen/ops/_upsample_nearest_exact2d_compositeimplicitautograd_dispatch.h>
  87. #include <ATen/ops/_upsample_nearest_exact3d_compositeimplicitautograd_dispatch.h>
  88. #include <ATen/ops/_use_cudnn_rnn_flatten_weight_compositeimplicitautograd_dispatch.h>
  89. #include <ATen/ops/_validate_sparse_bsc_tensor_args_compositeimplicitautograd_dispatch.h>
  90. #include <ATen/ops/_validate_sparse_bsr_tensor_args_compositeimplicitautograd_dispatch.h>
  91. #include <ATen/ops/_validate_sparse_compressed_tensor_args_compositeimplicitautograd_dispatch.h>
  92. #include <ATen/ops/_validate_sparse_coo_tensor_args_compositeimplicitautograd_dispatch.h>
  93. #include <ATen/ops/_validate_sparse_csc_tensor_args_compositeimplicitautograd_dispatch.h>
  94. #include <ATen/ops/_validate_sparse_csr_tensor_args_compositeimplicitautograd_dispatch.h>
  95. #include <ATen/ops/_version_compositeimplicitautograd_dispatch.h>
  96. #include <ATen/ops/_weight_norm_compositeimplicitautograd_dispatch.h>
  97. #include <ATen/ops/_weight_norm_differentiable_backward_compositeimplicitautograd_dispatch.h>
  98. #include <ATen/ops/absolute_compositeimplicitautograd_dispatch.h>
  99. #include <ATen/ops/adaptive_avg_pool1d_compositeimplicitautograd_dispatch.h>
  100. #include <ATen/ops/adaptive_avg_pool2d_compositeimplicitautograd_dispatch.h>
  101. #include <ATen/ops/adaptive_avg_pool3d_compositeimplicitautograd_dispatch.h>
  102. #include <ATen/ops/adaptive_max_pool1d_compositeimplicitautograd_dispatch.h>
  103. #include <ATen/ops/adjoint_compositeimplicitautograd_dispatch.h>
  104. #include <ATen/ops/affine_grid_generator_backward_compositeimplicitautograd_dispatch.h>
  105. #include <ATen/ops/align_as_compositeimplicitautograd_dispatch.h>
  106. #include <ATen/ops/align_tensors_compositeimplicitautograd_dispatch.h>
  107. #include <ATen/ops/align_to_compositeimplicitautograd_dispatch.h>
  108. #include <ATen/ops/all_compositeimplicitautograd_dispatch.h>
  109. #include <ATen/ops/alpha_dropout_compositeimplicitautograd_dispatch.h>
  110. #include <ATen/ops/and_compositeimplicitautograd_dispatch.h>
  111. #include <ATen/ops/any_compositeimplicitautograd_dispatch.h>
  112. #include <ATen/ops/arccos_compositeimplicitautograd_dispatch.h>
  113. #include <ATen/ops/arccosh_compositeimplicitautograd_dispatch.h>
  114. #include <ATen/ops/arcsin_compositeimplicitautograd_dispatch.h>
  115. #include <ATen/ops/arcsinh_compositeimplicitautograd_dispatch.h>
  116. #include <ATen/ops/arctan_compositeimplicitautograd_dispatch.h>
  117. #include <ATen/ops/arctan2_compositeimplicitautograd_dispatch.h>
  118. #include <ATen/ops/arctanh_compositeimplicitautograd_dispatch.h>
  119. #include <ATen/ops/argsort_compositeimplicitautograd_dispatch.h>
  120. #include <ATen/ops/argwhere_compositeimplicitautograd_dispatch.h>
  121. #include <ATen/ops/atleast_1d_compositeimplicitautograd_dispatch.h>
  122. #include <ATen/ops/atleast_2d_compositeimplicitautograd_dispatch.h>
  123. #include <ATen/ops/atleast_3d_compositeimplicitautograd_dispatch.h>
  124. #include <ATen/ops/avg_pool1d_compositeimplicitautograd_dispatch.h>
  125. #include <ATen/ops/batch_norm_compositeimplicitautograd_dispatch.h>
  126. #include <ATen/ops/bilinear_compositeimplicitautograd_dispatch.h>
  127. #include <ATen/ops/bitwise_and_compositeimplicitautograd_dispatch.h>
  128. #include <ATen/ops/bitwise_or_compositeimplicitautograd_dispatch.h>
  129. #include <ATen/ops/bitwise_xor_compositeimplicitautograd_dispatch.h>
  130. #include <ATen/ops/broadcast_tensors_compositeimplicitautograd_dispatch.h>
  131. #include <ATen/ops/broadcast_to_compositeimplicitautograd_dispatch.h>
  132. #include <ATen/ops/can_cast_compositeimplicitautograd_dispatch.h>
  133. #include <ATen/ops/cartesian_prod_compositeimplicitautograd_dispatch.h>
  134. #include <ATen/ops/cat_compositeimplicitautograd_dispatch.h>
  135. #include <ATen/ops/cdist_compositeimplicitautograd_dispatch.h>
  136. #include <ATen/ops/chain_matmul_compositeimplicitautograd_dispatch.h>
  137. #include <ATen/ops/chalf_compositeimplicitautograd_dispatch.h>
  138. #include <ATen/ops/choose_qparams_optimized_compositeimplicitautograd_dispatch.h>
  139. #include <ATen/ops/chunk_compositeimplicitautograd_dispatch.h>
  140. #include <ATen/ops/clip_compositeimplicitautograd_dispatch.h>
  141. #include <ATen/ops/coalesce_compositeimplicitautograd_dispatch.h>
  142. #include <ATen/ops/column_stack_compositeimplicitautograd_dispatch.h>
  143. #include <ATen/ops/combinations_compositeimplicitautograd_dispatch.h>
  144. #include <ATen/ops/concat_compositeimplicitautograd_dispatch.h>
  145. #include <ATen/ops/concatenate_compositeimplicitautograd_dispatch.h>
  146. #include <ATen/ops/conj_compositeimplicitautograd_dispatch.h>
  147. #include <ATen/ops/conj_physical_compositeimplicitautograd_dispatch.h>
  148. #include <ATen/ops/contiguous_compositeimplicitautograd_dispatch.h>
  149. #include <ATen/ops/conv1d_compositeimplicitautograd_dispatch.h>
  150. #include <ATen/ops/conv2d_compositeimplicitautograd_dispatch.h>
  151. #include <ATen/ops/conv3d_compositeimplicitautograd_dispatch.h>
  152. #include <ATen/ops/conv_tbc_backward_compositeimplicitautograd_dispatch.h>
  153. #include <ATen/ops/conv_transpose1d_compositeimplicitautograd_dispatch.h>
  154. #include <ATen/ops/conv_transpose2d_compositeimplicitautograd_dispatch.h>
  155. #include <ATen/ops/conv_transpose3d_compositeimplicitautograd_dispatch.h>
  156. #include <ATen/ops/corrcoef_compositeimplicitautograd_dispatch.h>
  157. #include <ATen/ops/cosine_embedding_loss_compositeimplicitautograd_dispatch.h>
  158. #include <ATen/ops/cosine_similarity_compositeimplicitautograd_dispatch.h>
  159. #include <ATen/ops/cov_compositeimplicitautograd_dispatch.h>
  160. #include <ATen/ops/cross_compositeimplicitautograd_dispatch.h>
  161. #include <ATen/ops/cross_entropy_loss_compositeimplicitautograd_dispatch.h>
  162. #include <ATen/ops/ctc_loss_compositeimplicitautograd_dispatch.h>
  163. #include <ATen/ops/cudnn_is_acceptable_compositeimplicitautograd_dispatch.h>
  164. #include <ATen/ops/cummax_compositeimplicitautograd_dispatch.h>
  165. #include <ATen/ops/cummaxmin_backward_compositeimplicitautograd_dispatch.h>
  166. #include <ATen/ops/cummin_compositeimplicitautograd_dispatch.h>
  167. #include <ATen/ops/cumprod_compositeimplicitautograd_dispatch.h>
  168. #include <ATen/ops/cumprod_backward_compositeimplicitautograd_dispatch.h>
  169. #include <ATen/ops/cumsum_compositeimplicitautograd_dispatch.h>
  170. #include <ATen/ops/cumulative_trapezoid_compositeimplicitautograd_dispatch.h>
  171. #include <ATen/ops/data_compositeimplicitautograd_dispatch.h>
  172. #include <ATen/ops/det_compositeimplicitautograd_dispatch.h>
  173. #include <ATen/ops/diag_compositeimplicitautograd_dispatch.h>
  174. #include <ATen/ops/diagflat_compositeimplicitautograd_dispatch.h>
  175. #include <ATen/ops/diagonal_compositeimplicitautograd_dispatch.h>
  176. #include <ATen/ops/diff_compositeimplicitautograd_dispatch.h>
  177. #include <ATen/ops/divide_compositeimplicitautograd_dispatch.h>
  178. #include <ATen/ops/dropout_compositeimplicitautograd_dispatch.h>
  179. #include <ATen/ops/dsplit_compositeimplicitautograd_dispatch.h>
  180. #include <ATen/ops/dstack_compositeimplicitautograd_dispatch.h>
  181. #include <ATen/ops/einsum_compositeimplicitautograd_dispatch.h>
  182. #include <ATen/ops/embedding_backward_compositeimplicitautograd_dispatch.h>
  183. #include <ATen/ops/embedding_bag_compositeimplicitautograd_dispatch.h>
  184. #include <ATen/ops/embedding_sparse_backward_compositeimplicitautograd_dispatch.h>
  185. #include <ATen/ops/empty_compositeimplicitautograd_dispatch.h>
  186. #include <ATen/ops/expand_as_compositeimplicitautograd_dispatch.h>
  187. #include <ATen/ops/fake_quantize_per_channel_affine_compositeimplicitautograd_dispatch.h>
  188. #include <ATen/ops/fake_quantize_per_channel_affine_cachemask_backward_compositeimplicitautograd_dispatch.h>
  189. #include <ATen/ops/fake_quantize_per_tensor_affine_compositeimplicitautograd_dispatch.h>
  190. #include <ATen/ops/fake_quantize_per_tensor_affine_cachemask_backward_compositeimplicitautograd_dispatch.h>
  191. #include <ATen/ops/fbgemm_linear_fp16_weight_compositeimplicitautograd_dispatch.h>
  192. #include <ATen/ops/fbgemm_linear_fp16_weight_fp32_activation_compositeimplicitautograd_dispatch.h>
  193. #include <ATen/ops/fbgemm_linear_int8_weight_compositeimplicitautograd_dispatch.h>
  194. #include <ATen/ops/fbgemm_linear_int8_weight_fp32_activation_compositeimplicitautograd_dispatch.h>
  195. #include <ATen/ops/fbgemm_linear_quantize_weight_compositeimplicitautograd_dispatch.h>
  196. #include <ATen/ops/fbgemm_pack_gemm_matrix_fp16_compositeimplicitautograd_dispatch.h>
  197. #include <ATen/ops/fbgemm_pack_quantized_matrix_compositeimplicitautograd_dispatch.h>
  198. #include <ATen/ops/feature_alpha_dropout_compositeimplicitautograd_dispatch.h>
  199. #include <ATen/ops/feature_dropout_compositeimplicitautograd_dispatch.h>
  200. #include <ATen/ops/fft_fft_compositeimplicitautograd_dispatch.h>
  201. #include <ATen/ops/fft_fft2_compositeimplicitautograd_dispatch.h>
  202. #include <ATen/ops/fft_fftn_compositeimplicitautograd_dispatch.h>
  203. #include <ATen/ops/fft_fftshift_compositeimplicitautograd_dispatch.h>
  204. #include <ATen/ops/fft_hfft_compositeimplicitautograd_dispatch.h>
  205. #include <ATen/ops/fft_hfft2_compositeimplicitautograd_dispatch.h>
  206. #include <ATen/ops/fft_hfftn_compositeimplicitautograd_dispatch.h>
  207. #include <ATen/ops/fft_ifft_compositeimplicitautograd_dispatch.h>
  208. #include <ATen/ops/fft_ifft2_compositeimplicitautograd_dispatch.h>
  209. #include <ATen/ops/fft_ifftn_compositeimplicitautograd_dispatch.h>
  210. #include <ATen/ops/fft_ifftshift_compositeimplicitautograd_dispatch.h>
  211. #include <ATen/ops/fft_ihfft_compositeimplicitautograd_dispatch.h>
  212. #include <ATen/ops/fft_ihfft2_compositeimplicitautograd_dispatch.h>
  213. #include <ATen/ops/fft_ihfftn_compositeimplicitautograd_dispatch.h>
  214. #include <ATen/ops/fft_irfft_compositeimplicitautograd_dispatch.h>
  215. #include <ATen/ops/fft_irfft2_compositeimplicitautograd_dispatch.h>
  216. #include <ATen/ops/fft_irfftn_compositeimplicitautograd_dispatch.h>
  217. #include <ATen/ops/fft_rfft_compositeimplicitautograd_dispatch.h>
  218. #include <ATen/ops/fft_rfft2_compositeimplicitautograd_dispatch.h>
  219. #include <ATen/ops/fft_rfftn_compositeimplicitautograd_dispatch.h>
  220. #include <ATen/ops/fill_diagonal_compositeimplicitautograd_dispatch.h>
  221. #include <ATen/ops/fix_compositeimplicitautograd_dispatch.h>
  222. #include <ATen/ops/flatten_compositeimplicitautograd_dispatch.h>
  223. #include <ATen/ops/flatten_dense_tensors_compositeimplicitautograd_dispatch.h>
  224. #include <ATen/ops/fliplr_compositeimplicitautograd_dispatch.h>
  225. #include <ATen/ops/flipud_compositeimplicitautograd_dispatch.h>
  226. #include <ATen/ops/float_power_compositeimplicitautograd_dispatch.h>
  227. #include <ATen/ops/floor_divide_compositeimplicitautograd_dispatch.h>
  228. #include <ATen/ops/frobenius_norm_compositeimplicitautograd_dispatch.h>
  229. #include <ATen/ops/fused_moving_avg_obs_fake_quant_compositeimplicitautograd_dispatch.h>
  230. #include <ATen/ops/gather_compositeimplicitautograd_dispatch.h>
  231. #include <ATen/ops/gather_backward_compositeimplicitautograd_dispatch.h>
  232. #include <ATen/ops/ger_compositeimplicitautograd_dispatch.h>
  233. #include <ATen/ops/gradient_compositeimplicitautograd_dispatch.h>
  234. #include <ATen/ops/greater_compositeimplicitautograd_dispatch.h>
  235. #include <ATen/ops/greater_equal_compositeimplicitautograd_dispatch.h>
  236. #include <ATen/ops/grid_sampler_compositeimplicitautograd_dispatch.h>
  237. #include <ATen/ops/group_norm_compositeimplicitautograd_dispatch.h>
  238. #include <ATen/ops/gru_compositeimplicitautograd_dispatch.h>
  239. #include <ATen/ops/gru_cell_compositeimplicitautograd_dispatch.h>
  240. #include <ATen/ops/hinge_embedding_loss_compositeimplicitautograd_dispatch.h>
  241. #include <ATen/ops/histogramdd_compositeimplicitautograd_dispatch.h>
  242. #include <ATen/ops/hsplit_compositeimplicitautograd_dispatch.h>
  243. #include <ATen/ops/hstack_compositeimplicitautograd_dispatch.h>
  244. #include <ATen/ops/imag_compositeimplicitautograd_dispatch.h>
  245. #include <ATen/ops/index_add_compositeimplicitautograd_dispatch.h>
  246. #include <ATen/ops/index_copy_compositeimplicitautograd_dispatch.h>
  247. #include <ATen/ops/index_fill_compositeimplicitautograd_dispatch.h>
  248. #include <ATen/ops/index_select_compositeimplicitautograd_dispatch.h>
  249. #include <ATen/ops/index_select_backward_compositeimplicitautograd_dispatch.h>
  250. #include <ATen/ops/infinitely_differentiable_gelu_backward_compositeimplicitautograd_dispatch.h>
  251. #include <ATen/ops/inner_compositeimplicitautograd_dispatch.h>
  252. #include <ATen/ops/instance_norm_compositeimplicitautograd_dispatch.h>
  253. #include <ATen/ops/inverse_compositeimplicitautograd_dispatch.h>
  254. #include <ATen/ops/is_complex_compositeimplicitautograd_dispatch.h>
  255. #include <ATen/ops/is_conj_compositeimplicitautograd_dispatch.h>
  256. #include <ATen/ops/is_distributed_compositeimplicitautograd_dispatch.h>
  257. #include <ATen/ops/is_floating_point_compositeimplicitautograd_dispatch.h>
  258. #include <ATen/ops/is_inference_compositeimplicitautograd_dispatch.h>
  259. #include <ATen/ops/is_leaf_compositeimplicitautograd_dispatch.h>
  260. #include <ATen/ops/is_neg_compositeimplicitautograd_dispatch.h>
  261. #include <ATen/ops/is_nonzero_compositeimplicitautograd_dispatch.h>
  262. #include <ATen/ops/is_signed_compositeimplicitautograd_dispatch.h>
  263. #include <ATen/ops/is_vulkan_available_compositeimplicitautograd_dispatch.h>
  264. #include <ATen/ops/isclose_compositeimplicitautograd_dispatch.h>
  265. #include <ATen/ops/isfinite_compositeimplicitautograd_dispatch.h>
  266. #include <ATen/ops/isreal_compositeimplicitautograd_dispatch.h>
  267. #include <ATen/ops/istft_compositeimplicitautograd_dispatch.h>
  268. #include <ATen/ops/item_compositeimplicitautograd_dispatch.h>
  269. #include <ATen/ops/kl_div_compositeimplicitautograd_dispatch.h>
  270. #include <ATen/ops/kron_compositeimplicitautograd_dispatch.h>
  271. #include <ATen/ops/kthvalue_compositeimplicitautograd_dispatch.h>
  272. #include <ATen/ops/l1_loss_compositeimplicitautograd_dispatch.h>
  273. #include <ATen/ops/layer_norm_compositeimplicitautograd_dispatch.h>
  274. #include <ATen/ops/ldexp_compositeimplicitautograd_dispatch.h>
  275. #include <ATen/ops/less_compositeimplicitautograd_dispatch.h>
  276. #include <ATen/ops/less_equal_compositeimplicitautograd_dispatch.h>
  277. #include <ATen/ops/linalg_cholesky_compositeimplicitautograd_dispatch.h>
  278. #include <ATen/ops/linalg_cond_compositeimplicitautograd_dispatch.h>
  279. #include <ATen/ops/linalg_det_compositeimplicitautograd_dispatch.h>
  280. #include <ATen/ops/linalg_diagonal_compositeimplicitautograd_dispatch.h>
  281. #include <ATen/ops/linalg_eigh_compositeimplicitautograd_dispatch.h>
  282. #include <ATen/ops/linalg_eigvals_compositeimplicitautograd_dispatch.h>
  283. #include <ATen/ops/linalg_eigvalsh_compositeimplicitautograd_dispatch.h>
  284. #include <ATen/ops/linalg_inv_compositeimplicitautograd_dispatch.h>
  285. #include <ATen/ops/linalg_ldl_factor_compositeimplicitautograd_dispatch.h>
  286. #include <ATen/ops/linalg_lu_factor_compositeimplicitautograd_dispatch.h>
  287. #include <ATen/ops/linalg_matmul_compositeimplicitautograd_dispatch.h>
  288. #include <ATen/ops/linalg_matrix_norm_compositeimplicitautograd_dispatch.h>
  289. #include <ATen/ops/linalg_matrix_power_compositeimplicitautograd_dispatch.h>
  290. #include <ATen/ops/linalg_matrix_rank_compositeimplicitautograd_dispatch.h>
  291. #include <ATen/ops/linalg_multi_dot_compositeimplicitautograd_dispatch.h>
  292. #include <ATen/ops/linalg_norm_compositeimplicitautograd_dispatch.h>
  293. #include <ATen/ops/linalg_pinv_compositeimplicitautograd_dispatch.h>
  294. #include <ATen/ops/linalg_slogdet_compositeimplicitautograd_dispatch.h>
  295. #include <ATen/ops/linalg_solve_compositeimplicitautograd_dispatch.h>
  296. #include <ATen/ops/linalg_solve_ex_compositeimplicitautograd_dispatch.h>
  297. #include <ATen/ops/linalg_svd_compositeimplicitautograd_dispatch.h>
  298. #include <ATen/ops/linalg_svdvals_compositeimplicitautograd_dispatch.h>
  299. #include <ATen/ops/linalg_tensorinv_compositeimplicitautograd_dispatch.h>
  300. #include <ATen/ops/linalg_tensorsolve_compositeimplicitautograd_dispatch.h>
  301. #include <ATen/ops/linalg_vander_compositeimplicitautograd_dispatch.h>
  302. #include <ATen/ops/linalg_vecdot_compositeimplicitautograd_dispatch.h>
  303. #include <ATen/ops/linear_compositeimplicitautograd_dispatch.h>
  304. #include <ATen/ops/log_sigmoid_compositeimplicitautograd_dispatch.h>
  305. #include <ATen/ops/log_softmax_compositeimplicitautograd_dispatch.h>
  306. #include <ATen/ops/logcumsumexp_compositeimplicitautograd_dispatch.h>
  307. #include <ATen/ops/logdet_compositeimplicitautograd_dispatch.h>
  308. #include <ATen/ops/logsumexp_compositeimplicitautograd_dispatch.h>
  309. #include <ATen/ops/lstm_compositeimplicitautograd_dispatch.h>
  310. #include <ATen/ops/lstm_cell_compositeimplicitautograd_dispatch.h>
  311. #include <ATen/ops/lu_solve_compositeimplicitautograd_dispatch.h>
  312. #include <ATen/ops/mH_compositeimplicitautograd_dispatch.h>
  313. #include <ATen/ops/mT_compositeimplicitautograd_dispatch.h>
  314. #include <ATen/ops/margin_ranking_loss_compositeimplicitautograd_dispatch.h>
  315. #include <ATen/ops/masked_select_backward_compositeimplicitautograd_dispatch.h>
  316. #include <ATen/ops/matmul_compositeimplicitautograd_dispatch.h>
  317. #include <ATen/ops/matrix_H_compositeimplicitautograd_dispatch.h>
  318. #include <ATen/ops/matrix_exp_compositeimplicitautograd_dispatch.h>
  319. #include <ATen/ops/matrix_exp_backward_compositeimplicitautograd_dispatch.h>
  320. #include <ATen/ops/matrix_power_compositeimplicitautograd_dispatch.h>
  321. #include <ATen/ops/max_compositeimplicitautograd_dispatch.h>
  322. #include <ATen/ops/max_pool1d_compositeimplicitautograd_dispatch.h>
  323. #include <ATen/ops/max_pool1d_with_indices_compositeimplicitautograd_dispatch.h>
  324. #include <ATen/ops/max_pool2d_compositeimplicitautograd_dispatch.h>
  325. #include <ATen/ops/max_pool3d_compositeimplicitautograd_dispatch.h>
  326. #include <ATen/ops/mean_compositeimplicitautograd_dispatch.h>
  327. #include <ATen/ops/median_compositeimplicitautograd_dispatch.h>
  328. #include <ATen/ops/meshgrid_compositeimplicitautograd_dispatch.h>
  329. #include <ATen/ops/min_compositeimplicitautograd_dispatch.h>
  330. #include <ATen/ops/mish_backward_compositeimplicitautograd_dispatch.h>
  331. #include <ATen/ops/mode_compositeimplicitautograd_dispatch.h>
  332. #include <ATen/ops/moveaxis_compositeimplicitautograd_dispatch.h>
  333. #include <ATen/ops/movedim_compositeimplicitautograd_dispatch.h>
  334. #include <ATen/ops/msort_compositeimplicitautograd_dispatch.h>
  335. #include <ATen/ops/multilabel_margin_loss_compositeimplicitautograd_dispatch.h>
  336. #include <ATen/ops/multiply_compositeimplicitautograd_dispatch.h>
  337. #include <ATen/ops/nanmean_compositeimplicitautograd_dispatch.h>
  338. #include <ATen/ops/nanmedian_compositeimplicitautograd_dispatch.h>
  339. #include <ATen/ops/nanquantile_compositeimplicitautograd_dispatch.h>
  340. #include <ATen/ops/narrow_compositeimplicitautograd_dispatch.h>
  341. #include <ATen/ops/native_channel_shuffle_compositeimplicitautograd_dispatch.h>
  342. #include <ATen/ops/negative_compositeimplicitautograd_dispatch.h>
  343. #include <ATen/ops/nested_to_padded_tensor_compositeimplicitautograd_dispatch.h>
  344. #include <ATen/ops/nll_loss_compositeimplicitautograd_dispatch.h>
  345. #include <ATen/ops/nll_loss2d_compositeimplicitautograd_dispatch.h>
  346. #include <ATen/ops/nll_loss_nd_compositeimplicitautograd_dispatch.h>
  347. #include <ATen/ops/nonzero_numpy_compositeimplicitautograd_dispatch.h>
  348. #include <ATen/ops/norm_compositeimplicitautograd_dispatch.h>
  349. #include <ATen/ops/norm_except_dim_compositeimplicitautograd_dispatch.h>
  350. #include <ATen/ops/not_equal_compositeimplicitautograd_dispatch.h>
  351. #include <ATen/ops/nuclear_norm_compositeimplicitautograd_dispatch.h>
  352. #include <ATen/ops/numpy_T_compositeimplicitautograd_dispatch.h>
  353. #include <ATen/ops/one_hot_compositeimplicitautograd_dispatch.h>
  354. #include <ATen/ops/or_compositeimplicitautograd_dispatch.h>
  355. #include <ATen/ops/orgqr_compositeimplicitautograd_dispatch.h>
  356. #include <ATen/ops/outer_compositeimplicitautograd_dispatch.h>
  357. #include <ATen/ops/output_nr_compositeimplicitautograd_dispatch.h>
  358. #include <ATen/ops/pad_compositeimplicitautograd_dispatch.h>
  359. #include <ATen/ops/pad_sequence_compositeimplicitautograd_dispatch.h>
  360. #include <ATen/ops/pairwise_distance_compositeimplicitautograd_dispatch.h>
  361. #include <ATen/ops/pdist_compositeimplicitautograd_dispatch.h>
  362. #include <ATen/ops/pin_memory_compositeimplicitautograd_dispatch.h>
  363. #include <ATen/ops/pinverse_compositeimplicitautograd_dispatch.h>
  364. #include <ATen/ops/poisson_nll_loss_compositeimplicitautograd_dispatch.h>
  365. #include <ATen/ops/positive_compositeimplicitautograd_dispatch.h>
  366. #include <ATen/ops/prelu_compositeimplicitautograd_dispatch.h>
  367. #include <ATen/ops/prod_compositeimplicitautograd_dispatch.h>
  368. #include <ATen/ops/promote_types_compositeimplicitautograd_dispatch.h>
  369. #include <ATen/ops/qr_compositeimplicitautograd_dispatch.h>
  370. #include <ATen/ops/quantile_compositeimplicitautograd_dispatch.h>
  371. #include <ATen/ops/quantized_gru_cell_compositeimplicitautograd_dispatch.h>
  372. #include <ATen/ops/quantized_lstm_cell_compositeimplicitautograd_dispatch.h>
  373. #include <ATen/ops/quantized_rnn_relu_cell_compositeimplicitautograd_dispatch.h>
  374. #include <ATen/ops/quantized_rnn_tanh_cell_compositeimplicitautograd_dispatch.h>
  375. #include <ATen/ops/rand_compositeimplicitautograd_dispatch.h>
  376. #include <ATen/ops/randn_compositeimplicitautograd_dispatch.h>
  377. #include <ATen/ops/ravel_compositeimplicitautograd_dispatch.h>
  378. #include <ATen/ops/real_compositeimplicitautograd_dispatch.h>
  379. #include <ATen/ops/refine_names_compositeimplicitautograd_dispatch.h>
  380. #include <ATen/ops/relu6_compositeimplicitautograd_dispatch.h>
  381. #include <ATen/ops/rename_compositeimplicitautograd_dispatch.h>
  382. #include <ATen/ops/repeat_interleave_compositeimplicitautograd_dispatch.h>
  383. #include <ATen/ops/requires_grad_compositeimplicitautograd_dispatch.h>
  384. #include <ATen/ops/reshape_compositeimplicitautograd_dispatch.h>
  385. #include <ATen/ops/reshape_as_compositeimplicitautograd_dispatch.h>
  386. #include <ATen/ops/resolve_conj_compositeimplicitautograd_dispatch.h>
  387. #include <ATen/ops/resolve_neg_compositeimplicitautograd_dispatch.h>
  388. #include <ATen/ops/result_type_compositeimplicitautograd_dispatch.h>
  389. #include <ATen/ops/retain_grad_compositeimplicitautograd_dispatch.h>
  390. #include <ATen/ops/retains_grad_compositeimplicitautograd_dispatch.h>
  391. #include <ATen/ops/rnn_relu_compositeimplicitautograd_dispatch.h>
  392. #include <ATen/ops/rnn_relu_cell_compositeimplicitautograd_dispatch.h>
  393. #include <ATen/ops/rnn_tanh_compositeimplicitautograd_dispatch.h>
  394. #include <ATen/ops/rnn_tanh_cell_compositeimplicitautograd_dispatch.h>
  395. #include <ATen/ops/row_stack_compositeimplicitautograd_dispatch.h>
  396. #include <ATen/ops/rrelu_compositeimplicitautograd_dispatch.h>
  397. #include <ATen/ops/scaled_dot_product_attention_compositeimplicitautograd_dispatch.h>
  398. #include <ATen/ops/scatter_compositeimplicitautograd_dispatch.h>
  399. #include <ATen/ops/scatter_add_compositeimplicitautograd_dispatch.h>
  400. #include <ATen/ops/select_compositeimplicitautograd_dispatch.h>
  401. #include <ATen/ops/selu_compositeimplicitautograd_dispatch.h>
  402. #include <ATen/ops/set_compositeimplicitautograd_dispatch.h>
  403. #include <ATen/ops/set_data_compositeimplicitautograd_dispatch.h>
  404. #include <ATen/ops/silu_backward_compositeimplicitautograd_dispatch.h>
  405. #include <ATen/ops/size_compositeimplicitautograd_dispatch.h>
  406. #include <ATen/ops/slogdet_compositeimplicitautograd_dispatch.h>
  407. #include <ATen/ops/slow_conv3d_compositeimplicitautograd_dispatch.h>
  408. #include <ATen/ops/smm_compositeimplicitautograd_dispatch.h>
  409. #include <ATen/ops/softmax_compositeimplicitautograd_dispatch.h>
  410. #include <ATen/ops/sort_compositeimplicitautograd_dispatch.h>
  411. #include <ATen/ops/sparse_bsc_tensor_compositeimplicitautograd_dispatch.h>
  412. #include <ATen/ops/sparse_bsr_tensor_compositeimplicitautograd_dispatch.h>
  413. #include <ATen/ops/sparse_compressed_tensor_compositeimplicitautograd_dispatch.h>
  414. #include <ATen/ops/sparse_coo_tensor_compositeimplicitautograd_dispatch.h>
  415. #include <ATen/ops/sparse_csc_tensor_compositeimplicitautograd_dispatch.h>
  416. #include <ATen/ops/sparse_csr_tensor_compositeimplicitautograd_dispatch.h>
  417. #include <ATen/ops/special_chebyshev_polynomial_t_compositeimplicitautograd_dispatch.h>
  418. #include <ATen/ops/special_chebyshev_polynomial_u_compositeimplicitautograd_dispatch.h>
  419. #include <ATen/ops/special_chebyshev_polynomial_v_compositeimplicitautograd_dispatch.h>
  420. #include <ATen/ops/special_chebyshev_polynomial_w_compositeimplicitautograd_dispatch.h>
  421. #include <ATen/ops/special_digamma_compositeimplicitautograd_dispatch.h>
  422. #include <ATen/ops/special_erf_compositeimplicitautograd_dispatch.h>
  423. #include <ATen/ops/special_erfc_compositeimplicitautograd_dispatch.h>
  424. #include <ATen/ops/special_erfinv_compositeimplicitautograd_dispatch.h>
  425. #include <ATen/ops/special_exp2_compositeimplicitautograd_dispatch.h>
  426. #include <ATen/ops/special_expit_compositeimplicitautograd_dispatch.h>
  427. #include <ATen/ops/special_expm1_compositeimplicitautograd_dispatch.h>
  428. #include <ATen/ops/special_gammainc_compositeimplicitautograd_dispatch.h>
  429. #include <ATen/ops/special_gammaincc_compositeimplicitautograd_dispatch.h>
  430. #include <ATen/ops/special_gammaln_compositeimplicitautograd_dispatch.h>
  431. #include <ATen/ops/special_hermite_polynomial_h_compositeimplicitautograd_dispatch.h>
  432. #include <ATen/ops/special_hermite_polynomial_he_compositeimplicitautograd_dispatch.h>
  433. #include <ATen/ops/special_i0_compositeimplicitautograd_dispatch.h>
  434. #include <ATen/ops/special_laguerre_polynomial_l_compositeimplicitautograd_dispatch.h>
  435. #include <ATen/ops/special_legendre_polynomial_p_compositeimplicitautograd_dispatch.h>
  436. #include <ATen/ops/special_log1p_compositeimplicitautograd_dispatch.h>
  437. #include <ATen/ops/special_log_softmax_compositeimplicitautograd_dispatch.h>
  438. #include <ATen/ops/special_logit_compositeimplicitautograd_dispatch.h>
  439. #include <ATen/ops/special_logsumexp_compositeimplicitautograd_dispatch.h>
  440. #include <ATen/ops/special_multigammaln_compositeimplicitautograd_dispatch.h>
  441. #include <ATen/ops/special_ndtr_compositeimplicitautograd_dispatch.h>
  442. #include <ATen/ops/special_polygamma_compositeimplicitautograd_dispatch.h>
  443. #include <ATen/ops/special_psi_compositeimplicitautograd_dispatch.h>
  444. #include <ATen/ops/special_round_compositeimplicitautograd_dispatch.h>
  445. #include <ATen/ops/special_shifted_chebyshev_polynomial_t_compositeimplicitautograd_dispatch.h>
  446. #include <ATen/ops/special_shifted_chebyshev_polynomial_u_compositeimplicitautograd_dispatch.h>
  447. #include <ATen/ops/special_shifted_chebyshev_polynomial_v_compositeimplicitautograd_dispatch.h>
  448. #include <ATen/ops/special_shifted_chebyshev_polynomial_w_compositeimplicitautograd_dispatch.h>
  449. #include <ATen/ops/special_sinc_compositeimplicitautograd_dispatch.h>
  450. #include <ATen/ops/special_softmax_compositeimplicitautograd_dispatch.h>
  451. #include <ATen/ops/special_xlogy_compositeimplicitautograd_dispatch.h>
  452. #include <ATen/ops/split_compositeimplicitautograd_dispatch.h>
  453. #include <ATen/ops/square_compositeimplicitautograd_dispatch.h>
  454. #include <ATen/ops/squeeze_compositeimplicitautograd_dispatch.h>
  455. #include <ATen/ops/sspaddmm_compositeimplicitautograd_dispatch.h>
  456. #include <ATen/ops/std_compositeimplicitautograd_dispatch.h>
  457. #include <ATen/ops/std_mean_compositeimplicitautograd_dispatch.h>
  458. #include <ATen/ops/stft_compositeimplicitautograd_dispatch.h>
  459. #include <ATen/ops/stride_compositeimplicitautograd_dispatch.h>
  460. #include <ATen/ops/subtract_compositeimplicitautograd_dispatch.h>
  461. #include <ATen/ops/sum_compositeimplicitautograd_dispatch.h>
  462. #include <ATen/ops/sum_to_size_compositeimplicitautograd_dispatch.h>
  463. #include <ATen/ops/svd_compositeimplicitautograd_dispatch.h>
  464. #include <ATen/ops/swapaxes_compositeimplicitautograd_dispatch.h>
  465. #include <ATen/ops/swapdims_compositeimplicitautograd_dispatch.h>
  466. #include <ATen/ops/take_along_dim_compositeimplicitautograd_dispatch.h>
  467. #include <ATen/ops/tensor_split_compositeimplicitautograd_dispatch.h>
  468. #include <ATen/ops/tensordot_compositeimplicitautograd_dispatch.h>
  469. #include <ATen/ops/thnn_conv2d_compositeimplicitautograd_dispatch.h>
  470. #include <ATen/ops/tile_compositeimplicitautograd_dispatch.h>
  471. #include <ATen/ops/to_compositeimplicitautograd_dispatch.h>
  472. #include <ATen/ops/to_dense_compositeimplicitautograd_dispatch.h>
  473. #include <ATen/ops/to_dense_backward_compositeimplicitautograd_dispatch.h>
  474. #include <ATen/ops/to_mkldnn_backward_compositeimplicitautograd_dispatch.h>
  475. #include <ATen/ops/trace_backward_compositeimplicitautograd_dispatch.h>
  476. #include <ATen/ops/transpose_compositeimplicitautograd_dispatch.h>
  477. #include <ATen/ops/trapezoid_compositeimplicitautograd_dispatch.h>
  478. #include <ATen/ops/trapz_compositeimplicitautograd_dispatch.h>
  479. #include <ATen/ops/triplet_margin_loss_compositeimplicitautograd_dispatch.h>
  480. #include <ATen/ops/true_divide_compositeimplicitautograd_dispatch.h>
  481. #include <ATen/ops/type_as_compositeimplicitautograd_dispatch.h>
  482. #include <ATen/ops/unbind_compositeimplicitautograd_dispatch.h>
  483. #include <ATen/ops/unflatten_compositeimplicitautograd_dispatch.h>
  484. #include <ATen/ops/unflatten_dense_tensors_compositeimplicitautograd_dispatch.h>
  485. #include <ATen/ops/unsafe_chunk_compositeimplicitautograd_dispatch.h>
  486. #include <ATen/ops/upsample_bicubic2d_compositeimplicitautograd_dispatch.h>
  487. #include <ATen/ops/upsample_bilinear2d_compositeimplicitautograd_dispatch.h>
  488. #include <ATen/ops/upsample_linear1d_compositeimplicitautograd_dispatch.h>
  489. #include <ATen/ops/upsample_nearest1d_compositeimplicitautograd_dispatch.h>
  490. #include <ATen/ops/upsample_nearest2d_compositeimplicitautograd_dispatch.h>
  491. #include <ATen/ops/upsample_nearest3d_compositeimplicitautograd_dispatch.h>
  492. #include <ATen/ops/upsample_trilinear3d_compositeimplicitautograd_dispatch.h>
  493. #include <ATen/ops/value_selecting_reduction_backward_compositeimplicitautograd_dispatch.h>
  494. #include <ATen/ops/vander_compositeimplicitautograd_dispatch.h>
  495. #include <ATen/ops/var_compositeimplicitautograd_dispatch.h>
  496. #include <ATen/ops/var_mean_compositeimplicitautograd_dispatch.h>
  497. #include <ATen/ops/view_as_compositeimplicitautograd_dispatch.h>
  498. #include <ATen/ops/vsplit_compositeimplicitautograd_dispatch.h>
  499. #include <ATen/ops/vstack_compositeimplicitautograd_dispatch.h>
  500. #include <ATen/ops/where_compositeimplicitautograd_dispatch.h>
  501. #include <ATen/ops/xor_compositeimplicitautograd_dispatch.h>