xref: /aosp_15_r20/external/executorch/codegen/templates/RegisterDispatchKeyCustomOps.cpp (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1*523fa7a6SAndroid Build Coastguard Worker /*
2*523fa7a6SAndroid Build Coastguard Worker  * Copyright (c) Meta Platforms, Inc. and affiliates.
3*523fa7a6SAndroid Build Coastguard Worker  * All rights reserved.
4*523fa7a6SAndroid Build Coastguard Worker  *
5*523fa7a6SAndroid Build Coastguard Worker  * This source code is licensed under the BSD-style license found in the
6*523fa7a6SAndroid Build Coastguard Worker  * LICENSE file in the root directory of this source tree.
7*523fa7a6SAndroid Build Coastguard Worker  */
8*523fa7a6SAndroid Build Coastguard Worker 
9*523fa7a6SAndroid Build Coastguard Worker // clang-format off
10*523fa7a6SAndroid Build Coastguard Worker #include <torch/library.h>
11*523fa7a6SAndroid Build Coastguard Worker #include <ATen/Tensor.h>
12*523fa7a6SAndroid Build Coastguard Worker 
13*523fa7a6SAndroid Build Coastguard Worker ${ops_headers}
14*523fa7a6SAndroid Build Coastguard Worker 
15*523fa7a6SAndroid Build Coastguard Worker namespace torch {
16*523fa7a6SAndroid Build Coastguard Worker namespace executor {
17*523fa7a6SAndroid Build Coastguard Worker namespace function {
18*523fa7a6SAndroid Build Coastguard Worker 
19*523fa7a6SAndroid Build Coastguard Worker ${dispatch_anonymous_definitions}
20*523fa7a6SAndroid Build Coastguard Worker 
21*523fa7a6SAndroid Build Coastguard Worker // All out variants ops
22*523fa7a6SAndroid Build Coastguard Worker ${static_init_dispatch_registrations}
23*523fa7a6SAndroid Build Coastguard Worker 
24*523fa7a6SAndroid Build Coastguard Worker namespace ${dispatch_namespace} {
25*523fa7a6SAndroid Build Coastguard Worker 
26*523fa7a6SAndroid Build Coastguard Worker ${dispatch_namespaced_definitions}
27*523fa7a6SAndroid Build Coastguard Worker 
28*523fa7a6SAndroid Build Coastguard Worker } // namespace ${dispatch_namespace}
29*523fa7a6SAndroid Build Coastguard Worker 
30*523fa7a6SAndroid Build Coastguard Worker } // namespace function
31*523fa7a6SAndroid Build Coastguard Worker } // namespace executor
32*523fa7a6SAndroid Build Coastguard Worker } // namespace torch
33