Skip to content
Snippets Groups Projects
Commit eb0e9ed9 authored by Maxence Naud's avatar Maxence Naud
Browse files

fix node_export.py for new scheduling that does not contain input Producer

parent cc2136e4
No related branches found
No related tags found
1 merge request!105version 0.2.0
Pipeline #42991 passed
......@@ -39,7 +39,11 @@ class ExportNode(ABC):
if parent_node is not None:
self.inputs_dims.append(self.operator.get_input(idx).dims())
else:
self.inputs_dims.append(None)
print(self.operator.get_input(idx))
if self.operator.get_input(idx) is not None:
self.inputs_dims.append(self.operator.get_input(idx).dims())
else:
self.inputs_dims.append(None)
for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node)
......
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