diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/KernelDescriptors.h | 12 | ||||
-rw-r--r-- | arm_compute/core/Types.h | 3 | ||||
-rw-r--r-- | arm_compute/runtime/CL/functions/CLRemap.h | 4 |
3 files changed, 14 insertions, 5 deletions
diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index 1a7ead4700..6c1fc74b1e 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -210,5 +210,17 @@ struct ScaleKernelInfo bool align_corners; /**< Align corners of input and output */ DataLayout data_layout; /**< Data layout to use */ }; + +struct RemapInfo +{ + RemapInfo() = default; + RemapInfo(InterpolationPolicy policy, BorderMode border_mode, PixelValue constant_border_value) + : policy(policy), border_mode(border_mode), constant_border_value(constant_border_value) + { + } + InterpolationPolicy policy; + BorderMode border_mode; + PixelValue constant_border_value; +}; } // namespace arm_compute #endif /* ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H */ diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 2dc9a77c39..48c87cd8ac 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -105,9 +105,6 @@ enum class SamplingPolicy TOP_LEFT /**< Samples are taken at pixel top left corner */ }; -/** Constant value of the border pixels when using BorderMode::CONSTANT */ -constexpr uint8_t CONSTANT_BORDER_VALUE = 199; - /** [DataLayout enum definition] **/ /** Supported tensor data layouts */ diff --git a/arm_compute/runtime/CL/functions/CLRemap.h b/arm_compute/runtime/CL/functions/CLRemap.h index 8466a79bb9..f69b045c9b 100644 --- a/arm_compute/runtime/CL/functions/CLRemap.h +++ b/arm_compute/runtime/CL/functions/CLRemap.h @@ -58,7 +58,7 @@ public: * @param[in] map_y Map for Y coords. Data types supported: F32. * @param[out] output Output tensor. Data types supported: U8. * @param[in] policy Interpolation policy to use. Only NEAREST and BILINEAR are supported. - * @param[in] border_mode Border mode to use on the input tensor. + * @param[in] border_mode Border mode to use on the input tensor. Only CONSTANT and UNDEFINED are supported. * @param[in] constant_border_value (Optional) Constant value to use for borders if border_mode is set to CONSTANT. * */ @@ -72,7 +72,7 @@ public: * @param[in] map_y Map for Y coords. Data types supported: F32. * @param[out] output Output tensor. Data types supported: U8. * @param[in] policy Interpolation policy to use. Only NEAREST and BILINEAR are supported. - * @param[in] border_mode Border mode to use on the input tensor. + * @param[in] border_mode Border mode to use on the input tensor. Only CONSTANT and UNDEFINED are supported. * @param[in] constant_border_value (Optional) Constant value to use for borders if border_mode is set to CONSTANT. * */ |