diff options
author | Finn Williams <finn.williams@arm.com> | 2022-01-21 19:33:46 +0000 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2022-01-24 12:24:56 +0000 |
commit | 35e7c1de53ef86b808fc8de0a0af8a747b4ef170 (patch) | |
tree | 42fed056f5adb2606723092dffb303cba7281b2c /tests/ExecuteNetwork/ExecuteNetwork.cpp | |
parent | 42d065f7966494a26be4a11844685ef13859bff3 (diff) | |
download | armnn-35e7c1de53ef86b808fc8de0a0af8a747b4ef170.tar.gz |
IVGCVSW-6733 Add missing qasymms8 output type to delegate
Signed-off-by: Finn Williams <finn.williams@arm.com>
Change-Id: Ic5ebf7b80468b7751c234c43a90ec4cbf4c59ffe
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 26 |
1 files changed, 4 insertions, 22 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 085721c6bb..f321a26009 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -154,7 +154,8 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, const armnn::IRun std::copy(tensorData.begin(), tensorData.end(), inputData); } - else if (params.m_InputTypes[inputIndex].compare("qsymms8") == 0) + else if (params.m_InputTypes[inputIndex].compare("qsymms8") == 0 || + params.m_InputTypes[inputIndex].compare("qasymms8") == 0) { auto inputData = tfLiteInterpreter->typed_tensor<int8_t>(input); @@ -215,26 +216,6 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, const armnn::IRun std::copy(tensorData.begin(), tensorData.end(), inputData); } - else if (params.m_InputTypes[inputIndex].compare("qasymms8") == 0) - { - auto inputData = tfLiteInterpreter->typed_tensor<int8_t>(input); - - if(inputData == NULL) - { - ARMNN_LOG(fatal) << "Input tensor is null, input type: " - "\"" << params.m_InputTypes[inputIndex] << "\" may be incorrect."; - return EXIT_FAILURE; - } - - std::vector<int8_t> tensorData; - PopulateTensorWithDataGeneric<int8_t>(tensorData, - inputSize, - dataFile, - [](const std::string& s) - { return armnn::numeric_cast<int8_t>(std::stoi(s)); }); - - std::copy(tensorData.begin(), tensorData.end(), inputData); - } else { ARMNN_LOG(fatal) << "Unsupported input tensor data type \"" << params.m_InputTypes[inputIndex] << "\". "; @@ -339,7 +320,8 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, const armnn::IRun } } } - else if (params.m_OutputTypes[outputIndex].compare("qsymms8") == 0) + else if (params.m_OutputTypes[outputIndex].compare("qsymms8") == 0 || + params.m_OutputTypes[outputIndex].compare("qasymms8") == 0) { auto tfLiteDelageOutputData = tfLiteInterpreter->typed_tensor<int8_t>(tfLiteDelegateOutputId); if(tfLiteDelageOutputData == NULL) |