diff options
author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-17 09:40:18 +0900 |
---|---|---|
committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-09-17 09:40:18 +0900 |
commit | 019a0cd293b2c698c6ca3c1085450b6e820a43f8 (patch) | |
tree | a4715c0ba6aa80156ac736b1c3f8fcdea6c1f2ee | |
parent | 93d85e1d258772b6fab86946bc1c4cfe60f18458 (diff) | |
download | nnfw-019a0cd293b2c698c6ca3c1085450b6e820a43f8.tar.gz nnfw-019a0cd293b2c698c6ca3c1085450b6e820a43f8.tar.bz2 nnfw-019a0cd293b2c698c6ca3c1085450b6e820a43f8.zip |
[exo-tflite] Add EltwiseMul and TensorBroadcast to Convert list (#7474)
This will update conversion list to convert EltwiseMul and TensorBroadcast node
Signed-off-by: SaeHie Park <saehie.park@samsung.com>
-rw-r--r-- | compiler/exo-tflite/src/Conversions.h | 3 | ||||
-rw-r--r-- | compiler/exo-tflite/src/Convert.cpp | 7 |
2 files changed, 8 insertions, 2 deletions
diff --git a/compiler/exo-tflite/src/Conversions.h b/compiler/exo-tflite/src/Conversions.h index 26a62584d..8f2e428b2 100644 --- a/compiler/exo-tflite/src/Conversions.h +++ b/compiler/exo-tflite/src/Conversions.h @@ -24,7 +24,7 @@ // TODO loco::DepthwiseFilterEncode #include "Conversion/EltwiseAddConverter.h" // TODO loco::EltwiseDiv -// TODO loco::EltwiseMul +#include "Conversion/EltwiseMulConverter.h" // TODO loco::EltwiseSqrt // TODO loco::EltwiseSub // TODO loco::FeatureBiasAdd @@ -35,6 +35,7 @@ // TODO loco::Tanh // TODO loco::TensorConcat // TODO loco::TensorBiasAdd +#include "Conversion/TensorBroadcastConverter.h" // TODO loco::TensorSoftmax #endif // __CONVERSIONS_H__ diff --git a/compiler/exo-tflite/src/Convert.cpp b/compiler/exo-tflite/src/Convert.cpp index 267cffaff..20035ffe9 100644 --- a/compiler/exo-tflite/src/Convert.cpp +++ b/compiler/exo-tflite/src/Convert.cpp @@ -66,7 +66,9 @@ void convert_to_TFLNodes(loco::Graph *graph) phase.emplace_back(stdex::make_unique<EltwiseAddConverter>()); // TODO loco::EltwiseDiv - // TODO loco::EltwiseMul + if (get_knob<Knob::ConvertEltwiseMul>()) + phase.emplace_back(stdex::make_unique<EltwiseMulConverter>()); + // TODO loco::EltwiseSqrt // TODO loco::EltwiseSub // TODO loco::FeatureBiasAdd @@ -82,6 +84,9 @@ void convert_to_TFLNodes(loco::Graph *graph) // TODO loco::Tanh // TODO loco::TensorConcat // TODO loco::TensorBiasAdd + if (get_knob<Knob::ConvertTensorBroadcast>()) + phase.emplace_back(stdex::make_unique<TensorBroadcastConverter>()); + // TODO loco::TensorSoftmax // Add optimization below |