From d473386e4d5e0edcf55e13a2bf3c422a23fac0de Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Tue, 9 Jul 2019 14:21:06 +0100 Subject: COMPMID-2447: Align TFlite nearest neighbor NE/CL functions with ACL Change-Id: Idd7b23247491d6e2e31d19b2a8aa522470ca174c Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/1500 Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- src/core/GLES_COMPUTE/cs_shaders/scale.cs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'src/core/GLES_COMPUTE/cs_shaders') diff --git a/src/core/GLES_COMPUTE/cs_shaders/scale.cs b/src/core/GLES_COMPUTE/cs_shaders/scale.cs index b72c3392aa..8a1d3e4c2d 100644 --- a/src/core/GLES_COMPUTE/cs_shaders/scale.cs +++ b/src/core/GLES_COMPUTE/cs_shaders/scale.cs @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016-2018 ARM Limited. + * Copyright (c) 2016-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -58,8 +58,15 @@ vec4[2] transform_nearest(vec2 coord, vec2 scale) vec4 in_x_coords = vec4(coord.x, 1.f + coord.x, 2.f + coord.x, 3.f + coord.x); vec4[2] t; +#if defined(SAMPLING_POLICY_CENTER) /* SAMPLING_POLICY_CENTER */ t[0] = (in_x_coords + (vec4(0.5f))) * scale.x; t[1] = vec4((coord.y + 0.5f) * scale.y); +#elif defined(SAMPLING_POLICY_TOP_LEFT) /* SAMPLING_POLICY_TOP_LEFT */ + t[0] = in_x_coords * scale.x; + t[1] = vec4(coord.y) * scale.y; +#else /* Unsupported sampling policy */ +#error Unsupported sampling policy +#endif return t; } -- cgit v1.2.1