diff options
author | Hyeongseok Oh <hseok82.oh@samsung.com> | 2023-04-12 15:42:02 +0900 |
---|---|---|
committer | Hyeongseok Oh <hseok82.oh@samsung.com> | 2023-04-12 15:42:02 +0900 |
commit | 323663bb115ef625642391a5a8e9b35fee8b2ae3 (patch) | |
tree | 17e2a6b91535e6f53f4cacda5e4db6aa0303dd22 /compiler/tflite2circle/src | |
parent | c690d52bdd137ed6a17353aa7af35e8141ece77b (diff) | |
download | nnfw-323663bb115ef625642391a5a8e9b35fee8b2ae3.tar.gz nnfw-323663bb115ef625642391a5a8e9b35fee8b2ae3.tar.bz2 nnfw-323663bb115ef625642391a5a8e9b35fee8b2ae3.zip |
Imported Upstream version 1.22.0upstream/1.22.0
Diffstat (limited to 'compiler/tflite2circle/src')
-rw-r--r-- | compiler/tflite2circle/src/DataLookup.cpp | 11 | ||||
-rw-r--r-- | compiler/tflite2circle/src/DataLookup.h | 1 | ||||
-rw-r--r-- | compiler/tflite2circle/src/TFLOperator.lst | 20 |
3 files changed, 30 insertions, 2 deletions
diff --git a/compiler/tflite2circle/src/DataLookup.cpp b/compiler/tflite2circle/src/DataLookup.cpp index 7c3aab089..c16e60196 100644 --- a/compiler/tflite2circle/src/DataLookup.cpp +++ b/compiler/tflite2circle/src/DataLookup.cpp @@ -36,15 +36,22 @@ circle::BuiltinOperator get_circle_builtin_code(tflite::BuiltinOperator tfl_bop) int8_t get_circle_builtin_code(int8_t tfl_bop_i8) { - tflite::BuiltinOperator tfl_bop = static_cast<tflite::BuiltinOperator>(tfl_bop_i8); + return get_circle_builtin_code(static_cast<int32_t>(tfl_bop_i8)); +} + +int32_t get_circle_builtin_code(int32_t tfl_bop_i32) +{ + tflite::BuiltinOperator tfl_bop = static_cast<tflite::BuiltinOperator>(tfl_bop_i32); switch (tfl_bop) { #define TFL_OPERATOR(OP) \ case tflite::BuiltinOperator_##OP: \ - return static_cast<int8_t>(circle::BuiltinOperator_##OP); + return static_cast<int32_t>(circle::BuiltinOperator_##OP); #include "TFLOperator.lst" #undef TFL_OPERATOR + case tflite::BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES: + return static_cast<int32_t>(circle::BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES); default: throw std::runtime_error("tflite2circle: wrong op"); } diff --git a/compiler/tflite2circle/src/DataLookup.h b/compiler/tflite2circle/src/DataLookup.h index 5aeeb6eca..f346b01f4 100644 --- a/compiler/tflite2circle/src/DataLookup.h +++ b/compiler/tflite2circle/src/DataLookup.h @@ -31,6 +31,7 @@ namespace tflite2circle circle::BuiltinOperator get_circle_builtin_code(tflite::BuiltinOperator tfl_bop); int8_t get_circle_builtin_code(int8_t tfl_bop_i8); +int32_t get_circle_builtin_code(int32_t tfl_bop_i32); /** * @brief Returns circle TensorType according to tflite. diff --git a/compiler/tflite2circle/src/TFLOperator.lst b/compiler/tflite2circle/src/TFLOperator.lst index 942c846c7..72a29fce5 100644 --- a/compiler/tflite2circle/src/TFLOperator.lst +++ b/compiler/tflite2circle/src/TFLOperator.lst @@ -131,3 +131,23 @@ TFL_OPERATOR(SELECT_V2) TFL_OPERATOR(DENSIFY) TFL_OPERATOR(SEGMENT_SUM) TFL_OPERATOR(BATCH_MATMUL) +// PLACEHOLDER_FOR_GREATER_OP_CODES = 127 +TFL_OPERATOR(CUMSUM) +TFL_OPERATOR(CALL_ONCE) +TFL_OPERATOR(BROADCAST_TO) +TFL_OPERATOR(RFFT2D) +TFL_OPERATOR(CONV_3D) +TFL_OPERATOR(IMAG) +TFL_OPERATOR(REAL) +TFL_OPERATOR(COMPLEX_ABS) +TFL_OPERATOR(HASHTABLE) +TFL_OPERATOR(HASHTABLE_FIND) +TFL_OPERATOR(HASHTABLE_IMPORT) +TFL_OPERATOR(HASHTABLE_SIZE) +TFL_OPERATOR(REDUCE_ALL) +TFL_OPERATOR(CONV_3D_TRANSPOSE) +TFL_OPERATOR(VAR_HANDLE) +TFL_OPERATOR(READ_VARIABLE) +TFL_OPERATOR(ASSIGN_VARIABLE) +TFL_OPERATOR(BROADCAST_ARGS) +TFL_OPERATOR(RANDOM_STANDARD_NORMAL) |