Merge pull request #2126 from lissyx/init-previous_state

Move init of previous_state_{c,h}
This commit is contained in:
lissyx 2019-05-22 20:06:14 +02:00 committed by GitHub
commit 325071a2a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -843,6 +843,12 @@ DS_SetupStream(ModelState* aCtx,
ctx->model = aCtx;
#ifdef USE_TFLITE
/* Ensure previous_state_{c,h} are not holding previous stream value */
memset(ctx->model->previous_state_c_.get(), 0, sizeof(float) * ctx->model->previous_state_size);
memset(ctx->model->previous_state_h_.get(), 0, sizeof(float) * ctx->model->previous_state_size);
#endif // USE_TFLITE
DecoderState *params = decoder_init(*aCtx->alphabet, num_classes, aCtx->scorer);
aCtx->decoder_state = params;