Browse Source

change int32 to int64

tags/v1.1.0
Xiaoda Zhang 5 years ago
parent
commit
a511826d4d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/core/abstract/prim_nn.cc

+ 1
- 1
mindspore/core/abstract/prim_nn.cc 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);


Loading…
Cancel
Save