diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2020-11-27 12:40:52 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-11-30 15:12:22 +0000 |
| commit | 34fa1bd7994af9abf52dbcc4aa808d0fa5f14aa3 (patch) | |
| tree | 8ff6041b92da57cd56ccbcd3b746bdc0fb29078a /delegate/src/test/TestUtils.hpp | |
| parent | b7fa5104715e097bf3778d2485d759a4612460cc (diff) | |
| download | armnn-34fa1bd7994af9abf52dbcc4aa808d0fa5f14aa3.tar.gz | |
IVGCVSW-5393 'TfLiteDelegate: Implement the split operators'
* Added SPLIT and SPLIT_V support to armnn_delegate
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I2def9b8be783b25ef17a997e521c6027553035d3
Diffstat (limited to 'delegate/src/test/TestUtils.hpp')
| -rw-r--r-- | delegate/src/test/TestUtils.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/delegate/src/test/TestUtils.hpp b/delegate/src/test/TestUtils.hpp index 57ae3ce6fe..284eaa74f5 100644 --- a/delegate/src/test/TestUtils.hpp +++ b/delegate/src/test/TestUtils.hpp @@ -51,12 +51,13 @@ template <typename T> void CompareOutputData(std::unique_ptr<tflite::Interpreter>& tfLiteInterpreter, std::unique_ptr<tflite::Interpreter>& armnnDelegateInterpreter, std::vector<int32_t>& expectedOutputShape, - std::vector<T>& expectedOutputValues) + std::vector<T>& expectedOutputValues, + unsigned int outputIndex = 0) { - auto tfLiteDelegateOutputId = tfLiteInterpreter->outputs()[0]; + auto tfLiteDelegateOutputId = tfLiteInterpreter->outputs()[outputIndex]; auto tfLiteDelegateOutputTensor = tfLiteInterpreter->tensor(tfLiteDelegateOutputId); auto tfLiteDelegateOutputData = tfLiteInterpreter->typed_tensor<T>(tfLiteDelegateOutputId); - auto armnnDelegateOutputId = armnnDelegateInterpreter->outputs()[0]; + auto armnnDelegateOutputId = armnnDelegateInterpreter->outputs()[outputIndex]; auto armnnDelegateOutputTensor = armnnDelegateInterpreter->tensor(armnnDelegateOutputId); auto armnnDelegateOutputData = armnnDelegateInterpreter->typed_tensor<T>(armnnDelegateOutputId); |