experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update quantization_config.cc

Browse Source
This commit is contained in:
tg-at-google 2020-06-16 21:31:09 -04:00 committed by GitHub
parent caf465347e
commit 8d8212ea81
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/mlir/lite/quantization/quantization_config.cc
View File

@ -48,9 +48,9 @@ bool ParseInputNodeQuantSpecs(absl::string_view node_names,
std::vector<double> node_mins;
if (!min_values.empty()) {
std::vector<std::string> node_mins_str = absl::StrSplit(min_values, ',');
for (const std::string&node_min : node_mins_str.size()) {
for (int i = 0, e = node_mins_str.size(); i < e; i++) {
double value;
if (!absl::SimpleAtod(node_min, &value)) {
if (!absl::SimpleAtod(node_mins_str[i], &value)) {
return true;
}
node_mins.push_back(value);
@ -60,9 +60,9 @@ bool ParseInputNodeQuantSpecs(absl::string_view node_names,
std::vector<double> node_maxs;
if (!max_values.empty()) {
std::vector<std::string> node_maxs_str = absl::StrSplit(max_values, ',');
for (const std::string&node_max : node_maxs_str.size()) {
for (int i = 0, e = node_maxs_str.size(); i < e; i++) {
double value;
if (!absl::SimpleAtod(node_max, &value)) {
if (!absl::SimpleAtod(node_maxs_str[i], &value)) {
llvm::errs() << "Unexpected mins: " << node_maxs_str[i] << "\n";
return true;
}