From 7bd2c0f9c1fcb563e6a9c667273c0142e0d94d95 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 13 Feb 2024 19:05:18 +0100
Subject: [PATCH] Fixed parameter input for Memorize
---
include/aidge/operator/Memorize.hpp | 2 +-
unit_tests/operator/Test_MetaOperator.cpp | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/operator/Memorize.hpp b/include/aidge/operator/Memorize.hpp
index 53340f866..8991ccb44 100644
--- a/include/aidge/operator/Memorize.hpp
+++ b/include/aidge/operator/Memorize.hpp
@@ -38,7 +38,7 @@ public:
using attr = typename Attributes_::template attr<e>;
Memorize_Op(const unsigned int endStep)
- : OperatorTensor(Type, 2, 0, 2),
+ : OperatorTensor(Type, 1, 1, 2),
Attributes_(attr<MemorizeAttr::ScheduleStep>(0),
attr<MemorizeAttr::ForwardStep>(0),
attr<MemorizeAttr::EndStep>(endStep))
diff --git a/unit_tests/operator/Test_MetaOperator.cpp b/unit_tests/operator/Test_MetaOperator.cpp
index 328492d43..421dec584 100644
--- a/unit_tests/operator/Test_MetaOperator.cpp
+++ b/unit_tests/operator/Test_MetaOperator.cpp
@@ -60,7 +60,7 @@ TEST_CASE("[core/operators] MetaOperator", "[Operator]") {
microGraph->save("lstm", false, false);
REQUIRE(myLSTM->nbInputs() == 3 + 8 + 8);
- REQUIRE(myLSTM->nbData() == 3);
+ REQUIRE(myLSTM->nbData() == 1);
REQUIRE(myLSTM->nbOutputs() == 2);
std::shared_ptr<Tensor> myInput = std::make_shared<Tensor>();
--
GitLab