ArmNN
NotReleased
NeonSubtractionWorkload.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
10
#include <arm_compute/core/Error.h>
11
#include <arm_compute/runtime/IFunction.h>
12
13
#include <memory>
14
15
namespace
armnn
16
{
17
18
arm_compute::Status
NeonSubtractionWorkloadValidate
(
const
TensorInfo& input0,
19
const
TensorInfo& input1,
20
const
TensorInfo& output);
21
22
class
NeonSubtractionWorkload
:
public
BaseWorkload
<SubtractionQueueDescriptor>
23
{
24
public
:
25
NeonSubtractionWorkload
(
const
SubtractionQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
26
virtual
void
Execute
()
const override
;
27
28
private
:
29
std::unique_ptr<arm_compute::IFunction> m_SubLayer;
30
};
31
32
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::BoostLogSeverityMapping::info
armnn::NeonSubtractionWorkload::Execute
virtual void Execute() const override
Definition:
NeonSubtractionWorkload.cpp:46
Workload.hpp
armnn::NeonSubtractionWorkload::NeonSubtractionWorkload
NeonSubtractionWorkload(const SubtractionQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonSubtractionWorkload.cpp:31
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn
Definition:
BackendHelper.hpp:11
armnn::NeonSubtractionWorkloadValidate
arm_compute::Status NeonSubtractionWorkloadValidate(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output)
Definition:
NeonSubtractionWorkload.cpp:17
armnn::SubtractionQueueDescriptor
Definition:
WorkloadData.hpp:234
armnn::NeonSubtractionWorkload
Definition:
NeonSubtractionWorkload.hpp:22
armnn::BaseWorkload
Definition:
Workload.hpp:28
src
backends
neon
workloads
NeonSubtractionWorkload.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13