From 71cf368fb319002fb3d74ba37204b44815bdeee2 Mon Sep 17 00:00:00 2001
From: bai-yangfan <baiyangfan@huawei.com>
Date: Thu, 28 Jan 2021 14:46:28 +0800
Subject: [PATCH] fix_l2norm_bug

---
 mindspore/core/utils/check_convert_utils.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mindspore/core/utils/check_convert_utils.cc b/mindspore/core/utils/check_convert_utils.cc
index a014a4c091..459d4033d0 100644
--- a/mindspore/core/utils/check_convert_utils.cc
+++ b/mindspore/core/utils/check_convert_utils.cc
@@ -230,8 +230,11 @@ namespace {
 typedef std::map<std::string, std::function<ValuePtr(ValuePtr)>> AttrFunction;
 
 ValuePtr L2NormalizeAttrConversion(ValuePtr attr) {
+  if (attr->isa<Int64Imm>()) {
+    return attr;
+  }
   auto attr_value = GetValue<std::vector<int64_t>>(attr);
-  return MakeValue<int64_t>(attr_value[0]);
+  return MakeValue(attr_value[0]);
 }
 
 std::map<std::string, AttrFunction> kIrAttrToOpAttr = {{"L2Normalize", {{"axis", L2NormalizeAttrConversion}}},