summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
author윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>2019-09-16 14:25:37 +0900
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>2019-09-16 14:25:37 +0900
commitde7d2d318b049d5797c597244b2a4e22a1c65a5e (patch)
tree935cb7df5b90a5c71b19d92eee8f1c76b683c22f
parent2bae7c172fc88e9ddcec3f3004064c1bed5597e0 (diff)
downloadnnfw-de7d2d318b049d5797c597244b2a4e22a1c65a5e.tar.gz
nnfw-de7d2d318b049d5797c597244b2a4e22a1c65a5e.tar.bz2
nnfw-de7d2d318b049d5797c597244b2a4e22a1c65a5e.zip
[exo-tflite] Adding optimization passes (#7402)
This adds SimplifyDomainConversionPass, RemoveForwardNodePass, and RemoveDeadNodePass into Convert.cpp. Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
-rw-r--r--compiler/exo-tflite/src/Convert.cpp9
1 files changed, 8 insertions, 1 deletions
diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp
index 11bdd9f9b..927d16410 100644
--- a/compiler/exo-tflite/src/Convert.cpp
+++ b/compiler/exo-tflite/src/Convert.cpp
@@ -28,6 +28,10 @@
#include <loco/Service/CanonicalShapeInferenceRule.h>
#include <loco/Service/TypeInference.h>
+#include <logo/SimplifyDomainConversionPass.h>
+#include <logo/RemoveDeadNodePass.h>
+#include <logo/RemoveForwardNodePass.h>
+
#include <logo/Phase.h>
#include <stdex/Memory.h>
@@ -75,7 +79,10 @@ void convert_to_TFLNodes(loco::Graph *graph)
// TODO loco::TensorBiasAdd
// TODO loco::TensorSoftmax
- // TODO Add optimization passes
+ // Add optimization below
+ phase.emplace_back(stdex::make_unique<logo::SimplifyDomainConversionPass>());
+ phase.emplace_back(stdex::make_unique<logo::RemoveForwardNodePass>());
+ phase.emplace_back(stdex::make_unique<logo::RemoveDeadNodePass>());
}
logo::PhaseRunner<logo::PhaseStrategy::Saturate> phase_runner{graph};