first commit

third_party/opus/dnn/pitchdnn_data.h

@@ -0,0 +1,41 @@
+/* Auto generated from checkpoint pitch_vsmallconv1.pth */
+
+
+#ifndef PITCHDNN_DATA_H
+#define PITCHDNN_DATA_H
+
+#include "nnet.h"
+
+
+#include "opus_types.h"
+
+#define DENSE_IF_UPSAMPLER_1_OUT_SIZE 64
+
+#define DENSE_IF_UPSAMPLER_2_OUT_SIZE 64
+
+#define DENSE_DOWNSAMPLER_OUT_SIZE 64
+
+#define DENSE_FINAL_UPSAMPLER_OUT_SIZE 192
+
+#define GRU_1_OUT_SIZE 64
+
+#define GRU_1_STATE_SIZE 64
+
+
+#define PITCH_DNN_MAX_RNN_UNITS 64
+
+
+struct PitchDNN {
+    LinearLayer dense_if_upsampler_1;
+    LinearLayer dense_if_upsampler_2;
+    LinearLayer dense_downsampler;
+    LinearLayer dense_final_upsampler;
+    Conv2dLayer conv2d_1;
+    Conv2dLayer conv2d_2;
+    LinearLayer gru_1_input;
+    LinearLayer gru_1_recurrent;
+};
+
+int init_pitchdnn(PitchDNN *model, const WeightArray *arrays);
+
+#endif /* PITCHDNN_DATA_H */