[PACL] Replace CLPixelWiseDivision with CLArithmeticDivision (#4302)
author장지섭/On-Device Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com>
Wed, 23 Jan 2019 07:35:21 +0000 (16:35 +0900)
committer오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Wed, 23 Jan 2019 07:35:21 +0000 (16:35 +0900)
This commit replaces CLPixelWiseDivision of ARMComputeEx with CLArithmeticDivision of ARMCompute.

Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
runtimes/pure_arm_compute/src/compilation.cc

index c48121e..590ba91 100644 (file)
@@ -32,9 +32,9 @@
 #include <arm_compute/runtime/CL/CLSubTensor.h>
 #include <arm_compute/runtime/CL/functions/CLArithmeticAddition.h>
 #include <arm_compute/runtime/CL/functions/CLArithmeticSubtraction.h>
+#include <arm_compute/runtime/CL/functions/CLArithmeticDivision.h>
 #include <arm_compute/runtime/CL/functions/CLPadLayerEx.h>
 #include <arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h>
-#include <arm_compute/runtime/CL/functions/CLPixelWiseDivision.h>
 #include <arm_compute/runtime/CL/functions/CLPoolingLayer.h>
 #include <arm_compute/runtime/CL/functions/CLActivationLayer.h>
 #include <arm_compute/runtime/CL/functions/CLActivationLayerEx.h>
@@ -905,11 +905,9 @@ void Planner::visit(const ::internal::tflite::op::Div::Node &node)
 
     if (::internal::arm_compute::isGpuMode())
     {
-      auto fn = nnfw::cpp14::make_unique<::arm_compute::CLPixelWiseDivision>();
+      auto fn = nnfw::cpp14::make_unique<::arm_compute::CLArithmeticDivision>();
 
-      fn->configure(CAST_CL(lhs_alloc), CAST_CL(rhs_alloc), CAST_CL(ofm_alloc),
-                    1.0, // scale
-                    arm_compute::ConvertPolicy::SATURATE, arm_compute::RoundingPolicy::TO_ZERO);
+      fn->configure(CAST_CL(lhs_alloc), CAST_CL(rhs_alloc), CAST_CL(ofm_alloc));
 
       builder.append("Div", std::move(fn));
     }