Skip to content

Commit 42df7a0

Browse files
author
Harish
committed
Revised version for QuantizeGraph
1 parent 3e132fe commit 42df7a0

File tree

3 files changed

+175
-191
lines changed

3 files changed

+175
-191
lines changed

src/qonnx/core/modelwrapper.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -542,6 +542,15 @@ def get_node_index(self, node):
542542
except ValueError:
543543
return None
544544

545+
def get_node_from_name(self, node_name):
546+
"""Returns the node with the specified name."""
547+
try:
548+
for node in self.graph.node:
549+
if node.name == node_name:
550+
return node
551+
except ValueError:
552+
return None
553+
545554
def get_tensor_layout(self, tensor_name):
546555
"""Returns the data layout annotation of tensor with given name.
547556
The data layout is expressed as a list of strings with as many

0 commit comments

Comments
 (0)