summaryrefslogtreecommitdiff
path: root/include/ct1973.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/ct1973.h')
-rw-r--r--include/ct1973.h22
1 files changed, 20 insertions, 2 deletions
diff --git a/include/ct1973.h b/include/ct1973.h
index 632f3ba..e1bb717 100644
--- a/include/ct1973.h
+++ b/include/ct1973.h
@@ -1,7 +1,25 @@
#include <stdint.h>
-
+#include <math.h>
#include "tak.h"
-#include "weights.h"
+
+#define KERN_NUM 12
+#define KERN_CHAN 8
+#define KERN_SIZE 3
+#define KERN_OSIZE (5-KERN_SIZE+1)
+#define CONV_NUM (KERN_NUM * KERN_OSIZE * KERN_OSIZE)
+
+#define DENSE1_NUM 9
+#define DENSE2_NUM 8
+#define OUTPUT_NUM 2
+
+extern const float conv2d_weights[KERN_SIZE][KERN_SIZE][KERN_CHAN][KERN_NUM];
+extern const float conv2d_biases[KERN_NUM];
+extern const float dense1_weights[CONV_NUM+2][DENSE1_NUM];
+extern const float dense1_biases[DENSE1_NUM];
+extern const float dense2_weights[DENSE1_NUM][DENSE2_NUM];
+extern const float dense2_biases[DENSE2_NUM];
+extern const float output_weights[DENSE2_NUM][OUTPUT_NUM];
+extern const float output_biases[2];
float
evaluate_black_win(void);