39 lines
740 B
C
39 lines
740 B
C
/* 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 */
|