Skip to content
Snippets Groups Projects
Commit daa641e1 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

[NodeExport] Update outputs_dims.

parent fb5101e6
No related branches found
No related tags found
1 merge request!43[NodeExport] Add node export class
Pipeline #33411 passed
......@@ -41,13 +41,9 @@ class ExportNode(ABC):
else:
self.inputs_dims.append(None)
# TODO
for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node)
if child_node is not None:
self.outputs_dims.append(self.operator.output(idx).dims())
else:
self.outputs_dims.append(None)
self.outputs_dims.append(self.operator.output(idx).dims())
@abstractmethod
def export(self, export_folder:str, list_configs:list):
......
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