blob: d60e15ec847503609d0934e36a761446e3697f78 [file] [log] [blame]
Michele Di Giorgiod02d5ed2021-01-22 09:47:04 +00001/*
ramelg018a164882022-04-07 02:42:52 +01002 * Copyright (c) 2021-2022 Arm Limited.
Michele Di Giorgiod02d5ed2021-01-22 09:47:04 +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
25#include "src/core/NEON/kernels/arm_gemm/utils.hpp"
26
27#include <cstdint>
28
29#pragma once
30
ramelg018a164882022-04-07 02:42:52 +010031#if defined(__aarch64__)
32
Michele Di Giorgiod02d5ed2021-01-22 09:47:04 +000033namespace arm_conv {
34namespace depthwise {
35
36void a64_fp32_packed_to_nhwc_generic_with_multiplier_output2x8_mla_depthfirst_impl(const float *const *const, float *const *const, const float *, const float *, const unsigned int, const unsigned int, const float, const float);
37
ramelg018a164882022-04-07 02:42:52 +010038struct a64_fp32_packed_to_nhwc_generic_with_multiplier_output2x8_mla_depthfirst : GenericDepthfirstMultiplierKernelStrategy<float, float, float, float>
Michele Di Giorgiod02d5ed2021-01-22 09:47:04 +000039{
ramelg018a164882022-04-07 02:42:52 +010040 using Parent = GenericDepthfirstMultiplierKernelStrategy<float, float, float, float>;
41 a64_fp32_packed_to_nhwc_generic_with_multiplier_output2x8_mla_depthfirst(const CPUInfo *)
42 : Parent(2, 8, arm_gemm::VLType::None)
43 {
44 }
45 Parent::KernelType kernel = a64_fp32_packed_to_nhwc_generic_with_multiplier_output2x8_mla_depthfirst_impl;
46 Parent::KernelType get_kernel(void) const override { return kernel; }
Michele Di Giorgiod02d5ed2021-01-22 09:47:04 +000047};
48
49} // namespace depthwise
50} // namespace arm_conv
ramelg018a164882022-04-07 02:42:52 +010051
52#endif // defined(__aarch64__)