From 7f09ecdf30523d8d15b6e6e8e3255d2b207f6f91 Mon Sep 17 00:00:00 2001
From: dongduo <dongduo5@huawei.com>
Date: Thu, 17 Dec 2020 19:20:17 +0800
Subject: [PATCH] Fix empty bug

---
 .../tensorflow_auto_mapping_parser_adapter.cc         | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
index 4d40e10..ebf8079 100644
--- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
+++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
@@ -56,6 +56,17 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
   }
   op.BreakConnect();
 
+  if (op_dest->GetType() == EMPTY) {
+    domi::tensorflow::AttrValue attr;
+    if (TensorFlowUtil::FindAttrValue(node, kShapeAttrDtype, attr)) {
+      ge::DataType data_type = domi::TensorAssign::ConvertTensorflowDataType(static_cast<uint32_t>(attr.type()));
+      AttrUtils::SetInt(op_dest, kShapeAttrDtype, data_type);
+      GETLOGD("Get dtype:%d success.", data_type);
+    } else {
+      GELOGW("Get dtype failed!");
+    }
+  }
+
   // add dynamic input/output
   if (op_dest->GetType() == IDENTITYN) {
     uint32_t dynamic_tensor_num = 0;