From 3fe3790aa788466d8feb86ad67bbad6eb53a63de Mon Sep 17 00:00:00 2001
From: Gregor Richards <hg-yff@gregor.im>
Date: Wed, 24 Apr 2019 20:04:33 -0400
Subject: [PATCH] Fix to RNN file reader to set _size correctly.

---
 src/rnn_reader.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/rnn_reader.c b/src/rnn_reader.c
index f71ead04..2a031db1 100644
--- a/src/rnn_reader.c
+++ b/src/rnn_reader.c
@@ -111,6 +111,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
 #define INPUT_DENSE(name) do { \
     INPUT_VAL(name->nb_inputs); \
     INPUT_VAL(name->nb_neurons); \
+    ret->name ## _size = name->nb_neurons; \
     INPUT_ACTIVATION(name->activation); \
     INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons); \
     INPUT_ARRAY(name->bias, name->nb_neurons); \
@@ -119,6 +120,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
 #define INPUT_GRU(name) do { \
     INPUT_VAL(name->nb_inputs); \
     INPUT_VAL(name->nb_neurons); \
+    ret->name ## _size = name->nb_neurons; \
     INPUT_ACTIVATION(name->activation); \
     INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons * 3); \
     INPUT_ARRAY(name->recurrent_weights, name->nb_neurons * name->nb_neurons * 3); \