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): ...@@ -41,13 +41,9 @@ class ExportNode(ABC):
else: else:
self.inputs_dims.append(None) self.inputs_dims.append(None)
# TODO
for idx, child_node in enumerate(self.node.get_children()): for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node) self.outputs.append(child_node)
if child_node is not None: self.outputs_dims.append(self.operator.output(idx).dims())
self.outputs_dims.append(self.operator.output(idx).dims())
else:
self.outputs_dims.append(None)
@abstractmethod @abstractmethod
def export(self, export_folder:str, list_configs:list): 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