pytorch-sbs/network/count_spikes_cpu_cpp/PyCountSpikesCPU.cpp
2023-02-04 14:28:11 +01:00

15 lines
292 B
C++

#include <pybind11/pybind11.h>
#include "CountSpikesCPU.h"
namespace py = pybind11;
PYBIND11_MODULE(PyCountSpikesCPU, m)
{
m.doc() = "CountSpikesCPU Module";
py::class_<CountSpikesCPU>(m, "CountSpikesCPU")
.def(py::init<>())
.def("process",
&CountSpikesCPU::process);
}