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