From 01fd6cf41254995539d553baed6093f377871837 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 9 Jul 2019 12:35:25 +0100 Subject: COMPMID-2336: Fix build for v8.2-a architectures Define fp16 specializations for Dildated Depthwise Convolution. Change-Id: I491d0927478956cc4e542a6d7d63048aaa80c901 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/1499 Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/core') diff --git a/src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp b/src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp index 3e2bbbb61a..c4f398e199 100644 --- a/src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp +++ b/src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp @@ -30,3 +30,9 @@ template class depthwise::DilatedDepthwiseConvolution<3, 3, 3, 3, 1, 1, float, f template class depthwise::DilatedDepthwiseConvolution<3, 3, 3, 3, 2, 2, float, float, float>; template class depthwise::DilatedDepthwiseConvolution<4, 4, 3, 3, 1, 1, float, float, float>; template class depthwise::DilatedDepthwiseConvolution<4, 4, 3, 3, 2, 2, float, float, float>; + +#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC +template class depthwise::DilatedDepthwiseConvolution<3, 3, 3, 3, 1, 1, float16_t, float16_t, float16_t>; +template class depthwise::DilatedDepthwiseConvolution<3, 3, 3, 3, 2, 2, float16_t, float16_t, float16_t>; +#endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC + -- cgit v1.2.1