From 02fdf34aaff55669667d6571cb8b630bc2780551 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Tue, 21 Nov 2023 15:28:09 +0100
Subject: [PATCH] add more Registrar dims for Transpose
---
include/aidge/operator/Transpose.hpp | 6 +++---
python_binding/operator/pybind_Transpose.cpp | 10 +++++++++-
2 files changed, 12 insertions(+), 4 deletions(-)
diff --git a/include/aidge/operator/Transpose.hpp b/include/aidge/operator/Transpose.hpp
index d6170396c..b19ec6473 100644
--- a/include/aidge/operator/Transpose.hpp
+++ b/include/aidge/operator/Transpose.hpp
@@ -103,19 +103,19 @@ class Transpose_Op : public Operator,
inline std::shared_ptr<Tensor> getInput(const IOIndex_t inputIdx) const override final {
- assert((inputIdx == 0) && "Transpose Operators supports only 1 input");
+ assert((inputIdx == 0) && "Transpose operator supports only 1 input");
(void) inputIdx; // avoid unused warning
return mInput;
}
inline std::shared_ptr<Tensor> getOutput(const IOIndex_t outputIdx) const override final {
- assert((outputIdx == 0) && "Transpose Operator has only 1 output");
+ assert((outputIdx == 0) && "Transpose operator has only 1 output");
(void) outputIdx; // avoid unused warning
return mOutput;
}
std::shared_ptr<Data> getRawInput(const IOIndex_t inputIdx) const override final {
- assert((inputIdx == 0) && "Transpose Operator supports only 1 input");
+ assert((inputIdx == 0) && "Transpose operator supports only 1 input");
return std::static_pointer_cast<Data>(mInput);
}
std::shared_ptr<Data> getRawOutput(const IOIndex_t outputIdx) const override final {
diff --git a/python_binding/operator/pybind_Transpose.cpp b/python_binding/operator/pybind_Transpose.cpp
index 59895cc44..2d9ce1796 100644
--- a/python_binding/operator/pybind_Transpose.cpp
+++ b/python_binding/operator/pybind_Transpose.cpp
@@ -26,7 +26,7 @@ namespace py = pybind11;
namespace Aidge {
template <DimIdx_t DIM>
-void init_Transpose(py::module &m) {
+void declare_Transpose(py::module &m) {
py::class_<Transpose_Op<DIM>, std::shared_ptr<Transpose_Op<DIM>>, Operator, Attributes>(
m, ("TransposeOp" + std::to_string(DIM) + "D").c_str(), py::multiple_inheritance())
.def("get_inputs_name", &Transpose_Op<DIM>::getInputsName)
@@ -41,4 +41,12 @@ void init_Transpose(py::module &m) {
}
+void init_Transpose(py::module &m) {
+ declare_Transpose<2>(m);
+ declare_Transpose<3>(m);
+ declare_Transpose<4>(m);
+ declare_Transpose<5>(m);
+ declare_Transpose<6>(m);
+
+}
} // namespace Aidge
--
GitLab