diff --git a/neureka/hal/neureka_task_defs.h b/neureka/hal/neureka_task_defs.h
index fa08289..dfd74d4 100644
--- a/neureka/hal/neureka_task_defs.h
+++ b/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)