24.02
Main Page
Related Pages
Namespaces
Namespace List
Namespace Members
All
a
b
c
d
e
f
g
h
i
j
l
m
n
o
p
q
r
s
t
u
v
w
Functions
a
b
c
d
e
f
g
h
i
l
m
n
o
p
q
r
s
t
v
w
Variables
Typedefs
a
b
c
d
e
f
g
h
i
l
m
n
o
p
q
r
s
t
u
w
Enumerations
a
b
c
d
e
f
g
j
l
m
n
o
p
q
r
s
t
u
Classes
Class List
Class Index
Class Hierarchy
Class Members
All
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
~
Functions
a
b
c
d
e
f
g
h
i
j
l
m
n
o
p
q
r
s
t
u
v
w
~
Variables
c
d
e
f
i
j
l
m
o
p
r
s
v
w
Typedefs
a
b
c
d
e
f
g
h
i
k
l
m
n
o
p
r
s
t
v
w
Enumerations
Enumerator
Related Functions
Files
File List
File Members
All
a
b
c
d
e
f
g
h
i
l
m
o
r
s
t
u
v
x
z
Functions
a
b
c
g
i
m
r
s
t
v
z
Variables
Typedefs
Macros
a
c
d
e
f
h
i
l
o
s
t
u
v
x
Examples
•
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Pages
WorkloadFactoryBase.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2019, 2023 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
armnn/backends/WorkloadFactory.hpp
>
9
10
namespace
armnn
11
{
12
13
class
WorkloadFactoryBase
:
public
IWorkloadFactory
14
{
15
public
:
16
bool
SupportsSubTensors
()
const override
17
{
return
false
; };
18
19
std::unique_ptr<ITensorHandle>
CreateSubTensorHandle
(
ITensorHandle
&
/*parent*/
,
20
TensorShape
const
&
/*subTensorShape*/
,
21
unsigned
int
const
*
/*subTensorOrigin*/
)
const override
22
{
return
nullptr
; };
23
24
25
std::unique_ptr<ITensorHandle>
CreateTensorHandle
(
const
TensorInfo
&
/*tensorInfo*/
,
26
const
bool
/*IsMemoryManaged*/
)
const override
27
{
return
nullptr
; }
28
29
std::unique_ptr<ITensorHandle>
CreateTensorHandle
(
const
TensorInfo
&
/*tensorInfo*/
,
30
DataLayout
/*dataLayout*/
,
31
const
bool
/*IsMemoryManaged*/
)
const override
32
{
return
nullptr
; }
33
34
std::unique_ptr<IWorkload>
CreateWorkload
(
LayerType
/*type*/
,
35
const
QueueDescriptor
&
/*descriptor*/
,
36
const
WorkloadInfo
&
/*info*/
)
const override
37
{
return
nullptr
; }
38
39
};
40
41
}
//namespace armnn
armnn::WorkloadFactoryBase::CreateTensorHandle
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, const bool) const override
Definition:
WorkloadFactoryBase.hpp:25
armnn::WorkloadFactoryBase
Definition:
WorkloadFactoryBase.hpp:13
armnn::DataLayout
DataLayout
Definition:
Types.hpp:62
armnn::TensorInfo
Definition:
Tensor.hpp:152
armnn::ITensorHandle
Definition:
ITensorHandle.hpp:16
WorkloadFactory.hpp
armnn::TensorShape
Definition:
Tensor.hpp:20
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
armnn::IWorkloadFactory
Definition:
WorkloadFactory.hpp:22
armnn::QueueDescriptor
Definition:
WorkloadData.hpp:24
armnn::WorkloadFactoryBase::CreateWorkload
std::unique_ptr< IWorkload > CreateWorkload(LayerType, const QueueDescriptor &, const WorkloadInfo &) const override
Backends should implement their own CreateWorkload function with a switch statement.
Definition:
WorkloadFactoryBase.hpp:34
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::WorkloadFactoryBase::CreateTensorHandle
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, DataLayout, const bool) const override
Definition:
WorkloadFactoryBase.hpp:29
armnn::WorkloadFactoryBase::SupportsSubTensors
bool SupportsSubTensors() const override
Definition:
WorkloadFactoryBase.hpp:16
armnn::LayerType
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.
Definition:
Types.hpp:491
armnn::WorkloadFactoryBase::CreateSubTensorHandle
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &, TensorShape const &, unsigned int const *) const override
Definition:
WorkloadFactoryBase.hpp:19
src
backends
backendsCommon
WorkloadFactoryBase.hpp
Generated on Wed Feb 14 2024 16:36:15 for Arm NN by
1.8.17