p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tflite_util

This commit is contained in:
xuanyue 2020-08-04 15:10:59 +08:00
parent ad2560bde6
commit ccb5d2fcd5
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mindspore/lite/tools/converter/parser/tflite/tflite_util.cc
View File

@ -103,6 +103,27 @@ std::map<tflite::BuiltinOperator, std::string> 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) {