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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
/* Auto generated from checkpoint lossgen2_2000.pth */
#ifndef LOSSGEN_DATA_H
#define LOSSGEN_DATA_H
#include "nnet.h"
#include "opus_types.h"
#define LOSSGEN_DENSE_IN_OUT_SIZE 8
#define LOSSGEN_DENSE_OUT_OUT_SIZE 1
#define LOSSGEN_GRU1_OUT_SIZE 16
#define LOSSGEN_GRU1_STATE_SIZE 16
#define LOSSGEN_GRU2_OUT_SIZE 32
#define LOSSGEN_GRU2_STATE_SIZE 32
#define LOSSGEN_MAX_RNN_UNITS 32
typedef struct {
LinearLayer lossgen_dense_in;
LinearLayer lossgen_dense_out;
LinearLayer lossgen_gru1_input;
LinearLayer lossgen_gru1_recurrent;
LinearLayer lossgen_gru2_input;
LinearLayer lossgen_gru2_recurrent;
} LossGen;
int init_lossgen(LossGen *model, const WeightArray *arrays);
#endif /* LOSSGEN_DATA_H */
|
