From ae03e3078243d78aee3ab25142bfa6acae652190 Mon Sep 17 00:00:00 2001 From: Olivier BICHLER <olivier.bichler@cea.fr> Date: Fri, 18 Apr 2025 14:56:44 +0200 Subject: [PATCH] Fixed indexing --- .../configuration/convolution_config.jinja | 16 ++++++++-------- .../templates/configuration/pooling_config.jinja | 12 ++++++------ 2 files changed, 14 insertions(+), 14 deletions(-) diff --git a/aidge_export_cpp/templates/configuration/convolution_config.jinja b/aidge_export_cpp/templates/configuration/convolution_config.jinja index a4a2462..beb7de3 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 afdb51f..57608fc 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 }} -- GitLab