Specify in-place operators

include/aidge/operator/Flatten.hpp

@@ -38,6 +38,10 @@ public:
      */
     Flatten_OpImpl(const Operator& op, const std::string& backend = ""): OperatorImpl(op, backend) {}
 
+    std::shared_ptr<ProdConso> getProdConso() const override {
+        return std::make_shared<ProdConso>(mOp, true);  // Flatten is an in-place operation!
+    }
+
     /**
      * @brief Compute the forward pass of the Flatten operation.
      */

include/aidge/operator/Identity.hpp

@@ -29,6 +29,11 @@ namespace Aidge {
 class Identity_OpImpl : public OperatorImpl {
 public:
     Identity_OpImpl(const Operator& op, const std::string& backend = ""): OperatorImpl(op, backend) {}
+
+    std::shared_ptr<ProdConso> getProdConso() const override {
+        return std::make_shared<ProdConso>(mOp, true);  // Identity is an in-place operation!
+    }
+
     void forward() override;
 };
 

include/aidge/operator/Reshape.hpp

@@ -37,6 +37,10 @@ public:
     Reshape_OpImpl(const Operator& op, const std::string& backend = "")
         : OperatorImpl(op, backend) {}
 
+    std::shared_ptr<ProdConso> getProdConso() const override {
+        return std::make_shared<ProdConso>(mOp, true);  // Reshape is an in-place operation!
+    }
+
     /**
      * @brief Perform the forward operation for the reshape.
      */

include/aidge/operator/Squeeze.hpp

@@ -38,6 +38,11 @@ class Squeeze_OpImpl : public OperatorImpl {
 public:
   Squeeze_OpImpl(const Operator &op, const std::string &backend = "")
       : OperatorImpl(op, backend) {}
+
+  std::shared_ptr<ProdConso> getProdConso() const override {
+      return std::make_shared<ProdConso>(mOp, true);  // Squeeze is an in-place operation!
+  }
+
   void forward() override;
 };
 } // namespace Aidge

include/aidge/operator/Unsqueeze.hpp

@@ -35,6 +35,11 @@ class Unsqueeze_OpImpl : public OperatorImpl {
 public:
   Unsqueeze_OpImpl(const Operator &op, const std::string &backend = "")
       : OperatorImpl(op, backend) {}
+
+  std::shared_ptr<ProdConso> getProdConso() const override {
+      return std::make_shared<ProdConso>(mOp, true);  // Unsqueeze is an in-place operation!
+  }
+
   void forward() override;
 };
 }  // namespace Aidge