diff options
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 |
commit | de7d2d318b049d5797c597244b2a4e22a1c65a5e (patch) | |
tree | 935cb7df5b90a5c71b19d92eee8f1c76b683c22f | |
parent | 2bae7c172fc88e9ddcec3f3004064c1bed5597e0 (diff) | |
download | nnfw-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.cpp | 9 |
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}; |