From 59a8b0cdfad0707e44c135f9f3e214321ad219ef Mon Sep 17 00:00:00 2001
From: l00591931 <liangzhibo@huawei.com>
Date: Thu, 23 Sep 2021 21:14:15 +0800
Subject: [PATCH] fix warning

---
 mindspore/core/abstract/prim_nn.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/core/abstract/prim_nn.cc b/mindspore/core/abstract/prim_nn.cc
index 516e14928d3..2e6455bc613 100644
--- a/mindspore/core/abstract/prim_nn.cc
+++ b/mindspore/core/abstract/prim_nn.cc
@@ -152,7 +152,7 @@ AbstractBasePtr InferImplBatchNorm(const AnalysisEnginePtr &, const PrimitivePtr
   MS_EXCEPTION_IF_NULL(data_format_ptr);
   int64_t data_format = GetAndCheckFormat(data_format_ptr);
 
-  int64_t c_axis = 1;
+  size_t c_axis = 1;
   if (data_format == Format::NHWC) {
     c_axis = 3;
   }