diff options
Diffstat (limited to 'tests/TfLiteResNetV2-50-Quantized-Armnn')
-rw-r--r-- | tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp b/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp index 48e63211a3..98235e3f8a 100644 --- a/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp +++ b/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp @@ -40,14 +40,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); } |