mirror of https://git.ffmpeg.org/ffmpeg.git
lavfi/dnn_backend_tensorflow.c: fix mem leak in load_tf_model
Signed-off-by: Ting Fu <ting.fu@intel.com>
This commit is contained in:
parent
a4e518c321
commit
3499ec84cc
|
@ -282,6 +282,9 @@ static DNNReturnType load_tf_model(TFModel *tf_model, const char *model_filename
|
||||||
TF_SetConfig(sess_opts, sess_config, sess_config_length,tf_model->status);
|
TF_SetConfig(sess_opts, sess_config, sess_config_length,tf_model->status);
|
||||||
av_freep(&sess_config);
|
av_freep(&sess_config);
|
||||||
if (TF_GetCode(tf_model->status) != TF_OK) {
|
if (TF_GetCode(tf_model->status) != TF_OK) {
|
||||||
|
TF_DeleteGraph(tf_model->graph);
|
||||||
|
TF_DeleteStatus(tf_model->status);
|
||||||
|
TF_DeleteSessionOptions(sess_opts);
|
||||||
av_log(ctx, AV_LOG_ERROR, "Failed to set config for sess options with %s\n",
|
av_log(ctx, AV_LOG_ERROR, "Failed to set config for sess options with %s\n",
|
||||||
tf_model->ctx.options.sess_config);
|
tf_model->ctx.options.sess_config);
|
||||||
return DNN_ERROR;
|
return DNN_ERROR;
|
||||||
|
@ -292,6 +295,8 @@ static DNNReturnType load_tf_model(TFModel *tf_model, const char *model_filename
|
||||||
TF_DeleteSessionOptions(sess_opts);
|
TF_DeleteSessionOptions(sess_opts);
|
||||||
if (TF_GetCode(tf_model->status) != TF_OK)
|
if (TF_GetCode(tf_model->status) != TF_OK)
|
||||||
{
|
{
|
||||||
|
TF_DeleteGraph(tf_model->graph);
|
||||||
|
TF_DeleteStatus(tf_model->status);
|
||||||
av_log(ctx, AV_LOG_ERROR, "Failed to create new session with model graph\n");
|
av_log(ctx, AV_LOG_ERROR, "Failed to create new session with model graph\n");
|
||||||
return DNN_ERROR;
|
return DNN_ERROR;
|
||||||
}
|
}
|
||||||
|
@ -304,6 +309,9 @@ static DNNReturnType load_tf_model(TFModel *tf_model, const char *model_filename
|
||||||
&init_op, 1, NULL, tf_model->status);
|
&init_op, 1, NULL, tf_model->status);
|
||||||
if (TF_GetCode(tf_model->status) != TF_OK)
|
if (TF_GetCode(tf_model->status) != TF_OK)
|
||||||
{
|
{
|
||||||
|
TF_DeleteSession(tf_model->session, tf_model->status);
|
||||||
|
TF_DeleteGraph(tf_model->graph);
|
||||||
|
TF_DeleteStatus(tf_model->status);
|
||||||
av_log(ctx, AV_LOG_ERROR, "Failed to run session when initializing\n");
|
av_log(ctx, AV_LOG_ERROR, "Failed to run session when initializing\n");
|
||||||
return DNN_ERROR;
|
return DNN_ERROR;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue