diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-07-09 12:35:25 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-07-09 13:12:44 +0000 |
| commit | 01fd6cf41254995539d553baed6093f377871837 (patch) | |
| tree | 9b4a99ab524156ade804223505e9528a76835a52 | |
| parent | f6f23ea562a448d72a8c62512e9e7fab92921b91 (diff) | |
| download | ComputeLibrary-01fd6cf41254995539d553baed6093f377871837.tar.gz | |
COMPMID-2336: Fix build for v8.2-a architectures
Define fp16 specializations for Dildated Depthwise Convolution.
Change-Id: I491d0927478956cc4e542a6d7d63048aaa80c901
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1499
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/core/NEON/kernels/convolution/depthwise/depthwise_dilated.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
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 + |