xref: /aosp_15_r20/external/pytorch/torch/script.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #pragma once
2*da0073e9SAndroid Build Coastguard Worker 
3*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/api/include/torch/types.h>
4*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/InferenceMode.h>
5*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/custom_function.h>
6*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/generated/variable_factories.h>
7*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/grad_mode.h>
8*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/jit/runtime/custom_operator.h>
9*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/jit/serialization/import.h>
10*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/jit/serialization/pickle.h>
11*da0073e9SAndroid Build Coastguard Worker #include <torch/custom_class.h>
12*da0073e9SAndroid Build Coastguard Worker 
13*da0073e9SAndroid Build Coastguard Worker #include <ATen/ATen.h>
14