diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-12-17 12:17:58 +0000 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-12-17 15:39:37 +0000 |
| commit | 958024be8f5c54f6e2a2930d40da62fda451bba7 (patch) | |
| tree | bf9cc93592a3a3fd9ca15d8a85edfda8204e67cf /delegate/src/test/TestUtils.hpp | |
| parent | 019840d1161738aefd6ebd32ccf4e72e618cae15 (diff) | |
| download | armnn-958024be8f5c54f6e2a2930d40da62fda451bba7.tar.gz | |
IVGCVSW-5383 TfLiteDelegate: Implement Pad and PadV2 operators
* Add Pad and PadV2 operators support to Armnn Delegate
* Add dimension check to CompareOutputData test utility
* Unit tests
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I9d00eb08f71e791498908fcbdb9de561e1c01aef
Diffstat (limited to 'delegate/src/test/TestUtils.hpp')
| -rw-r--r-- | delegate/src/test/TestUtils.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/delegate/src/test/TestUtils.hpp b/delegate/src/test/TestUtils.hpp index 284eaa74f5..b165920762 100644 --- a/delegate/src/test/TestUtils.hpp +++ b/delegate/src/test/TestUtils.hpp @@ -61,6 +61,9 @@ void CompareOutputData(std::unique_ptr<tflite::Interpreter>& tfLiteInterpreter, auto armnnDelegateOutputTensor = armnnDelegateInterpreter->tensor(armnnDelegateOutputId); auto armnnDelegateOutputData = armnnDelegateInterpreter->typed_tensor<T>(armnnDelegateOutputId); + CHECK(expectedOutputShape.size() == tfLiteDelegateOutputTensor->dims->size); + CHECK(expectedOutputShape.size() == armnnDelegateOutputTensor->dims->size); + for (size_t i = 0; i < expectedOutputShape.size(); i++) { CHECK(expectedOutputShape[i] == armnnDelegateOutputTensor->dims->data[i]); |