diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-14 12:25:50 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-14 14:38:24 +0000 |
commit | 4e3e818e1ea73544f3aec4e2ac2621a1c3380b54 (patch) | |
tree | 7a4cbdc3d95a1a2743194b7fdc61b81989f3ce2c /src/backends/neon/NeonBackend.hpp | |
parent | 4992c344c888774da0e5cffd4b607828bf46b74f (diff) | |
download | armnn-4e3e818e1ea73544f3aec4e2ac2621a1c3380b54.tar.gz |
IVGCVSW-3622 Add Neon TensorHandleFactory
Signed-off-by: James Conroy <james.conroy@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I9f24f7d97c1e7d37ee7d58ff85c6c3f8496e52ec
Diffstat (limited to 'src/backends/neon/NeonBackend.hpp')
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 398337f775..49b03b0fee 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -23,12 +23,19 @@ public: IWorkloadFactoryPtr CreateWorkloadFactory( const IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr) const override; + IWorkloadFactoryPtr CreateWorkloadFactory( + class TensorHandleFactoryRegistry& tensorHandleFactoryRegistry) const override; + IBackendInternal::IBackendContextPtr CreateBackendContext(const IRuntime::CreationOptions&) const override; IBackendInternal::Optimizations GetOptimizations() const override; IBackendInternal::ILayerSupportSharedPtr GetLayerSupport() const override; OptimizationViews OptimizeSubgraphView(const SubgraphView& subgraph) const override; + + std::vector<ITensorHandleFactory::FactoryId> GetHandleFactoryPreferences() const override; + + void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) override; }; } // namespace armnn |