We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 3faa830 + 7d90c15 commit f61f635Copy full SHA for f61f635
tools/src/tensil/tools/compiler/OnnxFrontend.scala
@@ -193,15 +193,16 @@ class OnnxFrontend(
193
.map(outputNodeNames(_))
194
.flatten
195
.distinct
196
- .foldLeft(prev)((prev, nodeName) => {
197
- recursiveTraverse(
198
- nodeName +: prev,
199
- nodeProtos(nodeName).input.toSet
200
- .diff(tensorProtos.keySet ++ inputValueInfoProtos.keySet)
201
- .toSeq
202
- )
203
- })
204
- .distinct
+ .foldLeft(prev)((prev, nodeName) =>
+ if (!prev.contains(nodeName))
+ recursiveTraverse(
+ prev,
+ nodeProtos(nodeName).input.toSet
+ .diff(tensorProtos.keySet ++ inputValueInfoProtos.keySet)
+ .toSeq
+ ) :+ nodeName
+ else prev
205
+ )
206
}
207
208
/*
0 commit comments