Skip to content
Snippets Groups Projects

[NodeExport] Add node export class

Merged Cyril Moineau requested to merge NodeExport into main
All threads resolved!
1 file
+ 1
5
Compare changes
  • Side-by-side
  • Inline
@@ -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):
Loading