From c21caa3039d4f847f7dcd7285ef79081ff019aaa Mon Sep 17 00:00:00 2001
From: zengxianglong <zengxianglong1@huawei.com>
Date: Wed, 4 Nov 2020 19:01:57 +0800
Subject: [PATCH] release the subgraph output tensor when not training

---
 mindspore/lite/src/executor.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/lite/src/executor.cc b/mindspore/lite/src/executor.cc
index b80e92dca7..96d5d4b383 100644
--- a/mindspore/lite/src/executor.cc
+++ b/mindspore/lite/src/executor.cc
@@ -43,10 +43,11 @@ int Executor::Run(std::vector<Tensor *> &in_tensors, std::vector<Tensor *> &out_
     return ret;
   }
   kernel::LiteKernelUtil::InitTensorRefCount(kernels);
+#ifdef SUPPORT_TRAIN
   for (auto out_tensor : out_tensors) {  // increase RefCount of output tensors, such that Run will not free them
     out_tensor->set_ref_count(out_tensor->ref_count() + 1);
   }
-
+#endif
   for (auto *kernel : kernels) {
     MS_ASSERT(nullptr != kernel);
     ret = kernel->PreProcess();