diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-07-22 12:46:51 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-07-23 21:17:37 +0000 |
commit | d6568772be726acd7dc2fc3e592f101c77d690a6 (patch) | |
tree | b385a4f1231c9cd15f61b93e83779b484593ccf6 /src | |
parent | 5b4d0d5070cee848901333e32cde7df2badf5d44 (diff) | |
download | armnn-d6568772be726acd7dc2fc3e592f101c77d690a6.tar.gz |
IVGCVSW-5010 Add GetCapabilities to ITensorHandleFactory
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie8acb9c729af4f95488aecf795f45ff12364f9ca
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/reference/test/RefTensorHandleTests.cpp | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefTensorHandleTests.cpp b/src/backends/reference/test/RefTensorHandleTests.cpp index 42f5664b72..3635a32472 100644 --- a/src/backends/reference/test/RefTensorHandleTests.cpp +++ b/src/backends/reference/test/RefTensorHandleTests.cpp @@ -141,6 +141,23 @@ BOOST_AUTO_TEST_CASE(RefTensorHandleImport) BOOST_CHECK(buffer[1] == 10.0f); } +BOOST_AUTO_TEST_CASE(RefTensorHandleGetCapabilities) +{ + std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); + RefTensorHandleFactory handleFactory(memoryManager); + + // Builds up the structure of the network. + INetworkPtr net(INetwork::Create()); + IConnectableLayer* input = net->AddInputLayer(0); + IConnectableLayer* output = net->AddOutputLayer(0); + input->GetOutputSlot(0).Connect(output->GetInputSlot(0)); + + std::vector<Capability> capabilities = handleFactory.GetCapabilities(input, + output, + CapabilityClass::PaddingRequired); + BOOST_CHECK(capabilities.empty()); +} + #if !defined(__ANDROID__) // Only run these tests on non Android platforms BOOST_AUTO_TEST_CASE(CheckSourceType) |