commit 5b7fd87b6ab73a7206a45d3621d484131c901143
author Sang-Hoon Park <sang-hoon.park@arm.com> Thu Jan 09 15:27:51 2020 +0000
committer Sang-Hoon Park <sang-hoon.park@arm.com> Fri Jan 10 11:12:39 2020 +0000
tree 27ce6a34e0e4f80de4cfc4aabbd72c43ee58d559
parent daa3aba64d01fd02fc74d0707d50d2571e5f0ce8

MLCE-139 add align_corners parameter handling to CLScale

Change-Id: I9d6047c306d7928258d5168eb4bfb96b09d2608f
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2569
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

arm_compute/core/CL/kernels/CLScaleKernel.h

diff --git a/arm_compute/core/CL/kernels/CLScaleKernel.h b/arm_compute/core/CL/kernels/CLScaleKernel.h
index 7d9ed41..f93286b 100644
--- a/arm_compute/core/CL/kernels/CLScaleKernel.h
+++ b/arm_compute/core/CL/kernels/CLScaleKernel.h
@@ -43,8 +43,9 @@
      * @param[in]  policy          Interpolation type to use
      * @param[in]  border_mode     Selected border mode.
      * @param[in]  sampling_policy (Optional) Sampling policy used by the interpolation. Defaults to @ref SamplingPolicy::CENTER
+     * @param[in]  align_corners   (Optional) Align corners of input and output, only affecting bilinear policy with TOP_LEFT sampling policy. Defaults to false.
      */
-    void configure(const ICLTensor *input, ICLTensor *output, InterpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy = SamplingPolicy::CENTER);
+    void configure(const ICLTensor *input, ICLTensor *output, InterpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy = SamplingPolicy::CENTER, bool align_corners = false);
 
     /** Static function to check if given info will lead to a valid configuration of @ref CLScaleKernel
      *
@@ -54,10 +55,12 @@
      * @param[in] policy          Interpolation type to use
      * @param[in] border_mode     Selected border mode.
      * @param[in] sampling_policy (Optional) Sampling policy used by the interpolation. Defaults to @ref SamplingPolicy::CENTER
+     * @param[in] align_corners   (Optional) Align corners of input and output, only affecting bilinear policy with TOP_LEFT sampling policy. Defaults to false.
      *
      * @return a status
      */
-    static Status validate(const ITensorInfo *input, const ITensorInfo *output, InterpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy = SamplingPolicy::CENTER);
+    static Status validate(const ITensorInfo *input, const ITensorInfo *output, InterpolationPolicy policy, BorderMode border_mode, SamplingPolicy sampling_policy = SamplingPolicy::CENTER,
+                           bool align_corners = false);
     /** Input tensor accessor.
      *
      * @return Pointer to input tensor.
@@ -76,6 +79,7 @@
 public:
     InterpolationPolicy _interpolationPolicy = InterpolationPolicy::BILINEAR;
     DataLayout          _data_layout         = DataLayout::UNKNOWN;
+    bool                _align_corners       = false;
 };
 } // namespace arm_compute
 #endif /*ARM_COMPUTE_CLSCALEKERNEL_H */