From c341b37318161d45aa9a1e09d8cff059de4b1566 Mon Sep 17 00:00:00 2001
From: Oceania2018 <haiping008@gmail.com>
Date: Sat, 2 Oct 2021 08:34:13 -0500
Subject: [PATCH] Don't dispose graph in LoadFromSavedModel.

---
 src/TensorFlowNET.Core/Sessions/Session.cs             | 2 +-
 src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/TensorFlowNET.Core/Sessions/Session.cs b/src/TensorFlowNET.Core/Sessions/Session.cs
index 1e94b882..5b842408 100644
--- a/src/TensorFlowNET.Core/Sessions/Session.cs
+++ b/src/TensorFlowNET.Core/Sessions/Session.cs
@@ -39,7 +39,7 @@ namespace Tensorflow
 
         public static Session LoadFromSavedModel(string path)
         {
-            using var graph = new Graph();
+            var graph = new Graph();
             using var status = new Status();
             using var opt = c_api.TF_NewSessionOptions();
 
diff --git a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
index 5cdb28f7..c060c5ed 100644
--- a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
+++ b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
@@ -10,7 +10,9 @@ using static Tensorflow.Binding;
 
 namespace Tensorflow.Benchmark.Leak
 {
-	
+	/// <summary>
+	/// https://github.com/SciSharp/TensorFlow.NET/issues/418
+	/// </summary>
 	public class SavedModelCleanup
 	{
 		[Benchmark]
@@ -22,6 +24,8 @@ namespace Tensorflow.Benchmark.Leak
 			for (var i = 0; i < 1024; i++)
             {
 				using var sess = Session.LoadFromSavedModel(ClassifierModelPath);
+				// destory graph
+				using var g = sess.graph;
 			}
 		}
 	}