From 96c73ff091a303940b2a5d2c48e365fa7dc1c073 Mon Sep 17 00:00:00 2001
From: lianghao <lianghao24@hisilicon.com>
Date: Sat, 3 Jul 2021 18:16:07 +0800
Subject: [PATCH] Size parser

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

diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
index dede14d..bdd525d 100644
--- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
+++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
@@ -84,6 +84,18 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
            op_dest->GetType().c_str(), dynamic_tensor_num);
   }
 
+  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))) {
+        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());
+        return FAILED;
+      }
+    }
+  }
+
   // add nodedef for shape insert by adapter when online_infer_dynamic
   if (op_dest->GetType() == SHAPE) {
     ge::DataType out_type = DT_INT32;