From 9de0607ef5c1bb8bfb24b25d53b90c7d5b900847 Mon Sep 17 00:00:00 2001
From: Jerome Hue <jerome.hue@cea.fr>
Date: Thu, 9 Jan 2025 14:45:47 +0100
Subject: [PATCH] fix: Change init input of Memorize from Param to Data input
category
---
src/operator/Memorize.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/Memorize.cpp b/src/operator/Memorize.cpp
index cd4a48081..c4f0bc4bf 100644
--- a/src/operator/Memorize.cpp
+++ b/src/operator/Memorize.cpp
@@ -76,7 +76,7 @@ void Aidge::Memorize_OpImpl::forward() {
const std::string Aidge::Memorize_Op::Type = "Memorize";
Aidge::Memorize_Op::Memorize_Op(const std::uint32_t endStep)
- : OperatorTensor(Type, {InputCategory::Data, InputCategory::Param}, 2),
+ : OperatorTensor(Type, {InputCategory::Data, InputCategory::Data}, 2),
mAttributes(std::make_shared<Attributes_>(
attr<MemorizeAttr::ScheduleStep>(0),
attr<MemorizeAttr::ForwardStep>(0),
--
GitLab