blob: d544ef5728de5d3a404994b536817907d14a2df5 [file] [log] [blame]
Georgios Pinitasbdcdc392021-04-22 16:42:03 +01001/*
Gunes Bayir9b921be2022-07-28 17:44:00 +01002 * Copyright (c) 2021-2022 Arm Limited.
Georgios Pinitasbdcdc392021-04-22 16:42:03 +01003 *
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 */
Dana Zlotnikbd2942d2021-11-15 08:46:04 +020024#ifndef SRC_CORE_NEON_KERNELS_ADD_IMPL_H
25#define SRC_CORE_NEON_KERNELS_ADD_IMPL_H
26#include "arm_compute/core/ITensor.h"
Georgios Pinitasbdcdc392021-04-22 16:42:03 +010027#include "arm_compute/core/Types.h"
Dana Zlotnikbd2942d2021-11-15 08:46:04 +020028#include "arm_compute/core/Window.h"
Georgios Pinitasbdcdc392021-04-22 16:42:03 +010029namespace arm_compute
30{
31namespace cpu
32{
33template <typename ScalarType>
Dana Zlotnikbd2942d2021-11-15 08:46:04 +020034void add_same_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &policy, const Window &window);
Gunes Bayir9b921be2022-07-28 17:44:00 +010035
Viet-Hoa Do40b44192022-09-22 10:24:23 +010036bool add_q8_neon_fixedpoint_possible(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst);
37
Omar Al Khatib939b21a2022-12-20 14:36:45 +000038bool sub_q8_neon_fixedpoint_possible(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst);
39
40bool add_sub_q8_neon_fixedpoint_possible(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *dst, bool is_addition);
41
42void add_sub_qasymm8_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &policy, const Window &window, bool is_addition);
43
44void add_sub_qasymm8_signed_neon(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &policy, const Window &window, bool is_addition);
45
Viet-Hoa Do40b44192022-09-22 10:24:23 +010046template <typename ScalarType>
47void add_q8_neon_fixedpoint(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &policy, const Window &window);
Omar Al Khatib939b21a2022-12-20 14:36:45 +000048
49template <typename ScalarType>
50void add_sub_q8_neon_fixedpoint(const ITensor *src0, const ITensor *src1, ITensor *dst, const ConvertPolicy &policy, const Window &window, bool is_addition);
Georgios Pinitasbdcdc392021-04-22 16:42:03 +010051} // namespace cpu
52} // namespace arm_compute
Dana Zlotnikbd2942d2021-11-15 08:46:04 +020053#endif // SRC_CORE_NEON_KERNELS_ADD_IMPL_H