From 681bd41dd55f053e9618d7bc7d1321da38119e4d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=B6=9B?= <wangtao123@huawei.com>
Date: Thu, 22 Jul 2021 15:35:55 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E9=80=80=20'Pull=20Request=20!345=20:?=
 =?UTF-8?q?=20size=20of=20parser'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 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 1ab6e2e..bdd525d 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::SetDataType(op_dest, kShapeAttrDtype, out_type)) {
+      if (!AttrUtils::SetInt(op_dest, kShapeAttrDtype, static_cast<int64_t>(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());