pytorch-sbs/network/multiplication_approximation_cpu_cpp/PyMultiplicationApproximationCPU.cpp

14 lines
400 B
C++
Raw Permalink Normal View History

2023-02-02 19:15:23 +01:00
#include <pybind11/pybind11.h>
#include "MultiplicationApproximationCPU.h"
namespace py = pybind11;
PYBIND11_MODULE(PyMultiplicationApproximationCPU, m) {
m.doc() = "MultiplicationApproximationCPU Module";
py::class_<MultiplicationApproximationCPU>(m, "MultiplicationApproximationCPU")
.def(py::init<>())
.def("update_entrypoint",
&MultiplicationApproximationCPU::entrypoint);
}