Browse Source

fix bug of reshape infershape

tags/v0.7.0-beta
fuzhiye 5 years ago
parent
commit
f0172aee4c
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