change int32 to int64

This commit is contained in:
Xiaoda Zhang 2020-10-20 18:36:08 +08:00
parent c8a8c2623d
commit a511826d4d
1 changed files with 1 additions and 1 deletions

View File

@ -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);