diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-06-26 11:00:21 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-06-30 08:39:55 +0000 |
| commit | 3b90af6bb79639d3813f21f3c3017503ee0c66af (patch) | |
| tree | c5cccdb45ed268a47f5b4c8ac39a389fe037597c /src/backends/backendsCommon/test/EndToEndTestImpl.hpp | |
| parent | c0761e9a82ec4d6bd4a81fda018d624659b629a7 (diff) | |
| download | armnn-3b90af6bb79639d3813f21f3c3017503ee0c66af.tar.gz | |
IVGCVSW-5036 Do not allocate memory when import is enabled
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ideaae5280702aae6c73f3b4e4cee9f71a8386fda
Diffstat (limited to 'src/backends/backendsCommon/test/EndToEndTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/EndToEndTestImpl.hpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp index a4d0d505f8..0d16bcd7a9 100644 --- a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp @@ -719,6 +719,11 @@ inline void ExportOutputWithSeveralOutputSlotConnectionsTest(std::vector<Backend std::vector<float> outputData0(4); std::vector<float> outputData1(4); + std::vector<float> expectedOutput + { + 1.0f, 4.0f, 9.0f, 16.0f + }; + InputTensors inputTensors { {0,armnn::ConstTensor(runtime->GetInputTensorInfo(netId, 0), inputData.data())}, @@ -764,6 +769,12 @@ inline void ExportOutputWithSeveralOutputSlotConnectionsTest(std::vector<Backend // Contains CopyMemGeneric found = dump.find("CopyMemGeneric"); BOOST_TEST(found != std::string::npos); + + // Check that the outputs are correct + BOOST_CHECK_EQUAL_COLLECTIONS(outputData0.begin(), outputData0.end(), + expectedOutput.begin(), expectedOutput.end()); + BOOST_CHECK_EQUAL_COLLECTIONS(outputData1.begin(), outputData1.end(), + expectedOutput.begin(), expectedOutput.end()); } inline void StridedSliceInvalidSliceEndToEndTest(std::vector<BackendId> backends) |