shithub: opus

Info  •  Files  •  Log  •  Branches

added destroy functions for encoder/decoder states

--- a/dnn/dred_rdovae.c

+++ b/dnn/dred_rdovae.c

@@ -42,6 +42,16 @@

     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);

--- a/dnn/include/dred_rdovae.h

+++ b/dnn/include/dred_rdovae.h

@@ -10,8 +10,10 @@

 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);

-- 

⑨