blob: 7ddc69b49a02c9560460c96d65f95fc5626197f9 [file] [log] [blame]
Sheri Zhang61243902021-01-12 18:25:16 +00001/*
2 * Copyright (c) 2021 Arm Limited.
3 *
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_CPU_ADD_H
25#define ARM_COMPUTE_CPU_ADD_H
26
27#include "src/runtime/cpu/ICpuOperator.h"
28
29namespace arm_compute
30{
31namespace cpu
32{
33/** Basic function to run @ref CpuAddKernel */
34class CpuAdd : public ICpuOperator
35{
36public:
37 /** Constructor */
38 CpuAdd() = default;
39 /** Initialise the kernel's input, dst and border mode.
40 *
41 * Valid configurations (src0,src1) -> dst :
42 *
43 * - (U8,U8) -> U8
44 * - (U8,U8) -> S16
45 * - (S16,U8) -> S16
46 * - (U8,S16) -> S16
47 * - (S16,S16) -> S16
48 * - (S32,S32) -> S32
49 * - (F16,F16) -> F16
50 * - (F32,F32) -> F32
51 * - (QASYMM8,QASYMM8) -> QASYMM8
52 * - (QASYMM8_SIGNED,QASYMM8_SIGNED) -> QASYMM8_SIGNED
53 * - (QSYMM16,QSYMM16) -> QSYMM16
54 *
55 * @param[in] src0 First input tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32
56 * @param[in] src1 Second input tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32
57 * @param[out] dst The dst tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32.
58 * @param[in] policy Overflow policy.
59 * @param[in] act_info (Optional) Activation layer information in case of a fused activation. Currently not supported.
60 *
61 */
62 void configure(const ITensorInfo *src0, const ITensorInfo *src1, ITensorInfo *dst, ConvertPolicy policy, const ActivationLayerInfo &act_info = ActivationLayerInfo());
63 /** Static function to check if given info will lead to a valid configuration of @ref CpuAddKernel
64 *
65 * @param[in] src0 First input tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32
66 * @param[in] src1 Second input tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32
67 * @param[in] dst The dst tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/QSYMM16/F16/S32/F32.
68 * @param[in] policy Overflow policy.
69 * @param[in] act_info (Optional) Activation layer information in case of a fused activation. Currently not supported.
70 *
71 * @return a status
72 */
73 static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst, ConvertPolicy policy, const ActivationLayerInfo &act_info = ActivationLayerInfo());
74};
75} // namespace cpu
76} // namespace arm_compute
77#endif /* ARM_COMPUTE_CPU_ADD_H */