Skip to content

Commit f61f635

Browse files
authored
Merge pull request #58 from tensil-ai/peter/sc-461/optimize-onnx-node-traversal
Optimize ONNX node traversal
2 parents 3faa830 + 7d90c15 commit f61f635

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

tools/src/tensil/tools/compiler/OnnxFrontend.scala

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -193,15 +193,16 @@ class OnnxFrontend(
193193
.map(outputNodeNames(_))
194194
.flatten
195195
.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
196+
.foldLeft(prev)((prev, nodeName) =>
197+
if (!prev.contains(nodeName))
198+
recursiveTraverse(
199+
prev,
200+
nodeProtos(nodeName).input.toSet
201+
.diff(tensorProtos.keySet ++ inputValueInfoProtos.keySet)
202+
.toSeq
203+
) :+ nodeName
204+
else prev
205+
)
205206
}
206207

207208
/*

0 commit comments

Comments
 (0)