From 3c884031c00ee99dbc54162635bc2898e14757d8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Staff=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Wed, 1 Aug 2018 17:11:37 +0900 Subject: [PATCH] Bug fix: support strided slice (#2136) Support strided slice operator STRIDED_SLICE_EX is removed Signed-off-by: Hyeongseok Oh --- runtimes/pure_arm_compute/src/model.cc | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/runtimes/pure_arm_compute/src/model.cc b/runtimes/pure_arm_compute/src/model.cc index a161443..2400754 100644 --- a/runtimes/pure_arm_compute/src/model.cc +++ b/runtimes/pure_arm_compute/src/model.cc @@ -392,6 +392,18 @@ int ANeuralNetworksModel_addOperation(ANeuralNetworksModel *model, break; } + case ANEURALNETWORKS_STRIDED_SLICE: + { + using internal::tflite::op::StridedSlice::Param; + using internal::tflite::op::StridedSlice::Node; + + // Add 'operations' + auto &operations = model->deref().operations(); + + operations.emplace_back(Param{inputCount, inputs, outputCount, outputs}); + + break; + } default: throw std::runtime_error{"Not supported operation"}; }; -- 2.7.4