From 84c1705319aa10cbe294b5fa6546a1c8bf6f7663 Mon Sep 17 00:00:00 2001
From: lixian <lixian16@huawei.com>
Date: Tue, 23 Mar 2021 16:12:19 +0800
Subject: [PATCH] add GetOutputsByNodeName

---
 include/api/model.h                              | 5 +++++
 mindspore/lite/src/cxx_api/tensor/tensor_impl.cc | 4 ++--
 mindspore/lite/src/cxx_api/tensor/tensor_impl.h  | 2 ++
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/include/api/model.h b/include/api/model.h
index 9837cc3e1b..0cb4992699 100644
--- a/include/api/model.h
+++ b/include/api/model.h
@@ -49,6 +49,7 @@ class MS_API Model {
   std::vector<MSTensor> GetOutputs();
   inline std::vector<std::string> GetOutputTensorNames();
   inline MSTensor GetOutputByTensorName(const std::string &tensor_name);
+  inline std::vector<MSTensor> GetOutputsByNodeName(const std::string &tensor_name);
 
   static bool CheckModelSupport(enum DeviceType device_type, ModelType model_type);
 
@@ -71,5 +72,9 @@ std::vector<std::string> Model::GetOutputTensorNames() { return VectorCharToStri
 MSTensor Model::GetOutputByTensorName(const std::string &tensor_name) {
   return GetOutputByTensorName(StringToChar(tensor_name));
 }
+
+std::vector<MSTensor> Model::GetOutputsByNodeName(const std::string &tensor_name) {
+  return GetOutputsByNodeName(StringToChar(tensor_name));
+}
 }  // namespace mindspore
 #endif  // MINDSPORE_INCLUDE_API_MODEL_H
diff --git a/mindspore/lite/src/cxx_api/tensor/tensor_impl.cc b/mindspore/lite/src/cxx_api/tensor/tensor_impl.cc
index 9667251bb3..a5c40304a6 100644
--- a/mindspore/lite/src/cxx_api/tensor/tensor_impl.cc
+++ b/mindspore/lite/src/cxx_api/tensor/tensor_impl.cc
@@ -45,12 +45,12 @@ MSTensor::Impl *MSTensor::Impl::CreateTensorImpl(const std::string &name, enum D
     MS_LOG(ERROR) << "Failed to allocate lite tensor.";
     return nullptr;
   }
-  auto impl = new (std::nothrow) Impl();
+  auto impl = new (std::nothrow) Impl(lite_tensor);
   if (impl == nullptr) {
     MS_LOG(ERROR) << "Failed to allocate tensor impl.";
     return nullptr;
   }
-  impl->set_lite_tensor(lite_tensor);
+  impl->set_from_session(false);
   return impl;
 }
 
diff --git a/mindspore/lite/src/cxx_api/tensor/tensor_impl.h b/mindspore/lite/src/cxx_api/tensor/tensor_impl.h
index b8e9d9833b..e500397861 100644
--- a/mindspore/lite/src/cxx_api/tensor/tensor_impl.h
+++ b/mindspore/lite/src/cxx_api/tensor/tensor_impl.h
@@ -149,6 +149,8 @@ class MSTensor::Impl {
 
   void set_own_data(bool own_data) { own_data_ = own_data; }
 
+  void set_from_session(bool from_session) { from_session_ = from_session; }
+
  private:
   tensor::MSTensor *lite_tensor_ = nullptr;
   std::string tensor_name_ = "";