shithub: opus

ref: 8e8edf71bde743c21960a97815f14dd48c86d6ad
dir: /dnn/pitchdnn.h/

View raw version 
#ifndef PITCHDNN_H
#define PITCHDNN_H


typedef struct PitchDNN PitchDNN;

#include "pitchdnn_data.h"
#include "lpcnet_private.h"

#define NB_XCORR_FEATURES (PITCH_MAX_PERIOD-PITCH_MIN_PERIOD)


typedef struct {
  PitchDNN model;
  float gru_state[GRU_1_STATE_SIZE];
  float xcorr_mem1[(NB_XCORR_FEATURES + 2)*2];
  float xcorr_mem2[(NB_XCORR_FEATURES + 2)*2*8];
  float xcorr_mem3[(NB_XCORR_FEATURES + 2)*2*8];
} PitchDNNState;


void pitchdnn_init(PitchDNNState *st);

int compute_pitchdnn(
    PitchDNNState *st,
    const float *if_features,
    const float *xcorr_features
    );

#endif