diff options
Diffstat (limited to 'src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp')
| -rw-r--r-- | src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp b/src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp index 63f5fc786c..d12f3c60c0 100644 --- a/src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp +++ b/src/core/NEON/kernels/convolution/winograd/weight_transforms_fp32.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022 Arm Limited. + * Copyright (c) 2022-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -33,12 +33,12 @@ namespace weight_transform { #if defined(ARM_COMPUTE_ENABLE_SVE) #endif // defined(ARM_COMPUTE_ENABLE_SVE) #endif // defined(__aarch64__) -void *arm_fp32_4x4_3x3(unsigned int, const float *, size_t, size_t, float *, size_t); -void *arm_fp32_2x2_3x3(unsigned int, const float *, size_t, size_t, float *, size_t); -void *arm_fp32_2x2_5x5(unsigned int, const float *, size_t, size_t, float *, size_t); -void *cpp_fp32_1x6_1x3(unsigned int, const float *, size_t, size_t, float *, size_t); -void *cpp_fp32_1x4_1x5(unsigned int, const float *, size_t, size_t, float *, size_t); -void *cpp_fp32_1x2_1x7(unsigned int, const float *, size_t, size_t, float *, size_t); +void arm_fp32_4x4_3x3(unsigned int, const float *, size_t, size_t, float *, size_t); +void arm_fp32_2x2_3x3(unsigned int, const float *, size_t, size_t, float *, size_t); +void arm_fp32_2x2_5x5(unsigned int, const float *, size_t, size_t, float *, size_t); +void cpp_fp32_1x6_1x3(unsigned int, const float *, size_t, size_t, float *, size_t); +void cpp_fp32_1x4_1x5(unsigned int, const float *, size_t, size_t, float *, size_t); +void cpp_fp32_1x2_1x7(unsigned int, const float *, size_t, size_t, float *, size_t); #define IMPL(KERN_ROWS, KERN_COLS, TRANS_ROWS, TRANS_COLS, KERN) \ new Transform<float>(#KERN, KERN_ROWS, KERN_COLS, TRANS_ROWS, TRANS_COLS, KERN) |