From 414578c9783ba63a0b5e8c5f14b3e9e2f83ee5dd Mon Sep 17 00:00:00 2001
From: Tobi-Alonso <tobi.alonso@gmail.com>
Date: Fri, 22 May 2020 12:48:34 +0100
Subject: [PATCH] [ModelWrapper] Small change to find_producer that should make
it run twice as fast on average
---
src/finn/core/modelwrapper.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py
index cdf99dc3b..6bf468368 100644
--- a/src/finn/core/modelwrapper.py
+++ b/src/finn/core/modelwrapper.py
@@ -259,11 +259,10 @@ class ModelWrapper:
def find_producer(self, tensor_name):
"""Finds and returns the node that produces the tensor with given name."""
- ret = None
for x in self._model_proto.graph.node:
if tensor_name in x.output:
- ret = x
- return ret
+ return x
+ return None
def find_upstream(self, tensor_name, finder_fxn):
"""Follow the producer chain upstream, calling finder_fxn on each upstream
--
GitLab