diff options
author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-16 17:15:37 +0900 |
---|---|---|
committer | 박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com> | 2019-09-16 17:15:37 +0900 |
commit | d610f092ec3d9f64162f69e4a6e93acdf687adf6 (patch) | |
tree | f9664a56d4b5141a7a0203a4ce3d0251dfda7a9e /compiler/exo-tflite | |
parent | 9a180abf363d24cb78e64d62db75db59e3c72dc5 (diff) | |
download | nnfw-d610f092ec3d9f64162f69e4a6e93acdf687adf6.tar.gz nnfw-d610f092ec3d9f64162f69e4a6e93acdf687adf6.tar.bz2 nnfw-d610f092ec3d9f64162f69e4a6e93acdf687adf6.zip |
[exo-tflite] Add EltwiseConverter to conversion list (#7463)
This will add EltwiseConverter to conversion list from knob status
Signed-off-by: SaeHie Park <saehie.park@samsung.com>
Diffstat (limited to 'compiler/exo-tflite')
-rw-r--r-- | compiler/exo-tflite/src/Conversions.h | 2 | ||||
-rw-r--r-- | compiler/exo-tflite/src/Convert.cpp | 4 |
2 files changed, 4 insertions, 2 deletions
diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h index 54c858459..26a62584d 100644 --- a/compiler/exo-tflite/src/Conversions.h +++ b/compiler/exo-tflite/src/Conversions.h @@ -22,7 +22,7 @@ // TODO loco::Conv2D // TODO loco::DepthwiseConv2D // TODO loco::DepthwiseFilterEncode -// TODO loco::EltwiseAdd +#include "Conversion/EltwiseAddConverter.h" // TODO loco::EltwiseDiv // TODO loco::EltwiseMul // TODO loco::EltwiseSqrt diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp index 6f5ffe8d4..267cffaff 100644 --- a/compiler/exo-tflite/src/Convert.cpp +++ b/compiler/exo-tflite/src/Convert.cpp @@ -62,7 +62,9 @@ void convert_to_TFLNodes(loco::Graph *graph) // TODO loco::Conv2D // TODO loco::DepthwiseConv2D // TODO loco::DepthwiseFilterEncode - // TODO loco::EltwiseAdd + if (get_knob<Knob::ConvertEltwiseAdd>()) + phase.emplace_back(stdex::make_unique<EltwiseAddConverter>()); + // TODO loco::EltwiseDiv // TODO loco::EltwiseMul // TODO loco::EltwiseSqrt |