From ccb5d2fcd5d5325190a80cae35316a51e3cbd619 Mon Sep 17 00:00:00 2001 From: xuanyue Date: Tue, 4 Aug 2020 15:10:59 +0800 Subject: [PATCH] tflite_util --- .../converter/parser/tflite/tflite_util.cc | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/mindspore/lite/tools/converter/parser/tflite/tflite_util.cc b/mindspore/lite/tools/converter/parser/tflite/tflite_util.cc index 7da8fb85d91..0ead307e90c 100644 --- a/mindspore/lite/tools/converter/parser/tflite/tflite_util.cc +++ b/mindspore/lite/tools/converter/parser/tflite/tflite_util.cc @@ -103,6 +103,27 @@ std::map tfMsOpTypeMap{ {tflite::BuiltinOperator_SCATTER_ND, "ScatterNd"}, {tflite::BuiltinOperator_MAXIMUM, "Maximum"}, {tflite::BuiltinOperator_MINIMUM, "Minimum"}, + {tflite::BuiltinOperator_ADD_N, "AddN"}, + {tflite::BuiltinOperator_CAST, "Cast"}, + {tflite::BuiltinOperator_EQUAL, "Equal"}, + {tflite::BuiltinOperator_NOT_EQUAL, "NotEqual"}, + {tflite::BuiltinOperator_GREATER, "Greater"}, + {tflite::BuiltinOperator_GREATER_EQUAL, "GreaterEqual"}, + {tflite::BuiltinOperator_LESS, "Less"}, + {tflite::BuiltinOperator_LESS_EQUAL, "LessEqual"}, + {tflite::BuiltinOperator_DEPTH_TO_SPACE, "DepthToSpace"}, + {tflite::BuiltinOperator_SPACE_TO_BATCH_ND, "SpaceToBatchND"}, + {tflite::BuiltinOperator_SPACE_TO_DEPTH, "SpaceToDepth"}, + {tflite::BuiltinOperator_PRELU, "Prelu"}, + {tflite::BuiltinOperator_ROUND, "Round"}, + {tflite::BuiltinOperator_WHERE, "Where"}, + {tflite::BuiltinOperator_SPARSE_TO_DENSE, "SparseToDense"}, + {tflite::BuiltinOperator_ZEROS_LIKE, "ZerosLike"}, + {tflite::BuiltinOperator_TILE, "Tile"}, + {tflite::BuiltinOperator_TOPK_V2, "TopKV2"}, + {tflite::BuiltinOperator_REVERSE_SEQUENCE, "ReverseSequence"}, + {tflite::BuiltinOperator_UNIQUE, "Unique"}, + {tflite::BuiltinOperator_UNPACK, "Unstack"}, }; std::string GetMSOpType(tflite::BuiltinOperator tfliteOpType) {