diff --git a/mindspore/lite/src/ops/batch_norm.cc b/mindspore/lite/src/ops/batch_norm.cc
index 736e6a9441..5ae19f1a76 100644
--- a/mindspore/lite/src/ops/batch_norm.cc
+++ b/mindspore/lite/src/ops/batch_norm.cc
@@ -38,12 +38,13 @@ int BatchNorm::UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &
   }
   if (this->primitive_->value.value == nullptr) {
     auto attr = new (std::nothrow) schema::FusedBatchNormT();
-    attr->epsilon = GetValue<float>(prim.GetAttr("epsilon"));
-    this->primitive_->value.value = attr;
-    if (this->primitive_->value.value == nullptr) {
-      MS_LOG(ERROR) << "new primitiveT value failed";
+    if (attr == nullptr) {
+      MS_LOG(ERROR) << "new FusedBatchNormT failed";
+      delete this->primitive_;
       return RET_ERROR;
     }
+    attr->epsilon = GetValue<float>(prim.GetAttr("epsilon"));
+    this->primitive_->value.value = attr;
   }
   return RET_OK;
 }
diff --git a/mindspore/lite/src/populate_parameter.cc b/mindspore/lite/src/populate_parameter.cc
index b7ff742dc8..c52b403dfa 100644
--- a/mindspore/lite/src/populate_parameter.cc
+++ b/mindspore/lite/src/populate_parameter.cc
@@ -1536,6 +1536,11 @@ OpParameter *PopulateL2NormParameter(const mindspore::lite::PrimitiveC *primitiv
   auto axis_vec = param->GetAxis();
   l2_norm_parameter->axis_num_ = axis_vec.size();
   l2_norm_parameter->axis_ = reinterpret_cast<int *>(malloc(axis_vec.size() * sizeof(int)));
+  if (l2_norm_parameter->axis_ == nullptr) {
+    MS_LOG(ERROR) << "malloc axis_ data failed";
+    free(l2_norm_parameter);
+    return nullptr;
+  }
   for (size_t i = 0; i < axis_vec.size(); i++) {
     l2_norm_parameter->axis_[i] = axis_vec[i];
   }