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

Update ReLU with new ExportNodeCpp

parent f2fb39f6
No related branches found
No related tags found
3 merge requests!27v0.2.0,!22v0.4.0,!15Export refactor
......@@ -4,7 +4,7 @@ import numpy as np
from pathlib import Path
from jinja2 import Environment, FileSystemLoader
from aidge_core import ExportNode
from aidge_core import ExportNode, ExportNodeCpp
from aidge_core.export_utils.code_generation import *
from aidge_export_cpp.utils import ROOT, operator_register
from aidge_export_cpp.utils.converter import numpy_dtype2ctype
......@@ -84,35 +84,17 @@ class ProducerCPP(ExportNode):
@operator_register("ReLU")
class ReLUCPP(ExportNode):
class ReLUCPP(ExportNodeCpp):
def __init__(self, node):
super().__init__(node)
def export(self, export_folder:Path, list_configs:list):
copyfile(str(ROOT / "kernels" / "activation.hpp"),
str(export_folder / "include" / "kernels"))
list_configs.append("kernels/activation.hpp")
list_configs.append(f"layers/{self.attributes['name']}.h")
generate_file(
str(export_folder / "layers" / f"{self.attributes['name']}.h"),
str(ROOT / "templates" / "configuration" / "activation_config.jinja"),
activation="Rectifier",
rescaling="NoScaling",
**self.attributes
)
return list_configs
def forward(self, list_actions:list):
list_actions.append(generate_str(
str(ROOT / "templates" / "kernel_forward" / "activation_forward.jinja"),
**self.attributes
))
return list_actions
self.attributes["activation"] = "Rectifier"
self.attributes["rescaling"] = "NoScaling"
self.config_template = str(ROOT / "templates" / "configuration" / "activation_config.jinja")
self.forward_template = str(ROOT / "templates" / "kernel_forward" / "activation_forward.jinja")
self.include_list = []
self.kernels_to_copy = [
str(ROOT / "kernels" / "activation.hpp"),
]
@operator_register("Conv")
......
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