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}}},