From de3fd61b5f8b5e9de2d54d4614bb83f6787e9d43 Mon Sep 17 00:00:00 2001
From: Maxence Naud <maxence.naud@cea.fr>
Date: Mon, 25 Sep 2023 12:57:06 +0000
Subject: [PATCH] Apply 1 suggestion(s) to 1 file(s)
---
include/aidge/operator/Scaling.hpp | 12 ++++++++++--
1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/include/aidge/operator/Scaling.hpp b/include/aidge/operator/Scaling.hpp
index 6140ff70c..114ce5c0a 100644
--- a/include/aidge/operator/Scaling.hpp
+++ b/include/aidge/operator/Scaling.hpp
@@ -71,8 +71,16 @@ public:
}
- inline Tensor& input(__attribute__((unused)) const IOIndex_t inputIdx) const override final { return *(mInput.get()); }
- inline Tensor& output(__attribute__((unused)) const IOIndex_t outputIdx) const override final { return *(mOutput.get()); }
+ inline Tensor& input(const IOIndex_t inputIdx) const override final {
+ assert((inputIdx == 0) && "Scaling Operator has only 1 input");
+ (void) inputIdx; // avoid unused warning
+ return *(mInput.get());
+ }
+ inline Tensor& output(const IOIndex_t outputIdx) const override final {
+ assert((outputIdx == 0) && "Scaling Operator has only 1 output");
+ (void) outputIdx; // avoid unused warning
+ return *(mOutput.get());
+ }
inline std::shared_ptr<Tensor> getInput(const IOIndex_t inputIdx) const override final {
--
GitLab