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

Path to Operator header file should be specified by hand because the file name...

Path to Operator header file should be specified by hand because the file name may not be exactly named like the associated Operator's type
parent e3207217
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
......@@ -93,7 +93,7 @@ def serialize_to_cpp(export_folder: str,
node, None)
set_operator.add(node.type())
# set_operator.add(node.type())
# TODO: list_configs and list_actions don't need to be passed by argument
# Export the configuration
......
......@@ -11,7 +11,7 @@ class Conv(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/conv.jinja")
self.forward_template = str(
ROOT_EXPORT /"templates/graph_ctor/conv.jinja")
self.include_list = []
self.include_list = ["aidge/operator/Conv.hpp"]
self.kernels_to_copy = []
self.config_path = "include/attributes"
self.config_extension = "hpp"
......@@ -12,7 +12,7 @@ class FC(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/fc.jinja")
self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/fc.jinja")
self.include_list = []
self.include_list = ["aidge/operator/FC.hpp"]
self.kernels_to_copy = []
self.config_path = "include/attributes"
self.config_extension = "hpp"
......@@ -11,7 +11,7 @@ class MaxPooling(ExportNodeCpp):
ROOT_EXPORT / "templates/attributes/maxpooling.jinja")
self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/maxpooling.jinja")
self.include_list = []
self.include_list = ["aidge/operator/MaxPooling.hpp"]
self.kernels_to_copy = []
self.config_path = "include/attributes"
self.config_extension = "hpp"
......@@ -23,7 +23,7 @@ class Producer(ExportNodeCpp):
ROOT_EXPORT / "templates/graph_ctor/producer.jinja")
self.attributes["tensor_name"] = f"{child.name()}_{in_idx}"
self.attributes["values"] = str(self.operator.get_output(0))
self.include_list = []
self.include_list = ["aidge/operator/Producer.hpp"]
self.kernels_to_copy = []
self.config_path = "include/attributes"
self.config_extension = "hpp"
......
......@@ -10,5 +10,5 @@ class ReLU(ExportNodeCpp):
self.config_template = ""
self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/relu.jinja")
self.include_list = []
self.include_list = ["aidge/operator/ReLU.hpp"]
self.kernels_to_copy = []
......@@ -10,5 +10,5 @@ class Sub(ExportNodeCpp):
self.config_template = ""
self.forward_template = str(
ROOT_EXPORT / "templates/graph_ctor/sub.jinja")
self.include_list = []
self.include_list = ["aidge/operator/Sub.hpp"]
self.kernels_to_copy = []
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