Make N-EUREKA size parametric (spatial dims)

Through the NNX_NEUREKA_PE_{H,W} defines.

neureka/hal/neureka_task_defs.h

@@ -21,18 +21,21 @@
 #ifndef __NEUREKA_DEFS_H__
 #define __NEUREKA_DEFS_H__
 
-/* ARHITECTURE */
+/* ARCHITECTURE */
 
-#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (6)
-#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1 (6)
+#define NNX_NEUREKA_PE_H (6)
+#define NNX_NEUREKA_PE_W (6)
+
+#define NEUREKA_SUBTILE_INPUT_HEIGHT_1x1 (NNX_NEUREKA_PE_H)
+#define NEUREKA_SUBTILE_INPUT_WIDTH_1x1  (NNX_NEUREKA_PE_W)
 #define NEUREKA_SUBTILE_INPUT_CHANNEL_1x1 (32)
 
-#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (8)
-#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3 (8)
-#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (28)
+#define NEUREKA_SUBTILE_INPUT_HEIGHT_3x3 (NNX_NEUREKA_PE_H+2)
+#define NEUREKA_SUBTILE_INPUT_WIDTH_3x3  (NNX_NEUREKA_PE_W+2)
+#define NEUREKA_SUBTILE_INPUT_CHANNEL_3x3 (32)
 
-#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (6)
-#define NEUREKA_SUBTILE_OUTPUT_WIDTH (6)
+#define NEUREKA_SUBTILE_OUTPUT_HEIGHT (4)
+#define NEUREKA_SUBTILE_OUTPUT_WIDTH (4)
 #define NEUREKA_SUBTILE_OUTPUT_CHANNEL (32)
 
 #define NEUREKA_OUTPUT_BANDWIDTH_BYTES (32)