From 63622616c52d4fd01ef2b16b507f98cd60e73d0b Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Thu, 24 Sep 2020 12:43:15 +0200
Subject: [PATCH] [Core] add ModelWrapper.get_nodes_by_name
---
src/finn/core/modelwrapper.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py
index 42acc6fd2..11fda3da5 100644
--- a/src/finn/core/modelwrapper.py
+++ b/src/finn/core/modelwrapper.py
@@ -463,7 +463,11 @@ class ModelWrapper:
self.model.metadata_props.append(metadata_prop)
else:
metadata_prop.value = value
-
+
+ def get_nodes_by_name(self, op_name):
+ """Returns a list of nodes with specified name."""
+ return list(filter(lambda x: x.name == op_name, self.graph.node))
+
def get_nodes_by_op_type(self, op_type):
"""Returns a list of nodes with specified op_type."""
return list(filter(lambda x: x.op_type == op_type, self.graph.node))
--
GitLab