From 23bb11dd484818bfae9ae0daba369ef4e5c31ab2 Mon Sep 17 00:00:00 2001
From: Jan Buethe <jbuethe@amazon.de>
Date: Tue, 25 Oct 2022 12:14:31 +0000
Subject: [PATCH] added destroy functions for encoder/decoder states
---
dnn/dred_rdovae.c | 10 ++++++++++
dnn/include/dred_rdovae.h | 4 +++-
2 files changed, 13 insertions(+), 1 deletion(-)
diff --git a/dnn/dred_rdovae.c b/dnn/dred_rdovae.c
index 3f6e738c3..c7d2d1820 100644
--- a/dnn/dred_rdovae.c
+++ b/dnn/dred_rdovae.c
@@ -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);
diff --git a/dnn/include/dred_rdovae.h b/dnn/include/dred_rdovae.h
index a5b5cc0c7..f9e7403dc 100644
--- a/dnn/include/dred_rdovae.h
+++ b/dnn/include/dred_rdovae.h
@@ -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);
--
GitLab