109 lines
2 KiB
C
109 lines
2 KiB
C
/* Auto generated from checkpoint rdovae_sparse5m_32.pth */
|
|
|
|
|
|
#ifndef DRED_RDOVAE_ENC_DATA_H
|
|
#define DRED_RDOVAE_ENC_DATA_H
|
|
|
|
#include "nnet.h"
|
|
|
|
|
|
#include "opus_types.h"
|
|
|
|
#include "dred_rdovae.h"
|
|
|
|
#include "dred_rdovae_constants.h"
|
|
|
|
|
|
#define ENC_DENSE1_OUT_SIZE 64
|
|
|
|
#define ENC_ZDENSE_OUT_SIZE 24
|
|
|
|
#define GDENSE1_OUT_SIZE 128
|
|
|
|
#define GDENSE2_OUT_SIZE 24
|
|
|
|
#define ENC_GRU1_OUT_SIZE 64
|
|
|
|
#define ENC_GRU1_STATE_SIZE 64
|
|
|
|
#define ENC_GRU2_OUT_SIZE 64
|
|
|
|
#define ENC_GRU2_STATE_SIZE 64
|
|
|
|
#define ENC_GRU3_OUT_SIZE 64
|
|
|
|
#define ENC_GRU3_STATE_SIZE 64
|
|
|
|
#define ENC_GRU4_OUT_SIZE 64
|
|
|
|
#define ENC_GRU4_STATE_SIZE 64
|
|
|
|
#define ENC_GRU5_OUT_SIZE 64
|
|
|
|
#define ENC_GRU5_STATE_SIZE 64
|
|
|
|
#define ENC_CONV1_OUT_SIZE 96
|
|
|
|
#define ENC_CONV1_IN_SIZE 128
|
|
|
|
#define ENC_CONV1_STATE_SIZE (128 * (1))
|
|
|
|
#define ENC_CONV1_DELAY 0
|
|
|
|
#define ENC_CONV2_OUT_SIZE 96
|
|
|
|
#define ENC_CONV2_IN_SIZE 288
|
|
|
|
#define ENC_CONV2_STATE_SIZE (288 * (1))
|
|
|
|
#define ENC_CONV2_DELAY 0
|
|
|
|
#define ENC_CONV3_OUT_SIZE 96
|
|
|
|
#define ENC_CONV3_IN_SIZE 448
|
|
|
|
#define ENC_CONV3_STATE_SIZE (448 * (1))
|
|
|
|
#define ENC_CONV3_DELAY 0
|
|
|
|
#define ENC_CONV4_OUT_SIZE 96
|
|
|
|
#define ENC_CONV4_IN_SIZE 608
|
|
|
|
#define ENC_CONV4_STATE_SIZE (608 * (1))
|
|
|
|
#define ENC_CONV4_DELAY 0
|
|
|
|
#define ENC_CONV5_OUT_SIZE 96
|
|
|
|
#define ENC_CONV5_IN_SIZE 768
|
|
|
|
#define ENC_CONV5_STATE_SIZE (768 * (1))
|
|
|
|
#define ENC_CONV5_DELAY 0
|
|
|
|
struct RDOVAEEnc {
|
|
LinearLayer enc_dense1;
|
|
LinearLayer enc_zdense;
|
|
LinearLayer gdense1;
|
|
LinearLayer gdense2;
|
|
LinearLayer enc_gru1_input;
|
|
LinearLayer enc_gru1_recurrent;
|
|
LinearLayer enc_gru2_input;
|
|
LinearLayer enc_gru2_recurrent;
|
|
LinearLayer enc_gru3_input;
|
|
LinearLayer enc_gru3_recurrent;
|
|
LinearLayer enc_gru4_input;
|
|
LinearLayer enc_gru4_recurrent;
|
|
LinearLayer enc_gru5_input;
|
|
LinearLayer enc_gru5_recurrent;
|
|
LinearLayer enc_conv1;
|
|
LinearLayer enc_conv2;
|
|
LinearLayer enc_conv3;
|
|
LinearLayer enc_conv4;
|
|
LinearLayer enc_conv5;
|
|
};
|
|
|
|
int init_rdovaeenc(RDOVAEEnc *model, const WeightArray *arrays);
|
|
|
|
#endif /* DRED_RDOVAE_ENC_DATA_H */
|