diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-07-27 15:52:13 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-07-27 15:52:13 +0100 |
commit | 1a26896fd8d48205393ba0f22db864b5302b703f (patch) | |
tree | f5bdc6414951f10c88b6375cec694685b0682c8c /src/backends/neon/NeonTensorHandleFactory.cpp | |
parent | 11c4efc983cacfae6b1442d18a1744e114a82e64 (diff) | |
download | armnn-1a26896fd8d48205393ba0f22db864b5302b703f.tar.gz |
IVGCVSW-5011 Implement GetCapabilities in NeonTensorHandleFactory
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I43591ec1250c1d84d286de85956a86eb5e2abc2a
Diffstat (limited to 'src/backends/neon/NeonTensorHandleFactory.cpp')
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.cpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/neon/NeonTensorHandleFactory.cpp b/src/backends/neon/NeonTensorHandleFactory.cpp index a8b5b81412..ec9e0631fe 100644 --- a/src/backends/neon/NeonTensorHandleFactory.cpp +++ b/src/backends/neon/NeonTensorHandleFactory.cpp @@ -6,6 +6,8 @@ #include "NeonTensorHandleFactory.hpp" #include "NeonTensorHandle.hpp" +#include "Layer.hpp" + #include <armnn/utility/IgnoreUnused.hpp> #include <armnn/utility/PolymorphicDowncast.hpp> @@ -108,4 +110,23 @@ MemorySourceFlags NeonTensorHandleFactory::GetImportFlags() const return 0; } +std::vector<Capability> NeonTensorHandleFactory::GetCapabilities(const IConnectableLayer* layer, + const IConnectableLayer* connectedLayer, + CapabilityClass capabilityClass) + +{ + IgnoreUnused(connectedLayer); + std::vector<Capability> capabilities; + if (capabilityClass == CapabilityClass::PaddingRequired) + { + auto search = paddingRequiredLayers.find((PolymorphicDowncast<const Layer*>(layer))->GetType()); + if ( search != paddingRequiredLayers.end()) + { + Capability paddingCapability(CapabilityClass::PaddingRequired, true); + capabilities.push_back(paddingCapability); + } + } + return capabilities; +} + } // namespace armnn |