diff --git a/aidge_export_cpp/templates/configuration/rescaling_config.jinja b/aidge_export_cpp/templates/configuration/rescaling_config.jinja new file mode 100644 index 0000000000000000000000000000000000000000..ee9c69e36c0c66c7e327824a8ea50ae1ff772896 --- /dev/null +++ b/aidge_export_cpp/templates/configuration/rescaling_config.jinja @@ -0,0 +1,15 @@ +{#- For name header -#} +#ifndef {{ name|upper }}_LAYER_H +#define {{ name|upper }}_LAYER_H + +{# For layer configuration -#} +{% include "./_def_io.jinja" %} +{% include "./_meminfo.jinja" %} + +#define {{ name|upper }}_NB_DATA {{ in_chan[0] * in_height[0] * in_width[0] }} + +// Activation +#define {{ name|upper }}_ACTIVATION {{ activation }} +{% include "./_rescaling.jinja" %} + +#endif /* {{ name|upper }}_LAYER_H */ diff --git a/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja b/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja new file mode 100644 index 0000000000000000000000000000000000000000..ce4ffb869c7f99ac789311a9bc98b926253f968f --- /dev/null +++ b/aidge_export_cpp/templates/kernel_forward/rescaling_forward.jinja @@ -0,0 +1,9 @@ +{% filter indent(width=4, first=False) %} +{% include "./_mem_offset.jinja" %} +rescaling_forward<{{name|upper}}_NB_DATA, + {{name|upper}}_ACTIVATION> + ({{in_name[0]}}, + {{out_name[0]}}, + {{name|upper}}_RESCALING); +{% include "./_save_outputs.jinja" %} +{% endfilter %}