Set output data type in constructor

src/operator/TopK.cpp

@@ -38,7 +38,10 @@ TopK_Op::TopK_Op(
           attr<TopKAttr::Axis>(axis),
           attr<TopKAttr::Largest>(largest),
           attr<TopKAttr::Sorted>(sorted),
-          attr<TopKAttr::K>(k))) {}
+          attr<TopKAttr::K>(k)))
+{
+    mOutputs[1]->setDataType(DataType::Int64);
+}
 
 TopK_Op::TopK_Op(const TopK_Op& op)
     : OperatorTensor(op),
@@ -79,7 +82,6 @@ bool Aidge::TopK_Op::forwardDims(bool allowDataDependency) {
         outDims[kAxis] = k();
         mOutputs[0]->resize(outDims);
         mOutputs[1]->resize(outDims);
-        mOutputs[1]->setDataType(DataType::Int64);
         return true;
     }