22.08
EthosnRefTransposeConvolution2dWorkload.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 <
backendsCommon/Workload.hpp
>
9
#include <
backendsCommon/WorkloadData.hpp
>
10
11
namespace
armnn
12
{
13
14
class
EthosnRefTransposeConvolution2dWorkload
:
public
BaseWorkload
<TransposeConvolution2dQueueDescriptor>
15
{
16
public
:
17
EthosnRefTransposeConvolution2dWorkload
(
const
TransposeConvolution2dQueueDescriptor
& descriptor,
18
const
WorkloadInfo
&
info
);
19
~EthosnRefTransposeConvolution2dWorkload
() =
default
;
20
21
void
Execute
()
const override
;
22
23
private
:
24
std::unique_ptr<ScopedCpuTensorHandle> m_Weight;
25
std::unique_ptr<ScopedCpuTensorHandle> m_Bias;
26
27
TensorShape
m_InputShape;
28
TensorShape
m_OutputShape;
29
};
30
31
}
// namespace armnn
armnn::EthosnRefTransposeConvolution2dWorkload
Definition:
EthosnRefTransposeConvolution2dWorkload.hpp:14
armnn::EthosnRefTransposeConvolution2dWorkload::~EthosnRefTransposeConvolution2dWorkload
~EthosnRefTransposeConvolution2dWorkload()=default
WorkloadData.hpp
armnn::EthosnRefTransposeConvolution2dWorkload::EthosnRefTransposeConvolution2dWorkload
EthosnRefTransposeConvolution2dWorkload(const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
EthosnRefTransposeConvolution2dWorkload.cpp:16
armnn::EthosnRefTransposeConvolution2dWorkload::Execute
void Execute() const override
Definition:
EthosnRefTransposeConvolution2dWorkload.cpp:30
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::TensorShape
Definition:
Tensor.hpp:20
Workload.hpp
armnn::BaseWorkload
Definition:
Workload.hpp:33
armnn::TransposeConvolution2dQueueDescriptor
Definition:
WorkloadData.hpp:584
armnn::BoostLogSeverityMapping::info
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition:
WorkloadInfo.hpp:16
src
backends
ethosnref
workloads
EthosnRefTransposeConvolution2dWorkload.hpp
Generated on Fri Aug 19 2022 14:38:30 for ArmNN by
1.8.13