aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp')
-rw-r--r--tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp4
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp b/tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp
index 220964d061..1b411f9b94 100644
--- a/tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp
+++ b/tests/TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp
@@ -108,14 +108,12 @@ int main(int argc, char* argv[])
[&imageSet](const char* dataDir, const ModelType & model) {
// we need to get the input quantization parameters from
// the parsed model
- auto inputBinding = model.GetInputBindingInfo();
return DatabaseType(
dataDir,
224,
224,
imageSet,
- inputBinding.second.GetQuantizationScale(),
- inputBinding.second.GetQuantizationOffset());
+ 1);
},
&inputTensorShape);
}