diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2021-02-08 17:43:04 +0000 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-02-10 10:58:45 +0000 |
commit | 79144a642b33ff1ac40a44aaa1881261d12e6376 (patch) | |
tree | 81e25f9f99c03436fa3563b94881ea879e9b15c1 /SConscript | |
parent | 8c23ba1c5041467f314a10c1da9147e41d056139 (diff) | |
download | ComputeLibrary-79144a642b33ff1ac40a44aaa1881261d12e6376.tar.gz |
Decouple CpuPoolingKernel data type and data layout
1. Decouple data layout for CpuPoolingKernel: NCHW & NHWC
2. Decouple data type for CpuPoolingKernel NHWC
Partially solves: COMPMID-3999
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I3c6535eebdddeb467b7c68a7287a16959b5b9695
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5039
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'SConscript')
-rw-r--r-- | SConscript | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/SConscript b/SConscript index 7d20ffe3f1..39e7b9d5b0 100644 --- a/SConscript +++ b/SConscript @@ -295,6 +295,9 @@ if env['neon']: core_files += Glob('src/core/cpu/kernels/*/*/qsymm16.cpp') if any(i in env['data_type_support'] for i in ['all', 'integer']): core_files += Glob('src/core/cpu/kernels/*/*/integer.cpp') + + if any(i in env['data_layout_support'] for i in ['all', 'nchw']): + core_files += Glob('src/core/cpu/kernels/*/*/nchw/all.cpp') runtime_files += Glob('src/runtime/cpu/*.cpp') runtime_files += Glob('src/runtime/cpu/operators/*.cpp') |