21 #include <arm_compute/runtime/Allocator.h> 23 #include <boost/cast.hpp> 24 #include <boost/polymorphic_pointer_cast.hpp> 37 return std::make_unique<NeonMemoryManager>(std::make_unique<arm_compute::Allocator>(),
44 return std::make_unique<NeonWorkloadFactory>(
45 boost::polymorphic_pointer_downcast<NeonMemoryManager>(memoryManager));
51 auto memoryManager = std::make_shared<NeonMemoryManager>(std::make_unique<arm_compute::Allocator>(),
55 return std::make_unique<NeonWorkloadFactory>(
56 boost::polymorphic_pointer_downcast<NeonMemoryManager>(memoryManager));
87 return optimizationViews;
92 return std::vector<ITensorHandleFactory::FactoryId>() = {
"Arm/Neon/TensorHandleFactory",
93 "Arm/Cl/TensorHandleFactory"};
98 auto memoryManager = std::make_shared<NeonMemoryManager>(std::make_unique<arm_compute::Allocator>(),
102 registry.
RegisterFactory(std::make_unique<NeonTensorHandleFactory>(memoryManager));
void RegisterMemoryManager(std::shared_ptr< IMemoryManager > memoryManger)
Register a memory manager with shared ownership.
std::unique_ptr< IWorkloadFactory > IWorkloadFactoryPtr
std::vector< OptimizationPtr > Optimizations
IBackendInternal::IBackendProfilingContextPtr CreateBackendProfilingContext(const IRuntime::CreationOptions &, IBackendProfilingPtr &backendProfiling) override
Create context specifically used for profiling interaction from backends.
std::vector< ITensorHandleFactory::FactoryId > GetHandleFactoryPreferences() const override
(Optional) Returns a vector of supported TensorHandleFactory ids in preference order.
OptimizationViews OptimizeSubgraphView(const SubgraphView &subgraph) const override
void RegisterFactory(std::unique_ptr< ITensorHandleFactory > allocator)
Register a TensorHandleFactory and transfer ownership.
IWorkloadFactoryPtr CreateWorkloadFactory(const IBackendInternal::IMemoryManagerSharedPtr &memoryManager=nullptr) const override
constexpr const char * NeonBackendId()
IBackendInternal::IBackendContextPtr CreateBackendContext(const IRuntime::CreationOptions &) const override
Create the runtime context of the backend.
Copyright (c) 2020 ARM Limited.
std::unique_ptr< IMemoryManager > IMemoryManagerUniquePtr
The SubgraphView class represents a subgraph of a Graph.
void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry ®istry) override
(Optional) Register TensorHandleFactories Either this method or CreateMemoryManager() and IWorkloadFa...
std::unique_ptr< armnn::profiling::IBackendProfiling > IBackendProfilingPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
IBackendInternal::ILayerSupportSharedPtr GetLayerSupport() const override
void AddUntouchedSubgraph(SubgraphView &&subgraph)
std::shared_ptr< ILayerSupport > ILayerSupportSharedPtr
IBackendInternal::Optimizations GetOptimizations() const override
static const BackendId & GetIdStatic()
std::shared_ptr< armnn::profiling::IBackendProfilingContext > IBackendProfilingContextPtr
This is the bridge between backend and backend profiling we'll keep it in the backend namespace...
IBackendInternal::IMemoryManagerUniquePtr CreateMemoryManager() const override
std::unique_ptr< IBackendContext > IBackendContextPtr