ArmNN
NotReleased
ClPreluWorkload.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/runtime/CL/functions/CLPReluLayer.h>
11
12
namespace
armnn
13
{
14
arm_compute::Status
ClPreluWorkloadValidate
(
const
TensorInfo& input,
15
const
TensorInfo& alpha,
16
const
TensorInfo& output);
17
18
class
ClPreluWorkload
:
public
BaseWorkload
<PreluQueueDescriptor>
19
{
20
public
:
21
ClPreluWorkload
(
const
PreluQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
22
void
Execute
()
const override
;
23
24
private
:
25
mutable
arm_compute::CLPReluLayer m_PreluLayer;
26
};
27
28
}
//namespace armnn
armnn::ClPreluWorkloadValidate
arm_compute::Status ClPreluWorkloadValidate(const TensorInfo &input, const TensorInfo &alpha, const TensorInfo &output)
Definition:
ClPreluWorkload.cpp:16
armnn::Status
Status
Definition:
Types.hpp:26
armnn::BoostLogSeverityMapping::info
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::ClPreluWorkload
Definition:
ClPreluWorkload.hpp:18
armnn::ClPreluWorkload::Execute
void Execute() const override
Definition:
ClPreluWorkload.cpp:42
armnn
Definition:
BackendHelper.hpp:11
armnn::PreluQueueDescriptor
Definition:
WorkloadData.hpp:489
armnn::ClPreluWorkload::ClPreluWorkload
ClPreluWorkload(const PreluQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
ClPreluWorkload.cpp:29
armnn::BaseWorkload
Definition:
Workload.hpp:28
src
backends
cl
workloads
ClPreluWorkload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13