From 45745c527e3c18c91e41df2cf6ea33e766cf4550 Mon Sep 17 00:00:00 2001
From: lvliang <lvliang18@huawei.com>
Date: Fri, 31 Jul 2020 09:48:13 +0800
Subject: [PATCH] sync-device-data-type-to-return-tensor

---
 mindspore/ccsrc/backend/session/session_basic.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/backend/session/session_basic.cc b/mindspore/ccsrc/backend/session/session_basic.cc
index dfde9336b2..d61e10e197 100644
--- a/mindspore/ccsrc/backend/session/session_basic.cc
+++ b/mindspore/ccsrc/backend/session/session_basic.cc
@@ -86,7 +86,10 @@ BaseRef CreateOneTensor(const AnfNodePtr &node, size_t output_index, const Kerne
   MS_EXCEPTION_IF_NULL(address);
   auto shape = AnfAlgo::GetOutputInferShape(node, output_index);
   TypeId type_id = kNumberTypeFloat32;
-  type_id = AnfAlgo::GetOutputInferDataType(node, output_index);
+  type_id = AnfAlgo::GetOutputDeviceDataType(node, output_index);
+  if (type_id == kTypeUnknown) {
+    type_id = AnfAlgo::GetOutputInferDataType(node, output_index);
+  }
   std::vector<int> temp_shape;
   if (graph.IsInternalOutput(node, output_index)) {
     temp_shape.emplace_back(1);