|
|
|
@@ -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) { |
|
|
|
|