22.05
RefDepthwiseConvolution2dWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#include "
RefBaseWorkload.hpp
"
6
#include <
armnn/backends/WorkloadData.hpp
>
7
#include "
Decoders.hpp
"
8
#include "
Encoders.hpp
"
9
10
#include <
armnn/TypesUtils.hpp
>
11
12
namespace
armnn
13
{
14
15
class
RefDepthwiseConvolution2dWorkload
:
public
RefBaseWorkload
<DepthwiseConvolution2dQueueDescriptor> {
16
public
:
17
explicit
RefDepthwiseConvolution2dWorkload
(
const
DepthwiseConvolution2dQueueDescriptor
&descriptor,
18
const
WorkloadInfo
&
info
);
19
void
Execute
()
const override
;
20
void
ExecuteAsync
(
WorkingMemDescriptor
& workingMemDescriptor)
override
;
21
22
private
:
23
void
Execute
(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs)
const
;
24
25
};
26
27
}
//namespace armnn
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
Encoders.hpp
armnn::experimental::WorkingMemDescriptor
Definition:
WorkingMemDescriptor.hpp:18
armnn::RefDepthwiseConvolution2dWorkload::ExecuteAsync
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
Definition:
RefDepthwiseConvolution2dWorkload.cpp:44
TypesUtils.hpp
armnn::RefDepthwiseConvolution2dWorkload::Execute
void Execute() const override
Definition:
RefDepthwiseConvolution2dWorkload.cpp:39
armnn::RefDepthwiseConvolution2dWorkload::RefDepthwiseConvolution2dWorkload
RefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
RefDepthwiseConvolution2dWorkload.cpp:18
armnn::RefBaseWorkload
Definition:
RefBaseWorkload.hpp:13
Decoders.hpp
armnn::RefDepthwiseConvolution2dWorkload
Definition:
RefDepthwiseConvolution2dWorkload.hpp:15
armnn::BoostLogSeverityMapping::info
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
RefBaseWorkload.hpp
armnn::DepthwiseConvolution2dQueueDescriptor
Depthwise Convolution 2D layer workload data.
Definition:
WorkloadData.hpp:247
WorkloadData.hpp
src
backends
reference
workloads
RefDepthwiseConvolution2dWorkload.hpp
Generated on Tue May 24 2022 11:27:14 for ArmNN by
1.8.13