1 #ifndef RNN_DATA_H
2 #define RNN_DATA_H
3 
4 #include "rnn.h"
5 
6 struct RNNModel {
7   int input_dense_size;
8   const DenseLayer *input_dense;
9 
10   int vad_gru_size;
11   const GRULayer *vad_gru;
12 
13   int noise_gru_size;
14   const GRULayer *noise_gru;
15 
16   int denoise_gru_size;
17   const GRULayer *denoise_gru;
18 
19   int denoise_output_size;
20   const DenseLayer *denoise_output;
21 
22   int vad_output_size;
23   const DenseLayer *vad_output;
24 };
25 
26 struct RNNState {
27   const RNNModel *model;
28   float *vad_gru_state;
29   float *noise_gru_state;
30   float *denoise_gru_state;
31 };
32 
33 
34 #endif
35