Skip to content
This repository was archived by the owner on Sep 27, 2019. It is now read-only.

Commit d5e8b40

Browse files
committed
Memory leak fix
1 parent 19dcdfe commit d5e8b40

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/brain/tf_session_entity/tf_session_entity.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,11 @@ TFSE_TYPE::TfSessionEntity() {
3131

3232
TFSE_TEMPLATE_ARGUMENTS
3333
TFSE_TYPE::~TfSessionEntity() {
34-
TF_DeleteStatus(status_);
34+
TF_CloseSession(session_, status_);
35+
TF_DeleteSession(session_, status_);
3536
TF_DeleteGraph(graph_);
37+
TF_DeleteStatus(status_);
38+
TF_DeleteSessionOptions(session_options_);
3639
}
3740

3841
/*

0 commit comments

Comments
 (0)