mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!45820 [MS][LITE] fix cxx converter api return code 

Merge pull request !45820 from luoyuan/fix-cxx-converter-returncode
This commit is contained in:
i-robot 2022-11-23 08:14:18 +00:00 committed by Gitee
parent 4ea8e84b1e f79e0e933e
commit f2e2e26efa
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -799,12 +799,12 @@ int CheckInputShape(const std::shared_ptr<ConverterPara> &param) {
std::vector<int64_t> dims = elem.second;
if (dims.empty()) {
MS_LOG(ERROR) << "INPUT MISSING: input tensor dim is empty";
return lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
bool has_negative_dim = std::any_of(dims.begin(), dims.end(), [](int64_t dim) { return dim < 0; });
if (has_negative_dim) {
MS_LOG(ERROR) << "INPUT ILLEGAL: Unsupported dim < 0.";
return lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
}
}

10
mindspore/lite/tools/converter/converter_lite/converter_flags.cc
View File

@ -158,7 +158,7 @@ int Flags::InitInTensorShape() const {
constexpr int kMinShapeSizeInStr = 2;
if (string_split.size() < kMinShapeSizeInStr) {
MS_LOG(ERROR) << "shape size must not be less than " << kMinShapeSizeInStr;
return mindspore::lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
auto name = string_split[0];
for (size_t i = 1; i < string_split.size() - 1; ++i) {
@ -166,17 +166,17 @@ int Flags::InitInTensorShape() const {
}
if (name.empty()) {
MS_LOG(ERROR) << "input tensor name is empty";
return lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
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;
return lite::RET_INPUT_PARAM_INVALID;
}
auto dims = lite::StrSplit(dim_strs, std::string(","));
if (dims.empty()) {
MS_LOG(ERROR) << "input tensor dim is empty";
return lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
for (const auto &dim : dims) {
int64_t dim_value;
@ -184,7 +184,7 @@ int Flags::InitInTensorShape() const {
dim_value = std::stoi(dim);
} catch (const std::exception &e) {
MS_LOG(ERROR) << "Get dim failed: " << e.what();
return lite::RET_ERROR;
return lite::RET_INPUT_PARAM_INVALID;
}
shape.push_back(dim_value);
}