From f525e33b07759e983110786708cae1498156c599 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: Mon, 23 Sep 2019 10:32:23 +0900 Subject: [PATCH] [exo-tflite] Enable Div,Sub conversion (#7669) This will enable Eltwise Div/Sub node conversion to TFL node by knob Signed-off-by: SaeHie Park --- compiler/exo-tflite/src/Conversions.h | 4 ++-- compiler/exo-tflite/src/Convert.cpp | 8 ++++++-- 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h index 6e88063..e9e3447 100644 --- a/compiler/exo-tflite/src/Conversions.h +++ b/compiler/exo-tflite/src/Conversions.h @@ -23,10 +23,10 @@ // TODO loco::DepthwiseConv2D // TODO loco::DepthwiseFilterEncode #include "Conversion/EltwiseAddConverter.h" -// TODO loco::EltwiseDiv +#include "Conversion/EltwiseDivConverter.h" #include "Conversion/EltwiseMulConverter.h" // TODO loco::EltwiseSqrt -// TODO loco::EltwiseSub +#include "Conversion/EltwiseSubConverter.h" #include "Conversion/FeatureBiasAddConverter.h" // TODO loco::FixedReshape #include "Conversion/MaxPool2DConverter.h" diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp index 516227d..9883c6a 100644 --- a/compiler/exo-tflite/src/Convert.cpp +++ b/compiler/exo-tflite/src/Convert.cpp @@ -67,12 +67,16 @@ void convert_to_TFLNodes(loco::Graph *graph) if (get()) phase.emplace_back(stdex::make_unique()); - // TODO loco::EltwiseDiv + if (get()) + phase.emplace_back(stdex::make_unique()); + if (get()) phase.emplace_back(stdex::make_unique()); // TODO loco::EltwiseSqrt - // TODO loco::EltwiseSub + + if (get()) + phase.emplace_back(stdex::make_unique()); if (get()) phase.emplace_back(stdex::make_unique()); -- 2.7.4