From 992ec681759fa6f5818bf4ff75b5f7f8740c840f Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=98=A4=ED=98=95=EC=84=9D/On-Device=20Lab=28SR=29/Staff?= =?utf8?q?=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 5 Sep 2019 20:16:03 +0900 Subject: [PATCH] Use type convert helper in tflite loader (#7232) Use activation type convert helper for sub operation Signed-off-by: Hyeongseok Oh --- runtimes/neurun/frontend/tflite/loader.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtimes/neurun/frontend/tflite/loader.cc b/runtimes/neurun/frontend/tflite/loader.cc index 6511eed..58a2a30 100644 --- a/runtimes/neurun/frontend/tflite/loader.cc +++ b/runtimes/neurun/frontend/tflite/loader.cc @@ -341,7 +341,7 @@ void Loader::loadSub(const tflite::Operator *op) model::operation::SubNode::Param param; const auto *options = op->builtin_options_as_SubOptions(); - param.activation = neurun::model::Activation(options->fused_activation_function()); + param.activation = convertActivation(options->fused_activation_function()); std::unique_ptr new_op(new model::operation::SubNode(inputs, outputs, param)); _graph.addOperation(std::move(new_op)); -- 2.7.4