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

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

diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
index 4d40e10..e92b089 100644
--- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
+++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
@@ -55,6 +55,16 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
     return FAILED;
   }
   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);
+      GELOGD("Get dtype:%d success.", data_type);
+    } else {
+      GELOGW("Get dtype failed!");
+    }
+  }
 
   // add dynamic input/output
   if (op_dest->GetType() == IDENTITYN) {