1 #pragma once 2 #include <torch/csrc/python_headers.h> 3 4 namespace torch { 5 namespace cpu { 6 7 void initModule(PyObject* module); 8 9 } // namespace cpu 10 } // namespace torch 11