diff --git a/aidge_export_cpp/templates/configuration/matmul_config.jinja b/aidge_export_cpp/templates/configuration/matmul_config.jinja index 0d941c0eca56278351b20fb0ee991832dafaf704..d0d4958e505b3208598fe387bba357a0c3d84602 100644 --- a/aidge_export_cpp/templates/configuration/matmul_config.jinja +++ b/aidge_export_cpp/templates/configuration/matmul_config.jinja @@ -1,6 +1,7 @@ {#- For name header -#} #ifndef {{ name|upper }}_LAYER_H #define {{ name|upper }}_LAYER_H +#include "network/rescaling_utils.hpp" {% include "./_def_io.jinja" %} {% include "./_meminfo.jinja" %} diff --git a/aidge_export_cpp/templates/configuration/rescaling_config.jinja b/aidge_export_cpp/templates/configuration/rescaling_config.jinja index ee9c69e36c0c66c7e327824a8ea50ae1ff772896..6f4e3ad80e4f72d180a9add5fd7978181a71031d 100644 --- a/aidge_export_cpp/templates/configuration/rescaling_config.jinja +++ b/aidge_export_cpp/templates/configuration/rescaling_config.jinja @@ -1,6 +1,7 @@ {#- For name header -#} #ifndef {{ name|upper }}_LAYER_H #define {{ name|upper }}_LAYER_H +#include "network/rescaling_utils.hpp" {# For layer configuration -#} {% include "./_def_io.jinja" %}