blob: e1bb7174cc6b51e966d941561dc066759486b673 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#include <stdint.h>
#include <math.h>
#include "tak.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);
|
