diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
index 4d40e10..2c4ad95 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);
+      GELOGD("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;