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 plc4ar_16.pth */
#ifndef PLC_DATA_H
#define PLC_DATA_H
#include "nnet.h"
#include "opus_types.h"
#define PLC_DENSE_IN_OUT_SIZE 128
#define PLC_DENSE_OUT_OUT_SIZE 20
#define PLC_GRU1_OUT_SIZE 192
#define PLC_GRU1_STATE_SIZE 192
#define PLC_GRU2_OUT_SIZE 192
#define PLC_GRU2_STATE_SIZE 192
#define PLC_MAX_RNN_UNITS 192
typedef struct {
LinearLayer plc_dense_in;
LinearLayer plc_dense_out;
LinearLayer plc_gru1_input;
LinearLayer plc_gru1_recurrent;
LinearLayer plc_gru2_input;
LinearLayer plc_gru2_recurrent;
} PLCModel;
int init_plcmodel(PLCModel *model, const WeightArray *arrays);
#endif /* PLC_DATA_H */
|