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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/* Auto generated from checkpoint rdovae_sparse5m_32.pth */
#ifndef DRED_RDOVAE_DEC_DATA_H
#define DRED_RDOVAE_DEC_DATA_H
#include "nnet.h"
#include "opus_types.h"
#include "dred_rdovae.h"
#include "dred_rdovae_constants.h"
#define DEC_DENSE1_OUT_SIZE 96
#define DEC_GLU1_OUT_SIZE 96
#define DEC_GLU2_OUT_SIZE 96
#define DEC_GLU3_OUT_SIZE 96
#define DEC_GLU4_OUT_SIZE 96
#define DEC_GLU5_OUT_SIZE 96
#define DEC_OUTPUT_OUT_SIZE 80
#define DEC_HIDDEN_INIT_OUT_SIZE 128
#define DEC_GRU_INIT_OUT_SIZE 480
#define DEC_GRU1_OUT_SIZE 96
#define DEC_GRU1_STATE_SIZE 96
#define DEC_GRU2_OUT_SIZE 96
#define DEC_GRU2_STATE_SIZE 96
#define DEC_GRU3_OUT_SIZE 96
#define DEC_GRU3_STATE_SIZE 96
#define DEC_GRU4_OUT_SIZE 96
#define DEC_GRU4_STATE_SIZE 96
#define DEC_GRU5_OUT_SIZE 96
#define DEC_GRU5_STATE_SIZE 96
#define DEC_CONV1_OUT_SIZE 32
#define DEC_CONV1_IN_SIZE 192
#define DEC_CONV1_STATE_SIZE (192 * (1))
#define DEC_CONV1_DELAY 0
#define DEC_CONV2_OUT_SIZE 32
#define DEC_CONV2_IN_SIZE 320
#define DEC_CONV2_STATE_SIZE (320 * (1))
#define DEC_CONV2_DELAY 0
#define DEC_CONV3_OUT_SIZE 32
#define DEC_CONV3_IN_SIZE 448
#define DEC_CONV3_STATE_SIZE (448 * (1))
#define DEC_CONV3_DELAY 0
#define DEC_CONV4_OUT_SIZE 32
#define DEC_CONV4_IN_SIZE 576
#define DEC_CONV4_STATE_SIZE (576 * (1))
#define DEC_CONV4_DELAY 0
#define DEC_CONV5_OUT_SIZE 32
#define DEC_CONV5_IN_SIZE 704
#define DEC_CONV5_STATE_SIZE (704 * (1))
#define DEC_CONV5_DELAY 0
struct RDOVAEDec {
LinearLayer dec_dense1;
LinearLayer dec_glu1;
LinearLayer dec_glu2;
LinearLayer dec_glu3;
LinearLayer dec_glu4;
LinearLayer dec_glu5;
LinearLayer dec_output;
LinearLayer dec_hidden_init;
LinearLayer dec_gru_init;
LinearLayer dec_gru1_input;
LinearLayer dec_gru1_recurrent;
LinearLayer dec_gru2_input;
LinearLayer dec_gru2_recurrent;
LinearLayer dec_gru3_input;
LinearLayer dec_gru3_recurrent;
LinearLayer dec_gru4_input;
LinearLayer dec_gru4_recurrent;
LinearLayer dec_gru5_input;
LinearLayer dec_gru5_recurrent;
LinearLayer dec_conv1;
LinearLayer dec_conv2;
LinearLayer dec_conv3;
LinearLayer dec_conv4;
LinearLayer dec_conv5;
};
int init_rdovaedec(RDOVAEDec *model, const WeightArray *arrays);
#endif /* DRED_RDOVAE_DEC_DATA_H */
|
