20.05
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 <
armnn/utility/PolymorphicDowncast.hpp
>
11
#include <
backendsCommon/test/WorkloadFactoryHelper.hpp
>
12
13
#include <
neon/NeonBackend.hpp
>
14
#include <
neon/NeonWorkloadFactory.hpp
>
15
16
namespace
17
{
18
19
template
<>
20
struct
WorkloadFactoryHelper<
armnn
::NeonWorkloadFactory>
21
{
22
static
armnn::IBackendInternal::IMemoryManagerSharedPtr
GetMemoryManager()
23
{
24
armnn::NeonBackend
backend;
25
return
backend.
CreateMemoryManager
();
26
}
27
28
static
armnn::NeonWorkloadFactory
GetFactory(
29
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager)
30
{
31
return
armnn::NeonWorkloadFactory
(
32
armnn::PolymorphicPointerDowncast<armnn::NeonMemoryManager>(memoryManager));
33
}
34
};
35
36
using
NeonWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::NeonWorkloadFactory>;
37
38
}
// anonymous namespace
NeonBackend.hpp
armnn::NeonWorkloadFactory
Definition:
NeonWorkloadFactory.hpp:17
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
PolymorphicDowncast.hpp
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:36
src
backends
neon
test
NeonWorkloadFactoryHelper.hpp
Generated on Fri May 29 2020 10:20:45 for ArmNN by
1.8.13