renamed functions to reflect kernel format (CHW)

aidge_export_arm_cortexm/_Aidge_Arm/kernels/Activation/Softmax/aidge_softmax_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/Activation/Softmax/aidge_softmax_chw_float32.c

 #include <math.h>
 
-void aidge_softmax_float32 (float* inputs, 
+void aidge_softmax_chw_float32 (float* inputs, 
                             float* outputs,
                             int inputDims[],
                             int axis,

aidge_export_arm_cortexm/_Aidge_Arm/kernels/BatchNorm/aidge_batchnorm2d_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/BatchNorm/aidge_batchnorm2d_chw_float32.c

 #include <math.h>
 
 
-void aidge_batchnorm2d_float32 (float* inputs,
+void aidge_batchnorm2d_chw_float32 (float* inputs,
                                 float* outputs,
                                 float* input_mean,
                                 float* input_var,

aidge_export_arm_cortexm/_Aidge_Arm/kernels/MatMul/aidge_matmul_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/MatMul/aidge_matmul_chw_float32.c

-void aidge_matmul_float32(float* input_a, 
+void aidge_matmul_chw_float32(float* input_a, 
                        float* input_b, 
                        float* output,
                        int dim_a[],

aidge_export_arm_cortexm/_Aidge_Arm/kernels/Reshape/aidge_reshape_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/Reshape/aidge_reshape_chw_float32.c

 
-void aidge_reshape_float32(float* inputs,
+void aidge_reshape_chw_float32(float* inputs,
                            float* outputs,
                            unsigned int size)
 {

aidge_export_arm_cortexm/_Aidge_Arm/kernels/Transform/Gather/aidge_gather_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/Transform/Gather/aidge_gather_chw_float32.c

 
 
-void aidge_gather_float32 (float* inputs, 
+void aidge_gather_chw_float32 (float* inputs, 
                             float* outputs,
                             int axis,
                             int indices[],

aidge_export_arm_cortexm/_Aidge_Arm/kernels/Transform/Transpose/aidge_transpose_float32.c → aidge_export_arm_cortexm/_Aidge_Arm/kernels/Transform/Transpose/aidge_transpose_chw_float32.c

-void aidge_transpose_float32   (float* inputs,    
+void aidge_transpose_chw_float32   (float* inputs,    
                                 float* outputs,
                                 int input_dims[],
                                 int perm[],

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/activation_chw.jinja

+    aidge_{{activation_type|lower}}_chw_{{dataformat}}({{input_name}}, {{output_name}}, {% if activation_type is eq('softmax') %} {{input_name}}_DIMS, {{name|upper}}_AXIS, {{name|upper}}_INPUT_DIMS_SIZE,{% endif %} {{name|upper}}_OUTPUTS_SIZE);
\ No newline at end of file

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/batchnorm2d.jinja

-aidge_batchnorm2d_{{dataformat}} ({{input_name}}, {{output_name}}, {{running_mean_name}}, {{running_var_name}}, {{weight_name}}, {{bias_name}}, {{ name|upper }}_EPSILON, {{ name|upper }}_NB_CHANNELS, {{ name|upper }}_CHANNELS_WIDTH, {{ name|upper }}_CHANNELS_HEIGHT);
+aidge_batchnorm2d_chw_{{dataformat}} ({{input_name}}, {{output_name}}, {{running_mean_name}}, {{running_var_name}}, {{weight_name}}, {{bias_name}}, {{ name|upper }}_EPSILON, {{ name|upper }}_NB_CHANNELS, {{ name|upper }}_CHANNELS_WIDTH, {{ name|upper }}_CHANNELS_HEIGHT);
\ No newline at end of file

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/gather.jinja

-aidge_gather_{{dataformat}} ({{input_name}}, {{output_name}}, {{name|upper}}_AXIS, {{name}}_INDEXES , {{input_name}}_DIMS, {{name|upper}}_INPUT_DIMS_SIZE,{{name|upper}}_INDEXES_DIMS_SIZE,{{name|upper}}_OUTPUT_SIZE);
+aidge_gather_chw_{{dataformat}} ({{input_name}}, {{output_name}}, {{name|upper}}_AXIS, {{name}}_INDEXES , {{input_name}}_DIMS, {{name|upper}}_INPUT_DIMS_SIZE,{{name|upper}}_INDEXES_DIMS_SIZE,{{name|upper}}_OUTPUT_SIZE);
\ No newline at end of file

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/matmul.jinja

-aidge_matmul_{{dataformat}} ({{inputa_name}}, {{inputb_name}}, {{output_name}}, {{inputa_name}}_DIMS, {{inputb_name}}_DIMS, {{output_name}}_DIMS ,{{name|upper}}_INPUT_A_DIMS_SIZE,{{name|upper}}_INPUT_B_DIMS_SIZE,{{name|upper}}_OUTPUT_DIMS_SIZE);
+aidge_matmul_chw_{{dataformat}} ({{inputa_name}}, {{inputb_name}}, {{output_name}}, {{inputa_name}}_DIMS, {{inputb_name}}_DIMS, {{output_name}}_DIMS ,{{name|upper}}_INPUT_A_DIMS_SIZE,{{name|upper}}_INPUT_B_DIMS_SIZE,{{name|upper}}_OUTPUT_DIMS_SIZE);
\ No newline at end of file

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/reshape.jinja

-aidge_reshape_{{dataformat}}({{input_name}}, {{output_name}}, {{name|upper}}_OUTPUTS_SIZE);
+aidge_reshape_chw_{{dataformat}}({{input_name}}, {{output_name}}, {{name|upper}}_OUTPUTS_SIZE);
\ No newline at end of file

aidge_export_arm_cortexm/_Aidge_Arm/templates/kernel/transpose.jinja

-aidge_transpose_{{dataformat}} ({{input_name}}, {{output_name}},{{input_name}}_DIMS, {{name}}_PERMUTATIONS, {{output_name}}_DIMS, {{name|upper}}_OUTPUT_DIMS_SIZE, {{name|upper}}_OUTPUT_SIZE);
+aidge_transpose_chw_{{dataformat}} ({{input_name}}, {{output_name}},{{input_name}}_DIMS, {{name}}_PERMUTATIONS, {{output_name}}_DIMS, {{name|upper}}_OUTPUT_DIMS_SIZE, {{name|upper}}_OUTPUT_SIZE);
\ No newline at end of file

aidge_export_arm_cortexm/operators.py

@@ -596,7 +596,6 @@ class FC_ARMCortexM(ExportNode):
         #     # transform to [nb_outputs, 1, 1]
         #     self.outputs_dims[0] = [self.outputs_dims[0][1], 1, 1]
 
-
     def export(self, export_folder:Path, list_configs:list):
 
         # Export weights to NHWC format
@@ -873,7 +872,7 @@ class Softmax_ARMCortexM(ExportNode):
         
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Activation" / "Softmax" / "aidge_softmax_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Activation" / "Softmax" / "aidge_softmax_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
 
             generate_file(
@@ -895,7 +894,7 @@ class Softmax_ARMCortexM(ExportNode):
 
         if self.library == "aidge":
             list_actions.append(generate_str(
-                str(ROOT / "_Aidge_Arm" / "templates" / "kernel" / "activation.jinja"),
+                str(ROOT / "_Aidge_Arm" / "templates" / "kernel" / "activation_chw.jinja"),
                 name=self.name,
                 activation_type="softmax",
                 dataformat=self.dataformat,
@@ -933,7 +932,7 @@ class BatchNorm2D_ARMCortexM(ExportNode):
         
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "BatchNorm" / "aidge_batchnorm2d_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "BatchNorm" / "aidge_batchnorm2d_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
 
             generate_file(
@@ -1028,7 +1027,7 @@ class Reshape_ARMCortexM(ExportNode):
 
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Reshape" / "aidge_reshape_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Reshape" / "aidge_reshape_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
 
             generate_file(
@@ -1091,7 +1090,7 @@ class Matmul_ARMCortexM(ExportNode):
 
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "MatMul" / "aidge_matmul_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "MatMul" / "aidge_matmul_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
                 copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "SupportFunctions" / "aidge_supportfunctions.h"),
                          str(Path(export_folder) / "include"))
@@ -1146,7 +1145,7 @@ class Gather_ARMCortexM(ExportNode):
 
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Transform" / "Gather" / "aidge_gather_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Transform" / "Gather" / "aidge_gather_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
 
             generate_file(
@@ -1203,7 +1202,7 @@ class Transpose_ARMCortexM(ExportNode):
         
         if self.library == "aidge":
             if self.dataformat == "float32":
-                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Transform" / "Transpose" / "aidge_transpose_float32.c"),
+                copyfile(str(ROOT / "_Aidge_Arm" / "kernels" / "Transform" / "Transpose" / "aidge_transpose_chw_float32.c"),
                          str(export_folder / "src" / "kernels"))
 
             generate_file(