pytorch-sbs/network/h_dynamic_cnn_cpu_cpp/PyHDynamicCNNCPU.cpp
2023-02-02 19:15:23 +01:00

14 lines
No EOL
310 B
C++

#include <pybind11/pybind11.h>
#include "HDynamicCNNCPU.h"
namespace py = pybind11;
PYBIND11_MODULE(PyHDynamicCNNCPU, m)
{
m.doc() = "HDynamicCNNCPU Module";
py::class_<HDynamicCNNCPU>(m, "HDynamicCNNCPU")
.def(py::init<>())
.def("update",
&HDynamicCNNCPU::entrypoint);
}