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) {