22.05
RefWorkloadFactoryHelper.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
backendsCommon/test/WorkloadFactoryHelper.hpp
>
9
10
#include <
armnn/utility/PolymorphicDowncast.hpp
>
11
12
#include <
reference/RefBackend.hpp
>
13
#include <
reference/RefWorkloadFactory.hpp
>
14
#include "
reference/RefTensorHandleFactory.hpp
"
15
16
namespace
17
{
18
19
template
<>
20
struct
WorkloadFactoryHelper<
armnn
::RefWorkloadFactory>
21
{
22
static
armnn::IBackendInternal::IMemoryManagerSharedPtr
GetMemoryManager()
23
{
24
armnn::RefBackend
backend;
25
return
backend.
CreateMemoryManager
();
26
}
27
28
static
armnn::RefWorkloadFactory
GetFactory(
29
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager =
nullptr
)
30
{
31
IgnoreUnused
(memoryManager);
32
return
armnn::RefWorkloadFactory
();
33
}
34
35
static
armnn::RefTensorHandleFactory
GetTensorHandleFactory(
36
const
armnn::IBackendInternal::IMemoryManagerSharedPtr
& memoryManager =
nullptr
)
37
{
38
39
return
armnn::RefTensorHandleFactory
(armnn::PolymorphicPointerDowncast<armnn::RefMemoryManager>(memoryManager));
40
}
41
};
42
43
using
RefWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::RefWorkloadFactory>;
44
45
}
// anonymous namespace
RefWorkloadFactory.hpp
armnn::RefBackend::CreateMemoryManager
IBackendInternal::IMemoryManagerUniquePtr CreateMemoryManager() const override
Definition:
RefBackend.cpp:62
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition:
IgnoreUnused.hpp:14
PolymorphicDowncast.hpp
armnn::IBackendInternal::IMemoryManagerSharedPtr
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
Definition:
IBackendInternal.hpp:97
RefBackend.hpp
armnn::RefBackend
Definition:
RefBackend.hpp:29
armnn::RefWorkloadFactory
Definition:
RefWorkloadFactory.hpp:30
WorkloadFactoryHelper.hpp
armnn::RefTensorHandleFactory
Definition:
RefTensorHandleFactory.hpp:17
RefTensorHandleFactory.hpp
src
backends
reference
test
RefWorkloadFactoryHelper.hpp
Generated on Tue May 24 2022 11:27:14 for ArmNN by
1.8.13