diff --git a/include/aidge/aidge.hpp b/include/aidge/aidge.hpp index 931b1b26a04e8886c211d77f8b0147c2140d350a..ec3a27538911435228c99b6ab2f2057308e2c2c5 100644 --- a/include/aidge/aidge.hpp +++ b/include/aidge/aidge.hpp @@ -59,6 +59,7 @@ #include "aidge/operator/ReduceMean.hpp" #include "aidge/operator/ReLU.hpp" #include "aidge/operator/Reshape.hpp" +#include "aidge/operator/Resize.hpp" #include "aidge/operator/Scaling.hpp" #include "aidge/operator/Slice.hpp" #include "aidge/operator/Softmax.hpp" diff --git a/include/aidge/operator/Resize.hpp b/include/aidge/operator/Resize.hpp index c498eaa2c1922a930c77327e2fbcffe61ff14473..3b8ce567adc1eceb092f942d5b3c103c156a1121 100644 --- a/include/aidge/operator/Resize.hpp +++ b/include/aidge/operator/Resize.hpp @@ -28,7 +28,7 @@ namespace Aidge { enum class ResizeAttr { NoROI, NoScales, NoSizes }; class Resize_Op : public OperatorTensor, - public Registrable<Resize_Op, std::string, std::unique_ptr<OperatorImpl>(const Resize_Op&)>, + public Registrable<Resize_Op, std::string, std::shared_ptr<OperatorImpl>(const Resize_Op&)>, public StaticAttributes<ResizeAttr, bool, bool, diff --git a/python_binding/operator/pybind_Resize.cpp b/python_binding/operator/pybind_Resize.cpp index 6f60762c477397ad143a59fdfcbbe559b0397a47..8a8a968bb34383053faaaced9230487814ab5dde 100644 --- a/python_binding/operator/pybind_Resize.cpp +++ b/python_binding/operator/pybind_Resize.cpp @@ -19,8 +19,10 @@ namespace Aidge { void init_Resize(py::module& m) { py::class_<Resize_Op, std::shared_ptr<Resize_Op>, Attributes, OperatorTensor>(m, "ResizeOp", py::multiple_inheritance()) - .def_static("get_inputs_name", &Resize_Op::getInputsName) - .def_static("get_outputs_name", &Resize_Op::getOutputsName); + .def_static("get_inputs_name", &Resize_Op::getInputsName) + .def_static("get_outputs_name", &Resize_Op::getOutputsName); + + declare_registrable<Resize_Op>(m, "ResizeOp"); m.def("Resize", &Resize, py::arg("nb_input_dims"), py::arg("no_roi"), py::arg("no_scales"), py::arg("no_sizes"), py::arg("name") = ""); }