shithub: opus

ref: d0b1cd886a43e588c8ed55f578c8a0c1f3f3f9ba
dir: /dnn/dred_rdovae.c/

#include "dred_rdovae.h"

#include "dred_rdovae_enc.h"
#include "dred_rdovae_dec.h"

size_t DRED_rdovae_get_enc_size()
{
    return sizeof(RDOVAEEnc);
}

size_t DRED_rdovae_get_dec_size()
{
    return sizeof(RDOVAEDec);
}

void DRED_rdovae_init_encoder(RDOVAEEnc *enc_state)
{
    memset(enc_state, 0, sizeof(*enc_state));

}

void DRED_rdovae_init_decoder(RDOVAEDec *dec_state)
{
    memset(dec_state, 0, sizeof(*dec_state));
}


RDOVAEEnc * DRED_rdovae_create_encoder()
{
    RDOVAEEnc *enc;
    enc = (RDOVAEEnc*) calloc(sizeof(*enc), 1);
    DRED_rdovae_init_encoder(enc);
    return enc;
}

RDOVAEDec * DRED_rdovae_create_decoder()
{
    RDOVAEDec *dec;
    dec = (RDOVAEDec*) calloc(sizeof(*dec), 1);
    DRED_rdovae_init_decoder(dec);
    return dec;
}

void DRED_rdovae_encode_dframe(RDOVAEEnc *enc_state, float *latents, float *initial_state, const float *input)
{
    dred_rdovae_encode_dframe(enc_state, latents, initial_state, input);
}

void DRED_rdovae_dec_init_states(RDOVAEDec *h, const float * initial_state)
{
    dred_rdovae_dec_init_states(h, initial_state);
}

void DRED_rdovae_decode_qframe(RDOVAEDec *h, float *qframe, const float *z)
{
    dred_rdovae_decode_qframe(h, qframe, z);
}