Browse Source

!7519 [Auto parallel] Change int32 to int64 in when inferring shape in DropoutGenMask

Merge pull request !7519 from Xiaoda/28-change-int32-to-int64-in-dropoutdomask
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
eb66f95dde
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(); 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) { for (std::size_t i = 0; i < x_shape->size(); ++i) {
auto value_track = x_shape_data[i]->GetValueTrack(); auto value_track = x_shape_data[i]->GetValueTrack();
MS_EXCEPTION_IF_NULL(value_track); MS_EXCEPTION_IF_NULL(value_track);


Loading…
Cancel
Save