From cfb51101f5bec9bfbe648cde7092bec46d6f6e13 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Siva=20Sai=20Vaddipati/System=20SW=20/SRI-Bangalore/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 12 Oct 2018 12:55:32 +0530 Subject: [PATCH] Adding Scale, Overflow policy and Rounding policy to Div op (#3086) Same as Mul op which uses the following values. Signed-off-by: Siva Sai --- runtimes/pure_arm_compute/src/compilation.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index 827f697..39d226b 100644 --- a/runtimes/pure_arm_compute/src/compilation.cc +++ b/runtimes/pure_arm_compute/src/compilation.cc @@ -836,9 +836,9 @@ void Planner::visit(const ::internal::tflite::op::Div::Node &node) { auto fn = nnfw::make_unique<::arm_compute::CLPixelWiseDivision>(); - // TODO Decide scale, overflow_policy, and rounding_policy. - // Currently, the default values are used. - fn->configure(CAST_CL(lhs_alloc), CAST_CL(rhs_alloc), CAST_CL(ofm_alloc)); + 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); builder.append("Div", std::move(fn)); } -- 2.7.4