diff options
author | David Monahan <david.monahan@arm.com> | 2019-04-23 10:54:34 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-04-23 10:53:23 +0000 |
commit | dc2797928dd0945f351445ba8d3be11b40376dd0 (patch) | |
tree | 4ce6f0597620a785d7abc2ba91039362ececb838 /tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | |
parent | 4de9f67c91b9224f447bd4a50b67129064b8c824 (diff) | |
download | armnn-dc2797928dd0945f351445ba8d3be11b40376dd0.tar.gz |
IVGCVSW-2657: Model update to fix failing timings test: TfLiteVGG16Quantized
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I49bd93766f8f3cf41a9471c87c1c61ed1ec6034c
Diffstat (limited to 'tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp')
-rw-r--r-- | tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp index 84d5292195..0ba1e5dc00 100644 --- a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp +++ b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp @@ -34,10 +34,10 @@ int main(int argc, char* argv[]) retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>( argc, argv, - "vgg_16_u8_batch1.tflite", // model name + "vgg_16_int8.tflite", // model name true, // model is binary - "content_vgg/concat", // input tensor name - "content_vgg/prob", // output tensor name + "input", // input tensor name + "vgg_16/fc8/squeezed", // output tensor name { 0, 1, 2 }, // test images to test with as above [&imageSet](const char* dataDir, const ModelType & model) { // we need to get the input quantization parameters from |