From 048dc86ea2a612a9443e5badb8fb8e5c18ac34df Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=82=A8=EA=B6=81=EC=84=9D/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 23 Aug 2019 10:04:30 +0900 Subject: [PATCH] [exo-tflite] Add Tanh visit fuction (#6847) This commit will add `Tanh` visit function Signed-off-by: Seok NamKoong --- compiler/exo-tflite/src/OperationExporter.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/compiler/exo-tflite/src/OperationExporter.cpp b/compiler/exo-tflite/src/OperationExporter.cpp index c322f65..c1092fe 100644 --- a/compiler/exo-tflite/src/OperationExporter.cpp +++ b/compiler/exo-tflite/src/OperationExporter.cpp @@ -41,6 +41,7 @@ public: public: void visit(loco::ReLU *) final; void visit(loco::ReLU6 *) final; + void visit(loco::Tanh *) final; void visit(loco::Push *) final { /* DO NOTHING */} void visit(loco::Pull *) final { /* DO NOTHING */} void visit(loco::FeatureEncode *) final; @@ -93,6 +94,17 @@ void OperationExporter::visit(loco::ReLU6 *node) gd._operators.push_back(op_offset); } +void OperationExporter::visit(loco::Tanh *node) +{ + uint32_t op_idx = gd.registerBuiltinOpcode(tflite::BuiltinOperator_TANH); + std::vector inputs_vec{get_tensor_index(node->input())}; + std::vector outputs_vec{get_tensor_index(static_cast(node))}; + auto inputs = builder.CreateVector(inputs_vec); + auto outputs = builder.CreateVector(outputs_vec); + auto op_offset = CreateOperator(builder, op_idx, inputs, outputs); + gd._operators.push_back(op_offset); +} + void OperationExporter::visit(loco::MaxPool2D *node) { uint32_t op_idx = gd.registerBuiltinOpcode(tflite::BuiltinOperator_MAX_POOL_2D); -- 2.7.4