Browse Source

codedex

pull/15628/head
yefeng 4 years ago
parent
commit
6a8e5adbf3
60 changed files with 190 additions and 0 deletions
  1. +14
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.h
  2. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_addn_parser.h
  3. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_argmax_parser.h
  4. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_argmin_parser.h
  5. +56
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_arithmetic_parser.h
  6. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_batch_to_space_parser.h
  7. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_broadcast_to_parser.h
  8. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_cast_parser.h
  9. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_concat_parser.h
  10. +4
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_conv_parser.h
  11. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_conv_transpose_parser.h
  12. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_custom_parser.h
  13. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.h
  14. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_dequantize_parser.h
  15. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_expand_dims_parser.h
  16. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_fill_parser.h
  17. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_fullyconnected_parser.h
  18. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_gather_nd_parser.h
  19. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_gather_parser.h
  20. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_hashtable_lookup_parser.h
  21. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_l2norm_parser.h
  22. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h
  23. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_lrn_parser.h
  24. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_lsh_projection_parser.h
  25. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_matmul_parser.h
  26. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_node_parser_registry.h
  27. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_one_hot_parser.h
  28. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_pad_parser.h
  29. +4
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_pooling_parser.h
  30. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_quantize_parser.h
  31. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_range_parser.h
  32. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_rank_parser.h
  33. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_reduce_parser.h
  34. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_reshape_parser.h
  35. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_resize_parser.h
  36. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_reverse_parser.h
  37. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_reverse_sequence_parser.h
  38. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_scatter_nd_parser.h
  39. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_shape_parser.h
  40. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_skip_gram_parser.h
  41. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_slice_parser.h
  42. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_softmax_parser.h
  43. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_space_to_batch_nd_parser.h
  44. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_space_to_depth_parser.h
  45. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_sparse_to_dense_parser.h
  46. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_split_parser.h
  47. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_split_v_parser.h
  48. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_squeeze_parser.h
  49. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_stack_parser.h
  50. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_strided_slice_parser.h
  51. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_tile_parser.h
  52. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.h
  53. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_transpose_parser.h
  54. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_unique_parser.h
  55. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_unstack_parser.h
  56. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_where_parser.h
  57. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_while_parser.h
  58. +2
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_zeros_like_parser.h
  59. +2
    -0
      mindspore/lite/tools/optimizer/graph/functionalize_cond.h
  60. +2
    -0
      mindspore/lite/tools/optimizer/graph/functionalize_while.h

+ 14
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.h View File

@@ -29,6 +29,8 @@ class TfliteReluParser : public TfliteNodeParser {
public: public:
TfliteReluParser() : TfliteNodeParser("Relu") {} TfliteReluParser() : TfliteNodeParser("Relu") {}


~TfliteReluParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -37,6 +39,8 @@ class TfliteRelu6Parser : public TfliteNodeParser {
public: public:
TfliteRelu6Parser() : TfliteNodeParser("Relu6") {} TfliteRelu6Parser() : TfliteNodeParser("Relu6") {}


~TfliteRelu6Parser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -45,6 +49,8 @@ class TfliteLeakyReluParser : public TfliteNodeParser {
public: public:
TfliteLeakyReluParser() : TfliteNodeParser("LeakyRelu") {} TfliteLeakyReluParser() : TfliteNodeParser("LeakyRelu") {}


~TfliteLeakyReluParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -53,6 +59,8 @@ class TflitePReLUParser : public TfliteNodeParser {
public: public:
TflitePReLUParser() : TfliteNodeParser("PReLU") {} TflitePReLUParser() : TfliteNodeParser("PReLU") {}


~TflitePReLUParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -61,6 +69,8 @@ class TfliteTanhParser : public TfliteNodeParser {
public: public:
TfliteTanhParser() : TfliteNodeParser("Tanh") {} TfliteTanhParser() : TfliteNodeParser("Tanh") {}


~TfliteTanhParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -69,6 +79,8 @@ class TfliteHardSwishParser : public TfliteNodeParser {
public: public:
TfliteHardSwishParser() : TfliteNodeParser("HardSwish") {} TfliteHardSwishParser() : TfliteNodeParser("HardSwish") {}


~TfliteHardSwishParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -77,6 +89,8 @@ class TfliteLogisticParser : public TfliteNodeParser {
public: public:
TfliteLogisticParser() : TfliteNodeParser("Logistic") {} TfliteLogisticParser() : TfliteNodeParser("Logistic") {}


~TfliteLogisticParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_addn_parser.h View File

@@ -29,6 +29,8 @@ class TfliteAddNParser : public TfliteNodeParser {
public: public:
TfliteAddNParser() : TfliteNodeParser("AddN") {} TfliteAddNParser() : TfliteNodeParser("AddN") {}


~TfliteAddNParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_argmax_parser.h View File

@@ -29,6 +29,8 @@ class TfliteArgmaxParser : public TfliteNodeParser {
public: public:
TfliteArgmaxParser() : TfliteNodeParser("Argmax") {} TfliteArgmaxParser() : TfliteNodeParser("Argmax") {}


~TfliteArgmaxParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_argmin_parser.h View File

@@ -29,6 +29,8 @@ class TfliteArgminParser : public TfliteNodeParser {
public: public:
TfliteArgminParser() : TfliteNodeParser("Argmin") {} TfliteArgminParser() : TfliteNodeParser("Argmin") {}


~TfliteArgminParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 56
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_arithmetic_parser.h View File

@@ -29,6 +29,8 @@ class TfliteAddParser : public TfliteNodeParser {
public: public:
TfliteAddParser() : TfliteNodeParser("Add") {} TfliteAddParser() : TfliteNodeParser("Add") {}


~TfliteAddParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -37,6 +39,8 @@ class TfliteSubParser : public TfliteNodeParser {
public: public:
TfliteSubParser() : TfliteNodeParser("Sub") {} TfliteSubParser() : TfliteNodeParser("Sub") {}


~TfliteSubParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -45,6 +49,8 @@ class TfliteMulParser : public TfliteNodeParser {
public: public:
TfliteMulParser() : TfliteNodeParser("Mul") {} TfliteMulParser() : TfliteNodeParser("Mul") {}


~TfliteMulParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -53,6 +59,8 @@ class TfliteDivParser : public TfliteNodeParser {
public: public:
TfliteDivParser() : TfliteNodeParser("Div") {} TfliteDivParser() : TfliteNodeParser("Div") {}


~TfliteDivParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -61,6 +69,8 @@ class TfliteFloorDivParser : public TfliteNodeParser {
public: public:
TfliteFloorDivParser() : TfliteNodeParser("FloorDiv") {} TfliteFloorDivParser() : TfliteNodeParser("FloorDiv") {}


~TfliteFloorDivParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -69,6 +79,8 @@ class TfliteFloorModParser : public TfliteNodeParser {
public: public:
TfliteFloorModParser() : TfliteNodeParser("FloorMod") {} TfliteFloorModParser() : TfliteNodeParser("FloorMod") {}


~TfliteFloorModParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -77,6 +89,8 @@ class TflitePowParser : public TfliteNodeParser {
public: public:
TflitePowParser() : TfliteNodeParser("PowFusion") {} TflitePowParser() : TfliteNodeParser("PowFusion") {}


~TflitePowParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -85,6 +99,8 @@ class TfliteSquaredDifferenceParser : public TfliteNodeParser {
public: public:
TfliteSquaredDifferenceParser() : TfliteNodeParser("SquaredDifference") {} TfliteSquaredDifferenceParser() : TfliteNodeParser("SquaredDifference") {}


~TfliteSquaredDifferenceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -93,6 +109,8 @@ class TfliteMaximumParser : public TfliteNodeParser {
public: public:
TfliteMaximumParser() : TfliteNodeParser("Maximum") {} TfliteMaximumParser() : TfliteNodeParser("Maximum") {}


~TfliteMaximumParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -101,6 +119,8 @@ class TfliteMinimumParser : public TfliteNodeParser {
public: public:
TfliteMinimumParser() : TfliteNodeParser("Minimum") {} TfliteMinimumParser() : TfliteNodeParser("Minimum") {}


~TfliteMinimumParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -109,6 +129,8 @@ class TfliteAbsParser : public TfliteNodeParser {
public: public:
TfliteAbsParser() : TfliteNodeParser("Abs") {} TfliteAbsParser() : TfliteNodeParser("Abs") {}


~TfliteAbsParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -117,6 +139,8 @@ class TfliteExpParser : public TfliteNodeParser {
public: public:
TfliteExpParser() : TfliteNodeParser("Exp") {} TfliteExpParser() : TfliteNodeParser("Exp") {}


~TfliteExpParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -125,6 +149,8 @@ class TfliteSqrtParser : public TfliteNodeParser {
public: public:
TfliteSqrtParser() : TfliteNodeParser("Sqrt") {} TfliteSqrtParser() : TfliteNodeParser("Sqrt") {}


~TfliteSqrtParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -133,6 +159,8 @@ class TfliteRsqrtParser : public TfliteNodeParser {
public: public:
TfliteRsqrtParser() : TfliteNodeParser("Rsqrt") {} TfliteRsqrtParser() : TfliteNodeParser("Rsqrt") {}


~TfliteRsqrtParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -141,6 +169,8 @@ class TfliteSquareParser : public TfliteNodeParser {
public: public:
TfliteSquareParser() : TfliteNodeParser("Square") {} TfliteSquareParser() : TfliteNodeParser("Square") {}


~TfliteSquareParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -149,6 +179,8 @@ class TfliteSinParser : public TfliteNodeParser {
public: public:
TfliteSinParser() : TfliteNodeParser("Sin") {} TfliteSinParser() : TfliteNodeParser("Sin") {}


~TfliteSinParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -157,6 +189,8 @@ class TfliteCosParser : public TfliteNodeParser {
public: public:
TfliteCosParser() : TfliteNodeParser("Cos") {} TfliteCosParser() : TfliteNodeParser("Cos") {}


~TfliteCosParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -165,6 +199,8 @@ class TfliteLogParser : public TfliteNodeParser {
public: public:
TfliteLogParser() : TfliteNodeParser("Log") {} TfliteLogParser() : TfliteNodeParser("Log") {}


~TfliteLogParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -173,6 +209,8 @@ class TfliteRoundParser : public TfliteNodeParser {
public: public:
TfliteRoundParser() : TfliteNodeParser("Round") {} TfliteRoundParser() : TfliteNodeParser("Round") {}


~TfliteRoundParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -181,6 +219,8 @@ class TfliteCeilParser : public TfliteNodeParser {
public: public:
TfliteCeilParser() : TfliteNodeParser("Ceil") {} TfliteCeilParser() : TfliteNodeParser("Ceil") {}


~TfliteCeilParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -189,6 +229,8 @@ class TfliteFloorParser : public TfliteNodeParser {
public: public:
TfliteFloorParser() : TfliteNodeParser("Floor") {} TfliteFloorParser() : TfliteNodeParser("Floor") {}


~TfliteFloorParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -197,6 +239,8 @@ class TfliteNegParser : public TfliteNodeParser {
public: public:
TfliteNegParser() : TfliteNodeParser("Neg") {} TfliteNegParser() : TfliteNodeParser("Neg") {}


~TfliteNegParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -205,6 +249,8 @@ class TfliteEqualParser : public TfliteNodeParser {
public: public:
TfliteEqualParser() : TfliteNodeParser("Equal") {} TfliteEqualParser() : TfliteNodeParser("Equal") {}


~TfliteEqualParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -213,6 +259,8 @@ class TfliteNotEqualParser : public TfliteNodeParser {
public: public:
TfliteNotEqualParser() : TfliteNodeParser("NotEqual") {} TfliteNotEqualParser() : TfliteNodeParser("NotEqual") {}


~TfliteNotEqualParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -221,6 +269,8 @@ class TfliteGreaterParser : public TfliteNodeParser {
public: public:
TfliteGreaterParser() : TfliteNodeParser("Greater") {} TfliteGreaterParser() : TfliteNodeParser("Greater") {}


~TfliteGreaterParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -229,6 +279,8 @@ class TfliteGreaterEqualParser : public TfliteNodeParser {
public: public:
TfliteGreaterEqualParser() : TfliteNodeParser("GreaterEqual") {} TfliteGreaterEqualParser() : TfliteNodeParser("GreaterEqual") {}


~TfliteGreaterEqualParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -237,6 +289,8 @@ class TfliteLessParser : public TfliteNodeParser {
public: public:
TfliteLessParser() : TfliteNodeParser("Less") {} TfliteLessParser() : TfliteNodeParser("Less") {}


~TfliteLessParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -245,6 +299,8 @@ class TfliteLessEqualParser : public TfliteNodeParser {
public: public:
TfliteLessEqualParser() : TfliteNodeParser("LessEqual") {} TfliteLessEqualParser() : TfliteNodeParser("LessEqual") {}


~TfliteLessEqualParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_batch_to_space_parser.h View File

@@ -29,6 +29,8 @@ class TfliteBatchToSpaceParser : public TfliteNodeParser {
public: public:
TfliteBatchToSpaceParser() : TfliteNodeParser("BatchToSpace") {} TfliteBatchToSpaceParser() : TfliteNodeParser("BatchToSpace") {}


~TfliteBatchToSpaceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_broadcast_to_parser.h View File

@@ -29,6 +29,8 @@ class TfliteBroadcastToParser : public TfliteNodeParser {
public: public:
TfliteBroadcastToParser() : TfliteNodeParser("BroadcastTo") {} TfliteBroadcastToParser() : TfliteNodeParser("BroadcastTo") {}


~TfliteBroadcastToParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_cast_parser.h View File

@@ -29,6 +29,8 @@ class TfliteCastParser : public TfliteNodeParser {
public: public:
TfliteCastParser() : TfliteNodeParser("Cast") {} TfliteCastParser() : TfliteNodeParser("Cast") {}


~TfliteCastParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_concat_parser.h View File

@@ -29,6 +29,8 @@ class TfliteConcatParser : public TfliteNodeParser {
public: public:
TfliteConcatParser() : TfliteNodeParser("Concat") {} TfliteConcatParser() : TfliteNodeParser("Concat") {}


~TfliteConcatParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 4
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_conv_parser.h View File

@@ -29,6 +29,8 @@ class TfliteConvParser : public TfliteNodeParser {
public: public:
TfliteConvParser() : TfliteNodeParser("Conv2D") {} TfliteConvParser() : TfliteNodeParser("Conv2D") {}


~TfliteConvParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -37,6 +39,8 @@ class TfliteDepthwiseConv2DParser : public TfliteNodeParser {
public: public:
TfliteDepthwiseConv2DParser() : TfliteNodeParser("DepthwiseConv2D") {} TfliteDepthwiseConv2DParser() : TfliteNodeParser("DepthwiseConv2D") {}


~TfliteDepthwiseConv2DParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_conv_transpose_parser.h View File

@@ -29,6 +29,8 @@ class TfliteDeConvParser : public TfliteNodeParser {
public: public:
TfliteDeConvParser() : TfliteNodeParser("DeConv2D") {} TfliteDeConvParser() : TfliteNodeParser("DeConv2D") {}


~TfliteDeConvParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_custom_parser.h View File

@@ -29,6 +29,8 @@ class TfliteCustomParser : public TfliteNodeParser {
public: public:
TfliteCustomParser() : TfliteNodeParser("Custom") {} TfliteCustomParser() : TfliteNodeParser("Custom") {}


~TfliteCustomParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;




+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.h View File

@@ -29,6 +29,8 @@ class TfliteDepthToSpaceParser : public TfliteNodeParser {
public: public:
TfliteDepthToSpaceParser() : TfliteNodeParser("DepthToSpace") {} TfliteDepthToSpaceParser() : TfliteNodeParser("DepthToSpace") {}


~TfliteDepthToSpaceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_dequantize_parser.h View File

@@ -28,6 +28,8 @@ class TfliteDequantizeParser : public TfliteNodeParser {
public: public:
TfliteDequantizeParser() : TfliteNodeParser("Dequantize") {} TfliteDequantizeParser() : TfliteNodeParser("Dequantize") {}


~TfliteDequantizeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_expand_dims_parser.h View File

@@ -29,6 +29,8 @@ class TfliteExpandDimsParser : public TfliteNodeParser {
public: public:
TfliteExpandDimsParser() : TfliteNodeParser("ExpandDims") {} TfliteExpandDimsParser() : TfliteNodeParser("ExpandDims") {}


~TfliteExpandDimsParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_fill_parser.h View File

@@ -29,6 +29,8 @@ class TfliteFillParser : public TfliteNodeParser {
public: public:
TfliteFillParser() : TfliteNodeParser("Fill") {} TfliteFillParser() : TfliteNodeParser("Fill") {}


~TfliteFillParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_fullyconnected_parser.h View File

@@ -29,6 +29,8 @@ class TfliteFullyConnectedParser : public TfliteNodeParser {
public: public:
TfliteFullyConnectedParser() : TfliteNodeParser("FullyConnected") {} TfliteFullyConnectedParser() : TfliteNodeParser("FullyConnected") {}


~TfliteFullyConnectedParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_gather_nd_parser.h View File

@@ -29,6 +29,8 @@ class TfliteGatherNdParser : public TfliteNodeParser {
public: public:
TfliteGatherNdParser() : TfliteNodeParser("GatherND") {} TfliteGatherNdParser() : TfliteNodeParser("GatherND") {}


~TfliteGatherNdParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_gather_parser.h View File

@@ -29,6 +29,8 @@ class TfliteGatherParser : public TfliteNodeParser {
public: public:
TfliteGatherParser() : TfliteNodeParser("Gather") {} TfliteGatherParser() : TfliteNodeParser("Gather") {}


~TfliteGatherParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_hashtable_lookup_parser.h View File

@@ -29,6 +29,8 @@ class TfliteHashtableLookupParser : public TfliteNodeParser {
public: public:
TfliteHashtableLookupParser() : TfliteNodeParser("HashtableLookup") {} TfliteHashtableLookupParser() : TfliteNodeParser("HashtableLookup") {}


~TfliteHashtableLookupParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_l2norm_parser.h View File

@@ -29,6 +29,8 @@ class TfliteL2NormParser : public TfliteNodeParser {
public: public:
TfliteL2NormParser() : TfliteNodeParser("L2_NORMALIZATION") {} TfliteL2NormParser() : TfliteNodeParser("L2_NORMALIZATION") {}


~TfliteL2NormParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h View File

@@ -29,6 +29,8 @@ class TfliteLogicalAndParser : public TfliteNodeParser {
public: public:
TfliteLogicalAndParser() : TfliteNodeParser("LogicalAnd") {} TfliteLogicalAndParser() : TfliteNodeParser("LogicalAnd") {}


~TfliteLogicalAndParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_lrn_parser.h View File

@@ -29,6 +29,8 @@ class TfliteLRNParser : public TfliteNodeParser {
public: public:
TfliteLRNParser() : TfliteNodeParser("LocalResponseNorm") {} TfliteLRNParser() : TfliteNodeParser("LocalResponseNorm") {}


~TfliteLRNParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_lsh_projection_parser.h View File

@@ -29,6 +29,8 @@ class TfliteLshProjectionParser : public TfliteNodeParser {
public: public:
TfliteLshProjectionParser() : TfliteNodeParser("LshProjection") {} TfliteLshProjectionParser() : TfliteNodeParser("LshProjection") {}


~TfliteLshProjectionParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_matmul_parser.h View File

@@ -28,6 +28,8 @@ class TfliteMatMulParser : public TfliteNodeParser {
public: public:
TfliteMatMulParser() : TfliteNodeParser("MatMul") {} TfliteMatMulParser() : TfliteNodeParser("MatMul") {}


~TfliteMatMulParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_node_parser_registry.h View File

@@ -42,6 +42,8 @@ class TfliteNodeRegister {
TfliteNodeRegister(const tflite::BuiltinOperator &type, TfliteNodeParser *parser) { TfliteNodeRegister(const tflite::BuiltinOperator &type, TfliteNodeParser *parser) {
TfliteNodeParserRegistry::GetInstance()->parsers[type] = parser; TfliteNodeParserRegistry::GetInstance()->parsers[type] = parser;
} }

~TfliteNodeRegister() = default;
}; };
} // namespace lite } // namespace lite
} // namespace mindspore } // namespace mindspore


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_one_hot_parser.h View File

@@ -29,6 +29,8 @@ class TfliteOneHotParser : public TfliteNodeParser {
public: public:
TfliteOneHotParser() : TfliteNodeParser("OneHot") {} TfliteOneHotParser() : TfliteNodeParser("OneHot") {}


~TfliteOneHotParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_pad_parser.h View File

@@ -29,6 +29,8 @@ class TflitePadParser : public TfliteNodeParser {
public: public:
TflitePadParser() : TfliteNodeParser("Pad") {} TflitePadParser() : TfliteNodeParser("Pad") {}


~TflitePadParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 4
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_pooling_parser.h View File

@@ -29,6 +29,8 @@ class TfliteAvgPoolParser : public TfliteNodeParser {
public: public:
TfliteAvgPoolParser() : TfliteNodeParser("avg_pool") {} TfliteAvgPoolParser() : TfliteNodeParser("avg_pool") {}


~TfliteAvgPoolParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -37,6 +39,8 @@ class TfliteMaxPoolParser : public TfliteNodeParser {
public: public:
TfliteMaxPoolParser() : TfliteNodeParser("max_pool") {} TfliteMaxPoolParser() : TfliteNodeParser("max_pool") {}


~TfliteMaxPoolParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_quantize_parser.h View File

@@ -28,6 +28,8 @@ class TfliteQuantizeParser : public TfliteNodeParser {
public: public:
TfliteQuantizeParser() : TfliteNodeParser("Quantize") {} TfliteQuantizeParser() : TfliteNodeParser("Quantize") {}


~TfliteQuantizeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_range_parser.h View File

@@ -29,6 +29,8 @@ class TfliteRangeParser : public TfliteNodeParser {
public: public:
TfliteRangeParser() : TfliteNodeParser("Range") {} TfliteRangeParser() : TfliteNodeParser("Range") {}


~TfliteRangeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_rank_parser.h View File

@@ -29,6 +29,8 @@ class TfliteRankParser : public TfliteNodeParser {
public: public:
TfliteRankParser() : TfliteNodeParser("Rank") {} TfliteRankParser() : TfliteNodeParser("Rank") {}


~TfliteRankParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_reduce_parser.h View File

@@ -28,6 +28,8 @@ class TfliteReduceParser : public TfliteNodeParser {
public: public:
TfliteReduceParser() : TfliteNodeParser("node_name") {} TfliteReduceParser() : TfliteNodeParser("node_name") {}


~TfliteReduceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_reshape_parser.h View File

@@ -29,6 +29,8 @@ class TfliteReshapeParser : public TfliteNodeParser {
public: public:
TfliteReshapeParser() : TfliteNodeParser("Reshape") {} TfliteReshapeParser() : TfliteNodeParser("Reshape") {}


~TfliteReshapeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_resize_parser.h View File

@@ -28,6 +28,8 @@ class TfliteResizeParser : public TfliteNodeParser {
public: public:
TfliteResizeParser() : TfliteNodeParser("resize_bilinear") {} TfliteResizeParser() : TfliteNodeParser("resize_bilinear") {}


~TfliteResizeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_reverse_parser.h View File

@@ -29,6 +29,8 @@ class TfliteReverseParser : public TfliteNodeParser {
public: public:
TfliteReverseParser() : TfliteNodeParser("reverse") {} TfliteReverseParser() : TfliteNodeParser("reverse") {}


~TfliteReverseParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_reverse_sequence_parser.h View File

@@ -29,6 +29,8 @@ class TfliteReverseSequenceParser : public TfliteNodeParser {
public: public:
TfliteReverseSequenceParser() : TfliteNodeParser("ReverseSequence") {} TfliteReverseSequenceParser() : TfliteNodeParser("ReverseSequence") {}


~TfliteReverseSequenceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_scatter_nd_parser.h View File

@@ -29,6 +29,8 @@ class TfliteScatterNdParser : public TfliteNodeParser {
public: public:
TfliteScatterNdParser() : TfliteNodeParser("ScatterNd") {} TfliteScatterNdParser() : TfliteNodeParser("ScatterNd") {}


~TfliteScatterNdParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_shape_parser.h View File

@@ -29,6 +29,8 @@ class TfliteShapeParser : public TfliteNodeParser {
public: public:
TfliteShapeParser() : TfliteNodeParser("Shape") {} TfliteShapeParser() : TfliteNodeParser("Shape") {}


~TfliteShapeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_skip_gram_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSkipGramParser : public TfliteNodeParser {
public: public:
TfliteSkipGramParser() : TfliteNodeParser("SkipGram") {} TfliteSkipGramParser() : TfliteNodeParser("SkipGram") {}


~TfliteSkipGramParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_slice_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSliceParser : public TfliteNodeParser {
public: public:
TfliteSliceParser() : TfliteNodeParser("Slice") {} TfliteSliceParser() : TfliteNodeParser("Slice") {}


~TfliteSliceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_softmax_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSoftmaxParser : public TfliteNodeParser {
public: public:
TfliteSoftmaxParser() : TfliteNodeParser("Softmax") {} TfliteSoftmaxParser() : TfliteNodeParser("Softmax") {}


~TfliteSoftmaxParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_space_to_batch_nd_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSpaceToBatchNDParser : public TfliteNodeParser {
public: public:
TfliteSpaceToBatchNDParser() : TfliteNodeParser("SpaceToBatchND") {} TfliteSpaceToBatchNDParser() : TfliteNodeParser("SpaceToBatchND") {}


~TfliteSpaceToBatchNDParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_space_to_depth_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSpaceToDepthParser : public TfliteNodeParser {
public: public:
TfliteSpaceToDepthParser() : TfliteNodeParser("SpaceToDepth") {} TfliteSpaceToDepthParser() : TfliteNodeParser("SpaceToDepth") {}


~TfliteSpaceToDepthParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_sparse_to_dense_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSparseToDenseParser : public TfliteNodeParser {
public: public:
TfliteSparseToDenseParser() : TfliteNodeParser("SparseToDense") {} TfliteSparseToDenseParser() : TfliteNodeParser("SparseToDense") {}


~TfliteSparseToDenseParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_split_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSplitParser : public TfliteNodeParser {
public: public:
TfliteSplitParser() : TfliteNodeParser("Split") {} TfliteSplitParser() : TfliteNodeParser("Split") {}


~TfliteSplitParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_split_v_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSplitVParser : public TfliteNodeParser {
public: public:
TfliteSplitVParser() : TfliteNodeParser("SplitV") {} TfliteSplitVParser() : TfliteNodeParser("SplitV") {}


~TfliteSplitVParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_squeeze_parser.h View File

@@ -29,6 +29,8 @@ class TfliteSqueezeParser : public TfliteNodeParser {
public: public:
TfliteSqueezeParser() : TfliteNodeParser("Squeeze") {} TfliteSqueezeParser() : TfliteNodeParser("Squeeze") {}


~TfliteSqueezeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_stack_parser.h View File

@@ -29,6 +29,8 @@ class TfliteStackParser : public TfliteNodeParser {
public: public:
TfliteStackParser() : TfliteNodeParser("Stack") {} TfliteStackParser() : TfliteNodeParser("Stack") {}


~TfliteStackParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_strided_slice_parser.h View File

@@ -29,6 +29,8 @@ class TfliteStridedSliceParser : public TfliteNodeParser {
public: public:
TfliteStridedSliceParser() : TfliteNodeParser("StridedSlice") {} TfliteStridedSliceParser() : TfliteNodeParser("StridedSlice") {}


~TfliteStridedSliceParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_tile_parser.h View File

@@ -29,6 +29,8 @@ class TfliteTileParser : public TfliteNodeParser {
public: public:
TfliteTileParser() : TfliteNodeParser("Tile") {} TfliteTileParser() : TfliteNodeParser("Tile") {}


~TfliteTileParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_topk_v2_parser.h View File

@@ -29,6 +29,8 @@ class TfliteTopKV2Parser : public TfliteNodeParser {
public: public:
TfliteTopKV2Parser() : TfliteNodeParser("TopKV2") {} TfliteTopKV2Parser() : TfliteNodeParser("TopKV2") {}


~TfliteTopKV2Parser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_transpose_parser.h View File

@@ -29,6 +29,8 @@ class TfliteTransposeParser : public TfliteNodeParser {
public: public:
TfliteTransposeParser() : TfliteNodeParser("Transpose") {} TfliteTransposeParser() : TfliteNodeParser("Transpose") {}


~TfliteTransposeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_unique_parser.h View File

@@ -29,6 +29,8 @@ class TfliteUniqueParser : public TfliteNodeParser {
public: public:
TfliteUniqueParser() : TfliteNodeParser("Unique") {} TfliteUniqueParser() : TfliteNodeParser("Unique") {}


~TfliteUniqueParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_unstack_parser.h View File

@@ -29,6 +29,8 @@ class TfliteUnstackParser : public TfliteNodeParser {
public: public:
TfliteUnstackParser() : TfliteNodeParser("Unstack") {} TfliteUnstackParser() : TfliteNodeParser("Unstack") {}


~TfliteUnstackParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_where_parser.h View File

@@ -29,6 +29,8 @@ class TfliteWhereParser : public TfliteNodeParser {
public: public:
TfliteWhereParser() : TfliteNodeParser("Where") {} TfliteWhereParser() : TfliteNodeParser("Where") {}


~TfliteWhereParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_while_parser.h View File

@@ -29,6 +29,8 @@ class TfliteWhileParser : public TfliteNodeParser {
public: public:
TfliteWhileParser() : TfliteNodeParser("While") {} TfliteWhileParser() : TfliteNodeParser("While") {}


~TfliteWhileParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_zeros_like_parser.h View File

@@ -29,6 +29,8 @@ class TfliteZerosLikeParser : public TfliteNodeParser {
public: public:
TfliteZerosLikeParser() : TfliteNodeParser("ZerosLike") {} TfliteZerosLikeParser() : TfliteNodeParser("ZerosLike") {}


~TfliteZerosLikeParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


+ 2
- 0
mindspore/lite/tools/optimizer/graph/functionalize_cond.h View File

@@ -36,6 +36,8 @@ class FunctionalizeCond {
public: public:
FunctionalizeCond(FuncGraphPtr fg, CNodePtr merge_node) : fg_(fg), merge_node_(merge_node) {} FunctionalizeCond(FuncGraphPtr fg, CNodePtr merge_node) : fg_(fg), merge_node_(merge_node) {}


~FunctionalizeCond() = default;

STATUS Process(); STATUS Process();


private: private:


+ 2
- 0
mindspore/lite/tools/optimizer/graph/functionalize_while.h View File

@@ -35,6 +35,8 @@ class FunctionalizeWhile {
FunctionalizeWhile(std::vector<AnfNodePtr> node_cluster, const CNodePtr &loop_cond_node, FuncGraphPtr fg) FunctionalizeWhile(std::vector<AnfNodePtr> node_cluster, const CNodePtr &loop_cond_node, FuncGraphPtr fg)
: node_cluster_(node_cluster), loop_cond_node_(loop_cond_node), fg_(fg) {} : node_cluster_(node_cluster), loop_cond_node_(loop_cond_node), fg_(fg) {}


~FunctionalizeWhile() = default;

// while // while
STATUS BuildWhileNode(); STATUS BuildWhileNode();
STATUS IdentifyWhileNodeInput(); STATUS IdentifyWhileNodeInput();


Loading…
Cancel
Save