diff options
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 |