From 4ff5a1c0fd58528f36895b59514bba3ea079ba1f Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 14 Aug 2019 10:53:42 +0900 Subject: [PATCH] [exo-tflite] Export FixedReshape operation (#6565) This commit introduces export of FixedReshape operation Signed-off-by: Cheongyo Bahk --- compiler/exo-tflite/src/OperationExporter.cpp | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/compiler/exo-tflite/src/OperationExporter.cpp b/compiler/exo-tflite/src/OperationExporter.cpp index 7f9c07f..d799e46 100644 --- a/compiler/exo-tflite/src/OperationExporter.cpp +++ b/compiler/exo-tflite/src/OperationExporter.cpp @@ -60,6 +60,7 @@ public: void visit(loco::EltwiseMul *) final; void visit(loco::EltwiseSub *) final; void visit(loco::EltwiseDiv *) final; + void visit(loco::FixedReshape *) final; void visit(locoex::COpCall *); @@ -427,6 +428,18 @@ void OperationExporter::visit(loco::EltwiseDiv *node) gd._operators.push_back(op_offset); } +void OperationExporter::visit(loco::FixedReshape *node) +{ + std::vector new_shape_vec; + for (uint32_t axis = 0; axis < node->rank(); ++axis) + { + assert(node->dim(axis).known()); + new_shape_vec.push_back(node->dim(axis).value()); + } + + exportAsReshape(node, builder, new_shape_vec, gd); +} + inline flatbuffers::Offset> CreateCOpCallOptions(flatbuffers::FlatBufferBuilder &fbb, locoex::COpCall *copCall) { -- 2.7.4