Browse Source

!4202 fix bug of reshape op infershape error

Merge pull request !4202 from fuzhiye/tmp
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
0ae5eeb33d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/ops/reshape.cc

+ 1
- 1
mindspore/lite/src/ops/reshape.cc View File

@@ -92,7 +92,7 @@ int Reshape::InferShape(std::vector<tensor::Tensor *> inputs_, std::vector<tenso
MS_LOG(INFO) << "Do infer shape in runtime.";
return RET_INFER_INVALID;
}
size_t shape_size = shape_tensor->shape().size();
size_t shape_size = shape_tensor->ElementsNum();
switch (shape_tensor->data_type()) {
case kNumberTypeInt8: {
auto data = reinterpret_cast<int8_t *>(shape_tensor->Data());


Loading…
Cancel
Save