mirror of
https://github.com/davrot/pytorch-sbs.git
synced 2025-07-02 00:00:02 +02:00
14 lines
No EOL
335 B
C++
14 lines
No EOL
335 B
C++
#include <pybind11/pybind11.h>
|
|
|
|
#include "HDynamicCNNManyIP.h"
|
|
|
|
namespace py = pybind11;
|
|
|
|
PYBIND11_MODULE(PyHDynamicCNNManyIP, m)
|
|
{
|
|
m.doc() = "HDynamicCNNManyIP Module";
|
|
py::class_<HDynamicCNNManyIP>(m, "HDynamicCNNManyIP")
|
|
.def(py::init<>())
|
|
.def("update",
|
|
&HDynamicCNNManyIP::update_entrypoint);
|
|
} |