aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp')
-rw-r--r--tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp2
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