/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python/python_custom_class.h (471B)
#pragma once
#include
#include
#include
namespace torch {
namespace jit {
void initPythonCustomClassBindings(PyObject* module);
struct ScriptClass {
ScriptClass(c10::StrongTypePtr class_type)
: class_type_(std::move(class_type)) {}
py::object __call__(py::args args, py::kwargs kwargs);
c10::StrongTypePtr class_type_;
};
} // namespace jit
} // namespace torch