xref: /aosp_15_r20/external/pytorch/aten/src/ATen/templates/DispatchKeyFunction.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 // ${generated_comment}
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 // Forward declarations of any types needed in the operator signatures.
12 // We can't directly include these classes because it will cause circular include dependencies.
13 // This file is included by TensorBody.h, which defines the Tensor class.
14 #include <ATen/core/ATen_fwd.h>
15 
16 namespace at {
17 
18 namespace ${dispatch_namespace} {
19 
20 ${dispatch_namespaced_declarations}
21 
22 } // namespace ${dispatch_namespace}
23 } // namespace at
24