xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix a bug in normalize operator

This commit is contained in:
zengxianglong 2020-12-02 15:24:38 +08:00
parent bc4f452aeb
commit d81cb6b5cf
4 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/lite/src/runtime/kernel/arm/string/normalize.cc
View File

@ -49,12 +49,13 @@ int NormalizeCPUKernel::Init() {
int NormalizeCPUKernel::ReSize() { return RET_OK; }
std::string NormalizeCPUKernel::Trim(const std::string &str, const std::string &whitespace /*= " \t\n\v\f\r"*/) {
if (str.empty()) {
std::string NormalizeCPUKernel::Trim(const std::string &str, const std::string &pattern /*= " \t\n\v\f\r"*/) {
auto begin = str.find_first_not_of(pattern);
if (begin == std::string::npos) {
MS_LOG(WARNING) << "Meaningless input string!";
return "";
}
auto begin = str.find_first_not_of(whitespace);
auto end = str.find_last_not_of(whitespace);
auto end = str.find_last_not_of(pattern);
const auto range = end - begin + 1;
return str.substr(begin, range);
}

2
mindspore/lite/src/runtime/kernel/arm/string/normalize.h
View File

@ -36,7 +36,7 @@ class NormalizeCPUKernel : public LiteKernel {
int Run() override;
private:
std::string Trim(const std::string &str, const std::string &whitespace = " \t\n\v\f\r");
std::string Trim(const std::string &str, const std::string &pattern = " \t\n\v\f\r");
std::string GlobalReplace(const std::string &str, const std::string &reg, const std::string &replace);
std::string Normalize(const std::string &str);
std::vector<char *> normalized_strs;

1
mindspore/lite/test/models_mindspore.cfg
View File

@ -15,3 +15,4 @@ mindspore_ghost-nose-pets-811.mindir 0.5
mindspore_ghost-pets-8244.mindir 1.5
mindspore_ghostnet600M-pets.mindir 1.5
mindspore_ghostnet_1x_pets_int8.mindir 12
mindspore_deeplab_v3_s16.mindir 6.5

1
mindspore/lite/test/models_onnx_fp16.cfg
View File

@ -33,3 +33,4 @@ rain-princess-9.onnx 5
udnie-9.onnx 3
adversarial_pruning.onnx 3
residual_distill_res34_cifar10_bs_1_update.onnx 2
residual_distill_res50_cifar10_bs_1_update.onnx 2