diff options
Diffstat (limited to 'src/armnn/test/RuntimeTests.cpp')
| -rw-r--r-- | src/armnn/test/RuntimeTests.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/test/RuntimeTests.cpp b/src/armnn/test/RuntimeTests.cpp index 6d5e2ae2b5..4bb2f44357 100644 --- a/src/armnn/test/RuntimeTests.cpp +++ b/src/armnn/test/RuntimeTests.cpp @@ -116,7 +116,9 @@ TEST_CASE("RuntimePreImportInputs") auto memHandle = runtime->CreateWorkingMemHandle(networkId); +ARMNN_NO_DEPRECATE_WARN_BEGIN runtime->Execute(*memHandle.get(), {{1, inputTensor2}}, {{2, outputTensor}}, {0 /* pre-imported id */}); +ARMNN_NO_DEPRECATE_WARN_END for (auto val: output) { CHECK(val == 30); } @@ -125,12 +127,16 @@ TEST_CASE("RuntimePreImportInputs") CHECK(importedInputVec2.size() == 1); CHECK(importedInputVec2[0] == 1); +ARMNN_NO_DEPRECATE_WARN_BEGIN runtime->Execute(*memHandle.get(), {{0, inputTensor1}}, {{2, outputTensor}}, {1 /* pre-imported id */}); +ARMNN_NO_DEPRECATE_WARN_END for (auto val: output) { CHECK(val == 30); } +ARMNN_NO_DEPRECATE_WARN_BEGIN runtime->Execute(*memHandle.get(), {}, {{2, outputTensor}}, {0, 1}); +ARMNN_NO_DEPRECATE_WARN_BEGIN for (auto val: output) { CHECK(val == 30); } |