1 #pragma once 2 3 #include <torch/csrc/api/include/torch/types.h> 4 #include <torch/csrc/autograd/InferenceMode.h> 5 #include <torch/csrc/autograd/custom_function.h> 6 #include <torch/csrc/autograd/generated/variable_factories.h> 7 #include <torch/csrc/autograd/grad_mode.h> 8 #include <torch/csrc/jit/runtime/custom_operator.h> 9 #include <torch/csrc/jit/serialization/import.h> 10 #include <torch/csrc/jit/serialization/pickle.h> 11 #include <torch/custom_class.h> 12 13 #include <ATen/ATen.h> 14