From bc09c6acc560191e0d941bfbe9d08dc1cb518058 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 28 Mar 2025 15:50:45 +0100
Subject: [PATCH] Set output data type in constructor
---
src/operator/TopK.cpp | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/operator/TopK.cpp b/src/operator/TopK.cpp
index 52bb37554..660865100 100644
--- a/src/operator/TopK.cpp
+++ b/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;
}
--
GitLab