aboutsummaryrefslogtreecommitdiff
path: root/include/weights.h
blob: 71d18dacba67707e10bc9f351a0719e0fc7b53fe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#define KERN_NUM  12
#define KERN_CHAN  6
#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 10

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[DENSE2_NUM];
extern const float output_bias;