blob: 348dc279297798191fcf3f2387aa7c1bbac868fd [file] [log] [blame]
Georgios Pinitasf47f7182021-01-15 09:29:50 +00001/*
Matthew Bentham043613f2023-05-30 16:43:14 +00002 * Copyright (c) 2021, 2023 Arm Limited.
Georgios Pinitasf47f7182021-01-15 09:29:50 +00003 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24#ifndef ARM_COMPUTE_CL_ACTIVATION_H
25#define ARM_COMPUTE_CL_ACTIVATION_H
26
SiCong Li91295492023-07-21 18:16:13 +010027#include "arm_compute/function_info/ActivationLayerInfo.h"
Georgios Pinitas7891a732021-08-20 21:39:25 +010028#include "src/gpu/cl/ClCompileContext.h"
29#include "src/gpu/cl/IClOperator.h"
Georgios Pinitasf47f7182021-01-15 09:29:50 +000030
31namespace arm_compute
32{
33namespace opencl
34{
35/** Basic function to run @ref kernels::ClActivationKernel */
36class ClActivation : public IClOperator
37{
38public:
Georgios Pinitasf47f7182021-01-15 09:29:50 +000039 /** Configure operator for a given list of arguments
40 *
41 * @param[in] compile_context The compile context to be used.
42 * @param[in] src Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/QSYMM16/F16/F32.
43 * @param[out] dst Destination tensor info. Data type supported: same as @p src
44 * @param[in] activation_info Activation layer parameters.
45 */
46 void configure(const ClCompileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const ActivationLayerInfo &activation_info);
Georgios Pinitas2eb5d162021-07-02 09:01:49 +010047 /** Static function to check if given info will lead to a valid configuration
Georgios Pinitasf47f7182021-01-15 09:29:50 +000048 *
Georgios Pinitas2eb5d162021-07-02 09:01:49 +010049 * Similar to @ref ClActivation::configure()
Georgios Pinitasf47f7182021-01-15 09:29:50 +000050 *
51 * @return a status
52 */
53 static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const ActivationLayerInfo &act_info);
54};
55} // namespace opencl
56} // namespace arm_compute
57#endif /* ARM_COMPUTE_CL_ACTIVATION_H */