diff options
author | 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com> | 2019-09-16 15:52:34 +0900 |
---|---|---|
committer | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-16 15:52:34 +0900 |
commit | 9452de4c1f6bf4e89d2ade9a4caac92d3bf3243b (patch) | |
tree | 38a2a01df7b4e209cb4463092f1a902088262d71 | |
parent | 885ac6f2422c0ac6950422cfbf18c67d3ab1b0f5 (diff) | |
download | nnfw-9452de4c1f6bf4e89d2ade9a4caac92d3bf3243b.tar.gz nnfw-9452de4c1f6bf4e89d2ade9a4caac92d3bf3243b.tar.bz2 nnfw-9452de4c1f6bf4e89d2ade9a4caac92d3bf3243b.zip |
[exo-tflite] Adding AvgPool2DConverter into convert.cpp (#7389)
AvgPool2DConverter is added into Convert.cpp. After this, turning on the knob will convert loco::AvgPool2D to TFLAveragePool2D.
Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
-rw-r--r-- | compiler/exo-tflite/src/Conversions.h | 2 | ||||
-rw-r--r-- | compiler/exo-tflite/src/Convert.cpp | 5 |
2 files changed, 5 insertions, 2 deletions
diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h index 804f9c690..54c858459 100644 --- a/compiler/exo-tflite/src/Conversions.h +++ b/compiler/exo-tflite/src/Conversions.h @@ -17,7 +17,7 @@ #ifndef __CONVERSIONS_H__ #define __CONVERSIONS_H__ -// TODO loco::AvgPool2D +#include "Conversion/AvgPool2DConverter.h" // TODO loco::ConstGen // TODO loco::Conv2D // TODO loco::DepthwiseConv2D diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp index 927d16410..6f5ffe8d4 100644 --- a/compiler/exo-tflite/src/Convert.cpp +++ b/compiler/exo-tflite/src/Convert.cpp @@ -54,7 +54,10 @@ void convert_to_TFLNodes(loco::Graph *graph) phase.emplace_back(stdex::make_unique<ShapeInferencePass>()); // Add converters for canonical nodes. Note: Not all loco canonical nodes are listed. - // TODO loco::AvgPool2D + + if (get_knob<Knob::ConvertAvgPool2D>()) + phase.emplace_back(stdex::make_unique<AvgPool2DConverter>()); + // TODO loco::ConstGen // TODO loco::Conv2D // TODO loco::DepthwiseConv2D |