From aab6aff4aa282810cb535eeec65e59741f1f4f0e Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Tue, 16 Apr 2019 10:28:19 +0100 Subject: [PATCH] IVGCVSW-2959: Update TfLiteResNetV2-50-Quantized-Armnn test * updated the output name parameter in TfLiteResNetV2-50-Quantized test to correct output name as in model Change-Id: Ic657d5e7972a610ea63d4fc709fe42c86c997249 Signed-off-by: Nina Drozd --- .../TfLiteResNetV2-50-Quantized-Armnn.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 03510f7..48e6321 100644 --- a/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp +++ b/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp @@ -35,7 +35,7 @@ int main(int argc, char* argv[]) "resnet_v2_50_default_minmax.tflite", // model name true, // model is binary "input", // input tensor name - "output", // output tensor name + "resnet_v2_50/predictions/Reshape_1", // 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 -- 2.7.4