From cee1749b1d749bdcc9c9bf99616d2fa959f946f9 Mon Sep 17 00:00:00 2001
From: Luke Bollam <lukeb@alertedh.com>
Date: Tue, 14 Sep 2021 14:40:29 +0800
Subject: [PATCH] actually calling the model

---
 src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
index 975968cb..3ed37316 100644
--- a/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
+++ b/src/TensorFlowNet.Benchmarks/Leak/SavedModelCleanup.cs
@@ -27,8 +27,8 @@ namespace Tensorflow.Benchmark.Leak
 						var inputOp = g.OperationByName("inference_input");
 						var outputOp = g.OperationByName("StatefulPartitionedCall");
 
-						var inp = np.zeros(new Shape(new int[] { 1, 96, 2 }));
-						var ops = g.OperationByName("StatefulPartitionedCall");
+						var inp = np.zeros(new Shape(new int[] { 1, 2, 96 }), TF_DataType.TF_FLOAT);
+						sess.run(outputOp.outputs[0], new FeedItem(inputOp.outputs[0], inp));
 					}
 				}
 			}