41 lines
795 B
C
41 lines
795 B
C
/* 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 */
|