This finally works!

1 files changed, 18 insertions, 0 deletions

diff --git a/include/weights.h b/include/weights.h
new file mode 100644
index 0000000..3c83010
--- /dev/null
+++ b/include/weights.h
@@ -0,0 +1,18 @@
+#define KERN_NUM  12
+#define KERN_CHAN  8
+#define KERN_SIZE  3
+#define KERN_OSIZE (5-KERN_SIZE+1) // 5
+#define CONV_NUM   (KERN_NUM * KERN_OSIZE * KERN_OSIZE) // 108
+
+#define DENSE1_NUM 9
+#define DENSE2_NUM 8
+#define OUTPUT_NUM 2
+
+extern const float conv2d_weights[KERN_NUM][KERN_SIZE][KERN_SIZE][KERN_CHAN];
+extern const float conv2d_biases[KERN_NUM];
+extern const float dense1_weights[DENSE1_NUM][CONV_NUM+2];
+extern const float dense1_biases[DENSE1_NUM];
+extern const float dense2_weights[DENSE2_NUM][DENSE1_NUM];
+extern const float dense2_biases[DENSE2_NUM];
+extern const float output_weights[OUTPUT_NUM][DENSE2_NUM];
+extern const float output_biases[2];