diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-04-16 10:28:19 +0100 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-04-16 10:29:12 +0100 |
commit | aab6aff4aa282810cb535eeec65e59741f1f4f0e (patch) | |
tree | 5e859ed1c4c4af623eceae211893ad4e7b225d14 | |
parent | 507f39d2bc6f1537faca86633bf756831f0c770d (diff) | |
download | armnn-aab6aff4aa282810cb535eeec65e59741f1f4f0e.tar.gz |
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 <nina.drozd@arm.com>
-rw-r--r-- | tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp | 2 |
1 files changed, 1 insertions, 1 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 03510f715d..48e63211a3 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 |