xref: /aosp_15_r20/external/pytorch/tools/jit/templates/external_functions_codegen_template.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 // ${generated_comment}
2 #include <torch/csrc/jit/tensorexpr/external_functions.h>
3 
4 #include <ATen/Functions.h>
5 #include <ATen/NativeFunctions.h>
6 #include <c10/util/irange.h>
7 #include <torch/csrc/jit/tensorexpr/external_functions_registry.h>
8 
9 namespace torch {
10 namespace jit {
11 namespace tensorexpr {
12 
13 #ifdef C10_MOBILE
14 extern "C" {
15 #endif
16 
17 ${external_functions}
18 
19 #ifndef C10_MOBILE
20 ${external_registrations}
21 #endif
22 
23 #ifdef C10_MOBILE
24 } // extern "C"
25 #endif
26 
27 } // namespace tensorexpr
28 } // namespace jit
29 } // namespace torch
30