IVGCVSW-2959: Update TfLiteResNetV2-50-Quantized-Armnn test
authorNina Drozd <nina.drozd@arm.com>
Tue, 16 Apr 2019 09:28:19 +0000 (10:28 +0100)
committerNina Drozd <nina.drozd@arm.com>
Tue, 16 Apr 2019 09:29:12 +0000 (10:29 +0100)
* 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 <nina.drozd@arm.com>
tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp

index 03510f7..48e6321 100644 (file)
@@ -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