diff --git a/aidge_export_cpp/templates/configuration/convolution_config.jinja b/aidge_export_cpp/templates/configuration/convolution_config.jinja index a4a2462d1f475424d799f38e52075fafb333c0d0..beb7de3f081dbbcdd8c5ff69864547444d1e62f5 100644 --- a/aidge_export_cpp/templates/configuration/convolution_config.jinja +++ b/aidge_export_cpp/templates/configuration/convolution_config.jinja @@ -5,14 +5,14 @@ {# For layer configuration -#} {% include "./_def_io.jinja" %} {% include "./_meminfo.jinja" %} -#define {{ name|upper }}_PADDING_Y {{ padding[1] }} -#define {{ name|upper }}_PADDING_X {{ padding[0] }} -#define {{ name|upper }}_STRIDE_Y {{ stride_dims[1] }} -#define {{ name|upper }}_STRIDE_X {{ stride_dims[0] }} -#define {{ name|upper }}_DILATION_Y {{ dilation_dims[1] }} -#define {{ name|upper }}_DILATION_X {{ dilation_dims[0] }} -#define {{ name|upper }}_KERNEL_HEIGHT {{ kernel_dims[1] }} -#define {{ name|upper }}_KERNEL_WIDTH {{ kernel_dims[0] }} +#define {{ name|upper }}_PADDING_Y {{ padding[0] }} +#define {{ name|upper }}_PADDING_X {{ padding[1] }} +#define {{ name|upper }}_STRIDE_Y {{ stride_dims[0] }} +#define {{ name|upper }}_STRIDE_X {{ stride_dims[1] }} +#define {{ name|upper }}_DILATION_Y {{ dilation_dims[0] }} +#define {{ name|upper }}_DILATION_X {{ dilation_dims[1] }} +#define {{ name|upper }}_KERNEL_HEIGHT {{ kernel_dims[0] }} +#define {{ name|upper }}_KERNEL_WIDTH {{ kernel_dims[1] }} #define {{ name|upper }}_ACTIVATION {{ activation }} static const {{ rescaling }} {{ name|upper }}_RESCALING = {}; diff --git a/aidge_export_cpp/templates/configuration/pooling_config.jinja b/aidge_export_cpp/templates/configuration/pooling_config.jinja index afdb51f21f1cea5679d6dc6dfab57ba313ed11a5..57608fcb9c8071a1c686b10565d6adab78735079 100644 --- a/aidge_export_cpp/templates/configuration/pooling_config.jinja +++ b/aidge_export_cpp/templates/configuration/pooling_config.jinja @@ -5,12 +5,12 @@ {# For layer configuration -#} {% include "./_def_io.jinja" %} {% include "./_meminfo.jinja" %} -#define {{ name|upper }}_PADDING_Y {{ padding[1] }} -#define {{ name|upper }}_PADDING_X {{ padding[0] }} -#define {{ name|upper }}_STRIDE_Y {{ stride_dims[1] }} -#define {{ name|upper }}_STRIDE_X {{ stride_dims[0] }} -#define {{ name|upper }}_KERNEL_HEIGHT {{ kernel_dims[1] }} -#define {{ name|upper }}_KERNEL_WIDTH {{ kernel_dims[0] }} +#define {{ name|upper }}_PADDING_Y {{ padding[0] }} +#define {{ name|upper }}_PADDING_X {{ padding[1] }} +#define {{ name|upper }}_STRIDE_Y {{ stride_dims[0] }} +#define {{ name|upper }}_STRIDE_X {{ stride_dims[1] }} +#define {{ name|upper }}_KERNEL_HEIGHT {{ kernel_dims[0] }} +#define {{ name|upper }}_KERNEL_WIDTH {{ kernel_dims[1] }} #define {{ name|upper }}_POOLING_TYPE {{ pool_type }} #define {{ name|upper }}_ACTIVATION {{ activation }}