|
|
|
@@ -29,6 +29,8 @@ class TfliteAddParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteAddParser() : TfliteNodeParser("Add") {} |
|
|
|
|
|
|
|
~TfliteAddParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -37,6 +39,8 @@ class TfliteSubParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteSubParser() : TfliteNodeParser("Sub") {} |
|
|
|
|
|
|
|
~TfliteSubParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -45,6 +49,8 @@ class TfliteMulParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteMulParser() : TfliteNodeParser("Mul") {} |
|
|
|
|
|
|
|
~TfliteMulParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -53,6 +59,8 @@ class TfliteDivParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteDivParser() : TfliteNodeParser("Div") {} |
|
|
|
|
|
|
|
~TfliteDivParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -61,6 +69,8 @@ class TfliteFloorDivParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteFloorDivParser() : TfliteNodeParser("FloorDiv") {} |
|
|
|
|
|
|
|
~TfliteFloorDivParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -69,6 +79,8 @@ class TfliteFloorModParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteFloorModParser() : TfliteNodeParser("FloorMod") {} |
|
|
|
|
|
|
|
~TfliteFloorModParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -77,6 +89,8 @@ class TflitePowParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TflitePowParser() : TfliteNodeParser("PowFusion") {} |
|
|
|
|
|
|
|
~TflitePowParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -85,6 +99,8 @@ class TfliteSquaredDifferenceParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteSquaredDifferenceParser() : TfliteNodeParser("SquaredDifference") {} |
|
|
|
|
|
|
|
~TfliteSquaredDifferenceParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -93,6 +109,8 @@ class TfliteMaximumParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteMaximumParser() : TfliteNodeParser("Maximum") {} |
|
|
|
|
|
|
|
~TfliteMaximumParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -101,6 +119,8 @@ class TfliteMinimumParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteMinimumParser() : TfliteNodeParser("Minimum") {} |
|
|
|
|
|
|
|
~TfliteMinimumParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -109,6 +129,8 @@ class TfliteAbsParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteAbsParser() : TfliteNodeParser("Abs") {} |
|
|
|
|
|
|
|
~TfliteAbsParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -117,6 +139,8 @@ class TfliteExpParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteExpParser() : TfliteNodeParser("Exp") {} |
|
|
|
|
|
|
|
~TfliteExpParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -125,6 +149,8 @@ class TfliteSqrtParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteSqrtParser() : TfliteNodeParser("Sqrt") {} |
|
|
|
|
|
|
|
~TfliteSqrtParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -133,6 +159,8 @@ class TfliteRsqrtParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteRsqrtParser() : TfliteNodeParser("Rsqrt") {} |
|
|
|
|
|
|
|
~TfliteRsqrtParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -141,6 +169,8 @@ class TfliteSquareParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteSquareParser() : TfliteNodeParser("Square") {} |
|
|
|
|
|
|
|
~TfliteSquareParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -149,6 +179,8 @@ class TfliteSinParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteSinParser() : TfliteNodeParser("Sin") {} |
|
|
|
|
|
|
|
~TfliteSinParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -157,6 +189,8 @@ class TfliteCosParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteCosParser() : TfliteNodeParser("Cos") {} |
|
|
|
|
|
|
|
~TfliteCosParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -165,6 +199,8 @@ class TfliteLogParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteLogParser() : TfliteNodeParser("Log") {} |
|
|
|
|
|
|
|
~TfliteLogParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -173,6 +209,8 @@ class TfliteRoundParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteRoundParser() : TfliteNodeParser("Round") {} |
|
|
|
|
|
|
|
~TfliteRoundParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -181,6 +219,8 @@ class TfliteCeilParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteCeilParser() : TfliteNodeParser("Ceil") {} |
|
|
|
|
|
|
|
~TfliteCeilParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -189,6 +229,8 @@ class TfliteFloorParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteFloorParser() : TfliteNodeParser("Floor") {} |
|
|
|
|
|
|
|
~TfliteFloorParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -197,6 +239,8 @@ class TfliteNegParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteNegParser() : TfliteNodeParser("Neg") {} |
|
|
|
|
|
|
|
~TfliteNegParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -205,6 +249,8 @@ class TfliteEqualParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteEqualParser() : TfliteNodeParser("Equal") {} |
|
|
|
|
|
|
|
~TfliteEqualParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -213,6 +259,8 @@ class TfliteNotEqualParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteNotEqualParser() : TfliteNodeParser("NotEqual") {} |
|
|
|
|
|
|
|
~TfliteNotEqualParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -221,6 +269,8 @@ class TfliteGreaterParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteGreaterParser() : TfliteNodeParser("Greater") {} |
|
|
|
|
|
|
|
~TfliteGreaterParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -229,6 +279,8 @@ class TfliteGreaterEqualParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteGreaterEqualParser() : TfliteNodeParser("GreaterEqual") {} |
|
|
|
|
|
|
|
~TfliteGreaterEqualParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -237,6 +289,8 @@ class TfliteLessParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteLessParser() : TfliteNodeParser("Less") {} |
|
|
|
|
|
|
|
~TfliteLessParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
@@ -245,6 +299,8 @@ class TfliteLessEqualParser : public TfliteNodeParser { |
|
|
|
public: |
|
|
|
TfliteLessEqualParser() : TfliteNodeParser("LessEqual") {} |
|
|
|
|
|
|
|
~TfliteLessEqualParser() override = default; |
|
|
|
|
|
|
|
ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, |
|
|
|
const std::unique_ptr<tflite::ModelT> &tflite_model) override; |
|
|
|
}; |
|
|
|
|