summaryrefslogtreecommitdiff
path: root/include/weights.h
blob: 3c830108589b0b8f8d81083710041aae529291c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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];