From a511826d4d59f7a6c18328b2587d640ebccd6ddc Mon Sep 17 00:00:00 2001
From: Xiaoda Zhang <zhangxiaoda@huawei.com>
Date: Tue, 20 Oct 2020 18:36:08 +0800
Subject: [PATCH] change int32 to int64

---
 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 039f7e35ca..b41d40594c 100644
--- a/mindspore/core/abstract/prim_nn.cc
+++ b/mindspore/core/abstract/prim_nn.cc
@@ -416,7 +416,7 @@ AbstractBasePtr InferImplDropoutGenMask(const AnalysisEnginePtr &, const Primiti
   }
 
   auto x_shape_data = x_shape->elements();
-  int count = 1;
+  int64_t count = 1;
   for (std::size_t i = 0; i < x_shape->size(); ++i) {
     auto value_track = x_shape_data[i]->GetValueTrack();
     MS_EXCEPTION_IF_NULL(value_track);