From d6cd3662abf0338a07528f818cd819bc2d14a607 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 9 Apr 2024 12:05:18 +0200
Subject: [PATCH] Fixed missing input for conv
---
aidge_core/unit_tests/test_impl.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/aidge_core/unit_tests/test_impl.py b/aidge_core/unit_tests/test_impl.py
index 4aacfafd7..6b83b048c 100644
--- a/aidge_core/unit_tests/test_impl.py
+++ b/aidge_core/unit_tests/test_impl.py
@@ -52,6 +52,7 @@ class test_OperatorImpl(unittest.TestCase):
self.assertTrue("cpu" in aidge_core.get_keys_ConvOp2D())
conv = aidge_core.Conv2D(2,2,[1,1], name="Conv0")
conv.get_operator().set_backend("cpu")
+ conv.get_operator().set_input(0, aidge_core.Tensor(np.arange(9).reshape(1,1,3,3)))
conv.get_operator().forward()
self.assertEqual(GLOBAL_CPT, 1)
@@ -65,6 +66,7 @@ class test_OperatorImpl(unittest.TestCase):
conv = aidge_core.Conv2D(2,2,[1,1], name="Conv0")
model = aidge_core.sequential([conv])
model.set_backend("cpu")
+ conv.get_operator().set_input(0, aidge_core.Tensor(np.arange(9).reshape(1,1,3,3)))
conv.get_operator().forward()
self.assertEqual(GLOBAL_CPT, 1)
--
GitLab