Skip to content
Snippets Groups Projects
Commit c2cfa8d1 authored by Mickael GUIBERT's avatar Mickael GUIBERT
Browse files

[Fix](Producer) Add ignore attribute to False by default

parent 3f58cc01
No related branches found
No related tags found
1 merge request!35New feature, bug fixes, code modifications to create a CPP export of Resnet18 (quantified model)
Pipeline #70892 failed
......@@ -47,7 +47,10 @@ class ProducerCPP(ExportNode):
def __init__(self, node, mem_info):
super().__init__(node, mem_info)
self.values = np.array(self.operator.get_output(0))
self.ignore = node.attributes().ignore
if node.attributes().has_attr("ignore"):
self.ignore = node.attributes().ignore
else:
self.ignore = False
if len(self.values.shape) == 4: # Note: export in HWC
self.values = np.transpose(self.values, (0, 2, 3, 1))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment