Removing unncessary CAST_NE from SquaredDifference Op (#2723)
authorPrasanna R/System SW /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Wed, 19 Sep 2018 06:58:11 +0000 (12:28 +0530)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 19 Sep 2018 06:58:11 +0000 (15:58 +0900)
This patch removes unncessary CAST_NE from SquaredDifference Op, Since the configure functions of NEON Kernels expects parameters of type ITensor and the parameters input, output are all of type ITensor.

Signed-off-by: prasannar <prasanna.r@samsung.com>
runtimes/pure_arm_compute/src/internal/layers/SquaredDifferenceOperation.cc

index 3f988a8..2b670bd 100644 (file)
@@ -19,9 +19,8 @@ void SquaredDifferenceOperation::configure(::arm_compute::ITensor *input1,
   }
   else
   {
-    _neon_sub.configure(CAST_NE(input1), CAST_NE(input2), CAST_NE(output), ConvertPolicy);
-    _neon_mul.configure(CAST_NE(output), CAST_NE(output), CAST_NE(output), scale, ConvertPolicy,
-                        RoundingPolicy);
+    _neon_sub.configure(input1, input2, output, ConvertPolicy);
+    _neon_mul.configure(output, output, output, scale, ConvertPolicy, RoundingPolicy);
   }
 }