diff options
Diffstat (limited to 'include/ct1973.h')
| -rw-r--r-- | include/ct1973.h | 22 |
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); |
