diff options
Diffstat (limited to 'src/core/NEON/kernels/NEColorConvertKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEColorConvertKernel.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEColorConvertKernel.cpp b/src/core/NEON/kernels/NEColorConvertKernel.cpp index 4582c88487..7a66b6cc3a 100644 --- a/src/core/NEON/kernels/NEColorConvertKernel.cpp +++ b/src/core/NEON/kernels/NEColorConvertKernel.cpp @@ -112,6 +112,10 @@ void NEColorConvertKernel::configure(const ITensor *input, ITensor *output) _func = colorconvert_rgb_to_rgbx; num_elems_processed_per_iteration = 16; break; + case Format::U8: + _func = colorconvert_rgb_to_u8; + num_elems_processed_per_iteration = 16; + break; default: ARM_COMPUTE_ERROR("Not supported"); break; |