Adding clearInput method to supress Int32 inputs after PTQ

include/aidge/quantization_version.h

@@ -3,9 +3,9 @@
 
 namespace Aidge {
 static constexpr const int PROJECT_VERSION_MAJOR = 0;
-static constexpr const int PROJECT_VERSION_MINOR = 3;
-static constexpr const int PROJECT_VERSION_PATCH = 1;
-static constexpr const char * PROJECT_VERSION = "0.3.1";
-static constexpr const char * PROJECT_GIT_HASH = "418bc3e";
+static constexpr const int PROJECT_VERSION_MINOR = 2;
+static constexpr const int PROJECT_VERSION_PATCH = 0;
+static constexpr const char * PROJECT_VERSION = "0.2.0";
+static constexpr const char * PROJECT_GIT_HASH = "f50c860";
 }
 #endif // VERSION_H

src/PTQ/PTQ.cpp

@@ -57,7 +57,16 @@ bool isNotQuantized(std::shared_ptr<Node> node)
 {
     return (notQuantizedNodeTypes.find(node->type()) != notQuantizedNodeTypes.end());
 }
-
+void clearGraphViewInputNodes(std::shared_ptr<GraphView> graphView)
+{
+    for (std::shared_ptr<Aidge::Node> input_node: graphView->inputNodes())
+    {
+        for (Aidge::IOIndex_t index = input_node->getFirstFreeDataInput();index < input_node->getNbFreeDataInputs(); index++)
+        {
+            std::static_pointer_cast<OperatorTensor>(input_node->getOperator())->resetInput(index);
+        }
+    }
+}
 bool checkArchitecture(std::shared_ptr<GraphView> graphView)
 {
     std::set<std::string> otherNodeTypes({"Flatten", "Softmax", "BatchNorm2D", "ReLU", "Producer"});
@@ -1275,12 +1284,14 @@ void quantizeNetwork(std::shared_ptr<GraphView> graphView,
     }
     //Mandatory to handle all of the newly added connections!
     graphView->updateInputsOutputs();
+    
+    //Clearing input nodes
+    Log::notice("Clearing all input nodes ...");
+    clearGraphViewInputNodes(graphView);
+
     if (verbose)
         printScalingFactors(graphView);
-
-    if (useCuda)
-       // graphView->setBackend("cuda");
-
+    
     Log::notice(" Reseting the scheduler ...");
     SequentialScheduler scheduler(graphView);
     scheduler.resetScheduling();