25.02
RefConvolution2dWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2022, 2024 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
RefBaseWorkload.hpp
"
9
#include <
armnn/backends/WorkloadData.hpp
>
10
#include "
Decoders.hpp
"
11
#include "
Encoders.hpp
"
12
13
namespace
armnn
14
{
15
16
class
RefConvolution2dWorkload
:
public
RefBaseWorkload
<Convolution2dQueueDescriptor>
17
{
18
public
:
19
explicit
RefConvolution2dWorkload
(
const
Convolution2dQueueDescriptor
& descriptor,
20
const
WorkloadInfo
&
info
);
21
22
void
Execute
()
const override
;
23
24
private
:
25
void
Execute
(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs)
const
;
26
27
const
TensorShape
m_InputShape;
28
const
TensorShape
m_FilterShape;
29
const
TensorShape
m_OutputShape;
30
};
31
32
}
//namespace armnn
Decoders.hpp
Encoders.hpp
RefBaseWorkload.hpp
WorkloadData.hpp
armnn::RefBaseWorkload
Definition:
RefBaseWorkload.hpp:14
armnn::RefConvolution2dWorkload
Definition:
RefConvolution2dWorkload.hpp:17
armnn::RefConvolution2dWorkload::RefConvolution2dWorkload
RefConvolution2dWorkload(const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
RefConvolution2dWorkload.cpp:15
armnn::RefConvolution2dWorkload::Execute
void Execute() const override
Definition:
RefConvolution2dWorkload.cpp:33
armnn::TensorShape
Definition:
Tensor.hpp:21
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:7
armnn::BoostLogSeverityMapping::info
@ info
armnn::Convolution2dQueueDescriptor
Definition:
WorkloadData.hpp:211
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:17
src
backends
reference
workloads
RefConvolution2dWorkload.hpp
Generated on Mon Feb 17 2025 21:34:26 for Arm NN by
1.9.1