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