20.02
NeonWorkloadFactoryHelper.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
armnn/backends/IBackendInternal.hpp
>
9
#include <
armnn/backends/IMemoryManager.hpp
>
10
#include <
backendsCommon/test/WorkloadFactoryHelper.hpp
>
11
12
#include <
neon/NeonBackend.hpp
>
13
#include <
neon/NeonWorkloadFactory.hpp
>
14
15
#include <boost/polymorphic_pointer_cast.hpp>
16
17
namespace
18
{
19
20
template
<>
21
struct
WorkloadFactoryHelper<
armnn
::NeonWorkloadFactory>
22
{
23
static
armnn::IBackendInternal::IMemoryManagerSharedPtr
GetMemoryManager()
24
{
25
armnn::NeonBackend
backend;
26
return
backend.
CreateMemoryManager
();
27
}
28
29
static
armnn::NeonWorkloadFactory
GetFactory(
30
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager)
31
{
32
return
armnn::NeonWorkloadFactory
(
33
boost::polymorphic_pointer_downcast<armnn::NeonMemoryManager>(memoryManager));
34
}
35
};
36
37
using
NeonWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::NeonWorkloadFactory>;
38
39
}
// anonymous namespace
NeonBackend.hpp
armnn::NeonWorkloadFactory
Definition:
NeonWorkloadFactory.hpp:17
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
IBackendInternal.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition:
IBackendInternal.hpp:90
WorkloadFactoryHelper.hpp
IMemoryManager.hpp
NeonWorkloadFactory.hpp
armnn::NeonBackend
Definition:
NeonBackend.hpp:12
armnn::NeonBackend::CreateMemoryManager
IBackendInternal::IMemoryManagerUniquePtr CreateMemoryManager() const override
Definition:
NeonBackend.cpp:35
src
backends
neon
test
NeonWorkloadFactoryHelper.hpp
Generated on Fri Mar 13 2020 16:09:12 for ArmNN by
1.8.13