xref: /aosp_15_r20/external/pytorch/test/edge/templates/Functions.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker // clang-format off
2*da0073e9SAndroid Build Coastguard Worker #pragma once
3*da0073e9SAndroid Build Coastguard Worker #include <ATen/Context.h>
4*da0073e9SAndroid Build Coastguard Worker #include <ATen/DeviceGuard.h>
5*da0073e9SAndroid Build Coastguard Worker #include <ATen/TensorUtils.h>
6*da0073e9SAndroid Build Coastguard Worker #include <ATen/TracerMode.h>
7*da0073e9SAndroid Build Coastguard Worker #include <ATen/core/Generator.h>
8*da0073e9SAndroid Build Coastguard Worker #include <ATen/core/Reduction.h>
9*da0073e9SAndroid Build Coastguard Worker #include <ATen/core/Tensor.h>
10*da0073e9SAndroid Build Coastguard Worker #include <c10/core/Scalar.h>
11*da0073e9SAndroid Build Coastguard Worker #include <c10/core/Storage.h>
12*da0073e9SAndroid Build Coastguard Worker #include <c10/core/TensorOptions.h>
13*da0073e9SAndroid Build Coastguard Worker #include <c10/util/Deprecated.h>
14*da0073e9SAndroid Build Coastguard Worker #include <c10/util/Optional.h>
15*da0073e9SAndroid Build Coastguard Worker // ${generated_comment}
16*da0073e9SAndroid Build Coastguard Worker 
17*da0073e9SAndroid Build Coastguard Worker ${static_dispatch_extra_headers}
18*da0073e9SAndroid Build Coastguard Worker 
19*da0073e9SAndroid Build Coastguard Worker namespace torch {
20*da0073e9SAndroid Build Coastguard Worker namespace executor {
21*da0073e9SAndroid Build Coastguard Worker 
22*da0073e9SAndroid Build Coastguard Worker ${Functions_declarations}
23*da0073e9SAndroid Build Coastguard Worker 
24*da0073e9SAndroid Build Coastguard Worker } // namespace executor
25*da0073e9SAndroid Build Coastguard Worker } // namespace torch
26