first commit

third_party/opus/dnn/lossgen_data.h

@@ -0,0 +1,39 @@
+/* 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 */