Make OperatorTensor::setDataType() more generic

include/aidge/operator/BatchNorm.hpp

@@ -105,16 +105,6 @@ public:
         getInput(4)->setBackend(name, device);
     }
 
-    void setDataType(const DataType& dt) const override {
-        mOutputs[0]->setDataType(dt);
-
-        // By default, automatically set data type for scale, shift, mean and variance
-        getInput(1)->setDataType(dt);
-        getInput(2)->setDataType(dt);
-        getInput(3)->setDataType(dt);
-        getInput(4)->setDataType(dt);
-    }
-
     static const std::vector<std::string> getInputsName() {
         return {"data_input", "scale", "shift", "mean", "variance"};
     }

include/aidge/operator/Conv.hpp

@@ -180,14 +180,6 @@ std::vector<std::pair<std::size_t, std::vector<DimSize_t>>> computeReceptiveFiel
         getInput(2)->setBackend(name, device);
     }
 
-    void setDataType(const DataType& dt) const override {
-        mOutputs[0]->setDataType(dt);
-
-        // By default, automatically set data type for weight and bias inputs
-        getInput(1)->setDataType(dt);
-        getInput(2)->setDataType(dt);
-    }
-
     static const std::vector<std::string> getInputsName(){
         return {"data_input", "weight", "bias"};
     }

include/aidge/operator/ConvDepthWise.hpp

@@ -174,14 +174,6 @@ public:
         getInput(2)->setBackend(name, device);
     }
 
-    void setDataType(const DataType& dt) const override {
-        mOutputs[0]->setDataType(dt);
-
-        // By default, automatically set data type for weight and bias inputs
-        getInput(1)->setDataType(dt);
-        getInput(2)->setDataType(dt);
-    }
-
     static const std::vector<std::string> getInputsName(){
         return {"data_input", "weight", "bias"};
     }

include/aidge/operator/FC.hpp

@@ -104,14 +104,6 @@ public:
         getInput(2)->setBackend(name, device);
     }
 
-    void setDataType(const DataType& dt) const override {
-        mOutputs[0]->setDataType(dt);
-
-        // By default, automatically set data type for weight and bias inputs
-        getInput(1)->setDataType(dt);
-        getInput(2)->setDataType(dt);
-    }
-
     static const std::vector<std::string> getInputsName(){
         return {"data_input", "weight", "bias"};
     }

src/operator/OperatorTensor.cpp

@@ -149,4 +149,8 @@ void Aidge::OperatorTensor::setDataType(const DataType& dataType) const {
     for (IOIndex_t i = 0; i < nbOutputs(); ++i) {
         getOutput(i)->setDataType(dataType);
     }
+
+    for (IOIndex_t i = nbData(); i < nbInputs(); ++i) {
+        getInput(i)->setDataType(dataType);
+    }
 }
\ No newline at end of file