Browse Source

!10170 reshape impl bug fix

From: @hwjiaorui
Reviewed-by: @zhoufeng54,@liangchenghui
Signed-off-by: @liangchenghui
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
433eea1526
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/core/abstract/prim_arrays.cc

+ 1
- 1
mindspore/core/abstract/prim_arrays.cc View File

@@ -856,7 +856,7 @@ AbstractBasePtr InferImplReshape(const AnalysisEnginePtr &, const PrimitivePtr &
if (it_second != shape.end()) {
MS_LOG(EXCEPTION) << "At most one component of input shape can be -1";
}
int index = std::distance(it_first, shape.begin());
int index = std::distance(shape.begin(), it_first);
int infer_value = x_num;
int infer_min_value = x_min_num;
int infer_max_value = x_max_num;


Loading…
Cancel
Save