diff options
| author | Frederick Liardet <frederick.liardet@arm.com> | 2021-04-22 21:13:21 +0100 |
|---|---|---|
| committer | frederick.liardet <frederick.liardet@arm.com> | 2021-06-15 11:24:53 +0000 |
| commit | 36dff9f81e3a95aea19fcc7246a4896930a14bc6 (patch) | |
| tree | 64f3194e806bb4a8a5e6f2f30c202295c5e853c6 /src/runtime/CL/functions/CLRemap.cpp | |
| parent | ee301b384f4aeb697a5c249b8bb848d784146582 (diff) | |
| download | ComputeLibrary-36dff9f81e3a95aea19fcc7246a4896930a14bc6.tar.gz | |
Add NHWC support to CLRemap
Add NHWC support to CLRemap, also add relevant tests.
Partially resolves COMPMID-4335.
Change-Id: I119bea99be497fb85d5cd83a10f8d4e8e1f97f17
Signed-off-by: Freddie Liardet <frederick.liardet@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5773
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLRemap.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLRemap.cpp | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/runtime/CL/functions/CLRemap.cpp b/src/runtime/CL/functions/CLRemap.cpp index a4cfc60368..0a1f864543 100644 --- a/src/runtime/CL/functions/CLRemap.cpp +++ b/src/runtime/CL/functions/CLRemap.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -44,14 +44,8 @@ void CLRemap::configure(const CLCompileContext &compile_context, ICLTensor *inpu BorderMode border_mode, uint8_t constant_border_value) { - ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::U8); - ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output, 1, DataType::U8); - ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(map_x, 1, DataType::F32); - ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(map_y, 1, DataType::F32); - ARM_COMPUTE_ERROR_ON_MSG(policy == InterpolationPolicy::AREA, "Area interpolation is not supported"); - auto k = std::make_unique<CLRemapKernel>(); - k->configure(compile_context, input, map_x, map_y, output, policy, border_mode == BorderMode::UNDEFINED); + k->configure(compile_context, input, map_x, map_y, output, RemapInfo{ policy, border_mode, PixelValue(constant_border_value) }); _kernel = std::move(k); _border_handler->configure(compile_context, input, _kernel->border_size(), border_mode, PixelValue(constant_border_value)); } |