From d95095d7e95850139d521ac9ddfd4fd229759fc4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 16 Aug 2019 12:41:45 +0900 Subject: [PATCH] [exo-tflite] Enable EltwiseSqrt (#6610) * [exo-tflite] Enable EltwiseSqrt This will enable export for EltwiseSqrt node Signed-off-by: SaeHie Park * remove for shapeinf * remove not related new line --- 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 533d2a0..916179c 100644 --- a/compiler/exo-tflite/src/OperationExporter.cpp +++ b/compiler/exo-tflite/src/OperationExporter.cpp @@ -59,6 +59,7 @@ public: void visit(loco::EltwiseMul *) final; void visit(loco::EltwiseSub *) final; void visit(loco::EltwiseDiv *) final; + void visit(loco::EltwiseSqrt *) final; void visit(loco::FixedReshape *) final; void visit(locoex::COpCall *); @@ -427,6 +428,17 @@ void OperationExporter::visit(loco::EltwiseDiv *node) gd._operators.push_back(op_offset); } +void OperationExporter::visit(loco::EltwiseSqrt *node) +{ + uint32_t op_idx = gd.registerBuiltinOpcode(tflite::BuiltinOperator_SQRT); + 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::FixedReshape *node) { std::vector new_shape_vec; -- 2.7.4