p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix converter input shape

This commit is contained in:
yeyunpeng2020 2021-11-24 09:04:19 +08:00
parent 77bf581af5
commit dc44f76ac2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/tools/converter/converter_flags.cc
View File

@ -183,14 +183,14 @@ int Flags::InitInTensorShape() {
auto string_split = lite::StrSplit(shape_str, std::string(":"));
CHECK_LESS_RETURN(string_split.size(), kMinShapeSizeInStr);
auto name = string_split[0];
for (size_t i = 0; i < string_split.size() - 1; ++i) {
for (size_t i = 1; i < string_split.size() - 1; ++i) {
name += ":" + string_split[i];
}
if (name.empty()) {
MS_LOG(ERROR) << "input tensor name is empty";
return lite::RET_ERROR;
}
auto dim_strs = string_split[1];
auto dim_strs = string_split[string_split.size() - 1];
if (dim_strs.empty()) {
MS_LOG(ERROR) << "input tensor dim string is empty";
return lite::RET_ERROR;