blob: 41584e954bd4aeb423ed1390166d228c9a105bd6 [file] [log] [blame]
Ramy Elgammal21fb2ad2024-05-13 11:12:11 +01001/*
2 * Copyright (c) 2024 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#if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
25
26#include "src/cpu/kernels/reduction_layer/generic/neon/impl_fp16.h"
27
28namespace arm_compute
29{
30namespace cpu
31{
32void reduce_RedOpX_reduceX_float16_8(const Window &window,
33 const ITensor *input,
34 ITensor *output,
35 const ReductionOperation op)
36{
37 return Reducer<RedOpX<float16_t, 8>>::reduceX(window, input, output, RedOpX<float16_t, 8>(), op);
38}
39
40void reduce_RedOpYZW_reduceY_float16_8(const Window &window,
41 const ITensor *input,
42 ITensor *output,
43 const ReductionOperation op)
44{
45 return Reducer<RedOpYZW<float16_t, 8>>::reduceY(window, input, output, RedOpYZW<float16_t, 8>(), op);
46}
47
48void reduce_RedOpYZW_reduceZ_float16_8(const Window &window,
49 const ITensor *input,
50 ITensor *output,
51 const ReductionOperation op)
52{
53 return Reducer<RedOpYZW<float16_t, 8>>::reduceZ(window, input, output, RedOpYZW<float16_t, 8>(), op);
54}
55
56void reduce_RedOpYZW_reduceW_float16_8(const Window &window,
57 const ITensor *input,
58 ITensor *output,
59 const ReductionOperation op)
60{
61 return Reducer<RedOpYZW<float16_t, 8>>::reduceW(window, input, output, RedOpYZW<float16_t, 8>(), op);
62}
63} // namespace cpu
64} // namespace arm_compute
65#endif /* defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) */