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.hpp | |
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.hpp')
-rw-r--r-- | src/backends/neon/NeonTensorHandleFactory.hpp | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/src/backends/neon/NeonTensorHandleFactory.hpp b/src/backends/neon/NeonTensorHandleFactory.hpp index e1cdc8bbac..0930d4e8d7 100644 --- a/src/backends/neon/NeonTensorHandleFactory.hpp +++ b/src/backends/neon/NeonTensorHandleFactory.hpp @@ -13,6 +13,30 @@ namespace armnn constexpr const char* NeonTensorHandleFactoryId() { return "Arm/Neon/TensorHandleFactory"; } +const std::set<armnn::LayerType> paddingRequiredLayers { + LayerType::ArgMinMax, + LayerType::Concat, + LayerType::Convolution2d, + LayerType::DepthToSpace, + LayerType::DepthwiseConvolution2d, + LayerType::Dequantize, + LayerType::FullyConnected, + LayerType::Gather, + LayerType::L2Normalization, + LayerType::Lstm, + LayerType::Mean, + LayerType::Multiplication, + LayerType::Normalization, + LayerType::Permute, + LayerType::Pooling2d, + LayerType::Quantize, + LayerType::QuantizedLstm, + LayerType::Resize, + LayerType::Stack, + LayerType::Transpose, + LayerType::TransposeConvolution2d +}; + class NeonTensorHandleFactory : public ITensorHandleFactory { public: @@ -46,6 +70,10 @@ public: MemorySourceFlags GetImportFlags() const override; + std::vector<Capability> GetCapabilities(const IConnectableLayer* layer, + const IConnectableLayer* connectedLayer, + CapabilityClass capabilityClass) override; + private: mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager; }; |