From fd9567884817a64d5191b61580cd1ced483ec12b Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 16 May 2024 12:39:22 +0000
Subject: [PATCH] Add support for export ofoperator Sub.
---
.../aidge_export_aidge/operator_export/sub.py | 21 +++++++++++++++++++
.../templates/graph_ctor/sub.jinja | 9 ++++++++
2 files changed, 30 insertions(+)
create mode 100644 aidge_core/aidge_export_aidge/operator_export/sub.py
create mode 100644 aidge_core/aidge_export_aidge/templates/graph_ctor/sub.jinja
diff --git a/aidge_core/aidge_export_aidge/operator_export/sub.py b/aidge_core/aidge_export_aidge/operator_export/sub.py
new file mode 100644
index 000000000..efcdd0924
--- /dev/null
+++ b/aidge_core/aidge_export_aidge/operator_export/sub.py
@@ -0,0 +1,21 @@
+from aidge_core.aidge_export_aidge.utils import operator_register, parse_node_input
+from aidge_core import ExportNode, generate_str
+from aidge_core.aidge_export_aidge import ROOT_EXPORT
+from pathlib import Path
+
+@operator_register("Sub")
+class Sub(ExportNode):
+ def __init__(self, node):
+ super().__init__(node)
+
+ def export(self, export_folder:Path, list_configs:list):
+ return list_configs
+
+ def forward(self, list_actions:list):
+ list_actions.append(generate_str(
+ ROOT_EXPORT / "templates/graph_ctor/sub.jinja",
+ name=self.name,
+ inputs=parse_node_input(self.node.inputs()),
+ **self.attributes
+ ))
+ return list_actions
diff --git a/aidge_core/aidge_export_aidge/templates/graph_ctor/sub.jinja b/aidge_core/aidge_export_aidge/templates/graph_ctor/sub.jinja
new file mode 100644
index 000000000..d9417751e
--- /dev/null
+++ b/aidge_core/aidge_export_aidge/templates/graph_ctor/sub.jinja
@@ -0,0 +1,9 @@
+{% filter indent(width=4, first=False) %}
+/*** {{name|upper}} ***/
+std::shared_ptr<Aidge::Node> {{name}} =
+ Aidge::Sub(
+ "{{name}}"
+ );
+{% include "./_set_input.jinja" %}
+graph->add({{name}});
+{% endfilter %}
--
GitLab