1 #pragma once 2 3 #include <torch/csrc/autograd/autograd.h> 4 #include <torch/csrc/autograd/autograd_not_implemented_fallback.h> 5 #include <torch/csrc/autograd/custom_function.h> 6