diff options
author | ramelg01 <ramy.elgammal@arm.com> | 2022-02-26 22:06:20 +0000 |
---|---|---|
committer | Ramy Elgammal <ramy.elgammal@arm.com> | 2022-03-15 16:18:13 +0000 |
commit | 375156937a0783432c5d18e199b5d8d2b3ec33f7 (patch) | |
tree | 4c755770ac569c286179e5ce40df1c820baa73d8 /Android.bp | |
parent | 4e66d707a292b90a344e32c59eb1dacb67a0e4c1 (diff) | |
download | ComputeLibrary-375156937a0783432c5d18e199b5d8d2b3ec33f7.tar.gz |
Implementation of ClPooling3d
- For NDHWC layout
- For F16 and F32 data types
- Mixed Precision stil not supported
Resolves: COMPMID-4670
Signed-off-by: ramy.elgammal@arm.com
Change-Id: I0e14a13e4625569e8e5ee67e6033bd1efe0da469
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7262
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'Android.bp')
-rw-r--r-- | Android.bp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/Android.bp b/Android.bp index 491dbca165..c80bba0998 100644 --- a/Android.bp +++ b/Android.bp @@ -109,6 +109,7 @@ opencl_srcs = [ "src/core/CL/cl_kernels/nhwc/normalization_layer.cl", "src/core/CL/cl_kernels/nhwc/normalize_planar_yuv_layer.cl", "src/core/CL/cl_kernels/nhwc/normalize_planar_yuv_layer_quantized.cl", + "src/core/CL/cl_kernels/nhwc/pooling_3d_layer.cl", "src/core/CL/cl_kernels/nhwc/pooling_layer.cl", "src/core/CL/cl_kernels/nhwc/pooling_layer_quantized.cl", "src/core/CL/cl_kernels/nhwc/reorg_layer.cl", @@ -604,6 +605,7 @@ cc_library_static { "src/gpu/cl/kernels/ClMulKernel.cpp", "src/gpu/cl/kernels/ClPermuteKernel.cpp", "src/gpu/cl/kernels/ClPool2dKernel.cpp", + "src/gpu/cl/kernels/ClPool3dKernel.cpp", "src/gpu/cl/kernels/ClQuantizeKernel.cpp", "src/gpu/cl/kernels/ClReshapeKernel.cpp", "src/gpu/cl/kernels/ClScaleKernel.cpp", @@ -651,6 +653,7 @@ cc_library_static { "src/gpu/cl/operators/ClPRelu.cpp", "src/gpu/cl/operators/ClPermute.cpp", "src/gpu/cl/operators/ClPool2d.cpp", + "src/gpu/cl/operators/ClPool3d.cpp", "src/gpu/cl/operators/ClQuantize.cpp", "src/gpu/cl/operators/ClReshape.cpp", "src/gpu/cl/operators/ClScale.cpp", @@ -734,6 +737,7 @@ cc_library_static { "src/runtime/CL/functions/CLPadLayer.cpp", "src/runtime/CL/functions/CLPermute.cpp", "src/runtime/CL/functions/CLPixelWiseMultiplication.cpp", + "src/runtime/CL/functions/CLPooling3dLayer.cpp", "src/runtime/CL/functions/CLPoolingLayer.cpp", "src/runtime/CL/functions/CLPriorBoxLayer.cpp", "src/runtime/CL/functions/CLQLSTMLayer.cpp", |