xref: /aosp_15_r20/external/pytorch/test/edge/templates/RegisterDispatchKeyCustomOps.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker // clang-format off
2*da0073e9SAndroid Build Coastguard Worker // Generated code for registering custom operators into the dispatcher.
3*da0073e9SAndroid Build Coastguard Worker 
4*da0073e9SAndroid Build Coastguard Worker #include <torch/library.h>
5*da0073e9SAndroid Build Coastguard Worker #include <ATen/Tensor.h>
6*da0073e9SAndroid Build Coastguard Worker 
7*da0073e9SAndroid Build Coastguard Worker $ops_headers
8*da0073e9SAndroid Build Coastguard Worker 
9*da0073e9SAndroid Build Coastguard Worker namespace torch {
10*da0073e9SAndroid Build Coastguard Worker namespace executor {
11*da0073e9SAndroid Build Coastguard Worker namespace function {
12*da0073e9SAndroid Build Coastguard Worker 
13*da0073e9SAndroid Build Coastguard Worker 
14*da0073e9SAndroid Build Coastguard Worker ${dispatch_anonymous_definitions}
15*da0073e9SAndroid Build Coastguard Worker 
16*da0073e9SAndroid Build Coastguard Worker // All out variants ops
17*da0073e9SAndroid Build Coastguard Worker ${static_init_dispatch_registrations}
18*da0073e9SAndroid Build Coastguard Worker 
19*da0073e9SAndroid Build Coastguard Worker namespace ${dispatch_namespace}
20*da0073e9SAndroid Build Coastguard Worker {
21*da0073e9SAndroid Build Coastguard Worker   ${dispatch_namespaced_definitions}
22*da0073e9SAndroid Build Coastguard Worker 
23*da0073e9SAndroid Build Coastguard Worker } // namespace ${dispatch_namespace}
24*da0073e9SAndroid Build Coastguard Worker 
25*da0073e9SAndroid Build Coastguard Worker } // namespace function
26*da0073e9SAndroid Build Coastguard Worker } // namespace executor
27*da0073e9SAndroid Build Coastguard Worker } // namespace torch
28