Skip to content
Snippets Groups Projects
Commit 23bb11dd authored by Jan Buethe's avatar Jan Buethe
Browse files

added destroy functions for encoder/decoder states

parent 7cac35e6
No related branches found
No related tags found
Loading
......@@ -42,6 +42,16 @@ RDOVAEDec * DRED_rdovae_create_decoder()
return dec;
}
void DRED_rdovae_destroy_decoder(RDOVAEDec* dec)
{
free(dec);
}
void DRED_rdovae_destroy_encoder(RDOVAEEnc* enc)
{
free(enc);
}
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);
......
......@@ -10,8 +10,10 @@ size_t DRED_rdovae_get_enc_size(void);
size_t DRED_rdovae_get_dec_size(void);
RDOVAEDec * DRED_rdovae_create_decoder(void);
RDOVAEEnc * DRED_rdovae_create_encoder(void);
void DRED_rdovae_destroy_decoder(RDOVAEDec* h);
void DRED_rdovae_destroy_encoder(RDOVAEEnc* h);
void DRED_rdovae_init_encoder(RDOVAEEnc *enc_state);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment