diff options
author | 장지섭/On-Device Lab(SR)/Engineer/삼성전자 <jiseob.jang@samsung.com> | 2019-01-24 08:44:12 +0900 |
---|---|---|
committer | 오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com> | 2019-01-24 08:44:12 +0900 |
commit | b0867f5971185354e9cdc62e98a61742c9d55e26 (patch) | |
tree | f78a14efbaacd09721e9fcd91d4d5f022b3ba30d /runtimes/pure_arm_compute | |
parent | 0f812d444edb37d399e371a1a03476cb061f6465 (diff) | |
download | nnfw-b0867f5971185354e9cdc62e98a61742c9d55e26.tar.gz nnfw-b0867f5971185354e9cdc62e98a61742c9d55e26.tar.bz2 nnfw-b0867f5971185354e9cdc62e98a61742c9d55e26.zip |
[PACL] Replace CLLogicalNot with CLBitwiseNot (#4299)
This commit replaces CLLogicalNot of ARMComputeEx with CLBitwiseNot of ARMCompute
Signed-off-by: jiseob.jang <jiseob.jang@samsung.com>
Diffstat (limited to 'runtimes/pure_arm_compute')
-rw-r--r-- | runtimes/pure_arm_compute/src/compilation.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index 590ba9152..d32487fae 100644 --- a/runtimes/pure_arm_compute/src/compilation.cc +++ b/runtimes/pure_arm_compute/src/compilation.cc @@ -33,6 +33,7 @@ #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/CLBitwiseNot.h> #include <arm_compute/runtime/CL/functions/CLPadLayerEx.h> #include <arm_compute/runtime/CL/functions/CLPixelWiseMultiplication.h> #include <arm_compute/runtime/CL/functions/CLPoolingLayer.h> @@ -68,7 +69,6 @@ #include <arm_compute/runtime/CL/functions/CLPReLU.h> #include <arm_compute/runtime/CL/functions/CLBinaryLogicalOp.h> #include <arm_compute/runtime/CL/functions/CLComparisonOp.h> -#include <arm_compute/runtime/CL/functions/CLLogicalNot.h> #include <arm_compute/runtime/SubTensor.h> #include <arm_compute/runtime/NEON/functions/NESoftmaxLayer.h> @@ -5812,13 +5812,13 @@ void Planner::visit(const ::internal::tflite::op::LogicalNot::Node &node) // Set Shape Constraints and TensorInfo _builder.addShapeConstr(output_index, - asTensorInfo(asTensorShape(_ctx.at(output_index).shape(), false), - _ctx.at(output_index).type(), _ctx.at(output_index).scale(), + asTensorInfo(asTensorShape(_ctx.at(output_index).shape()), + ::arm_compute::DataType::U8, _ctx.at(output_index).scale(), _ctx.at(output_index).zeroPoint())); _builder.addShapeConstr(input_index, - asTensorInfo(asTensorShape(_ctx.at(input_index).shape(), false), - _ctx.at(input_index).type(), _ctx.at(input_index).scale(), + asTensorInfo(asTensorShape(_ctx.at(input_index).shape()), + ::arm_compute::DataType::U8, _ctx.at(input_index).scale(), _ctx.at(input_index).zeroPoint())); // Construct operation parameters @@ -5837,7 +5837,7 @@ void Planner::visit(const ::internal::tflite::op::LogicalNot::Node &node) auto input_alloc = ctx.at(::internal::tflite::operand::Index{param.input_index}); if (::internal::arm_compute::isGpuMode()) { - auto fn = nnfw::cpp14::make_unique<::arm_compute::CLLogicalNot>(); + auto fn = nnfw::cpp14::make_unique<::arm_compute::CLBitwiseNot>(); fn->configure(CAST_CL(input_alloc), CAST_CL(output_alloc)); |