From b1543278fcefdf9fced64a2b4d5398c8cf9bbffa Mon Sep 17 00:00:00 2001
From: wangxiaotian22 <wangxiaotian4@huawei.com>
Date: Tue, 20 Jul 2021 20:03:13 +0800
Subject: [PATCH] size parse modify

---
 parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
index bdd525d..1ab6e2e 100644
--- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
+++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
@@ -87,7 +87,7 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
   if (op_dest->GetType() == SIZE) {
     ge::DataType out_type = DT_INT32;
     if (AttrUtils::GetDataType(op_dest, kShapeAttrOutType, out_type)) {
-      if (!AttrUtils::SetInt(op_dest, kShapeAttrDtype, static_cast<int64_t>(out_type))) {
+      if (!AttrUtils::SetDataType(op_dest, kShapeAttrDtype, out_type)) {
         REPORT_CALL_ERROR("E19999", "Set Attr:%s to op:%s(%s) failed", kShapeAttrDtype,
                           op_dest->GetName().c_str(), op_dest->GetType().c_str());
         GELOGE(FAILED, "Set attr dtype for op:%s failed.", op_dest->GetName().c_str());