diff options
Diffstat (limited to 'tests/nnfw_api/src/CircleGen.cc')
-rw-r--r-- | tests/nnfw_api/src/CircleGen.cc | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tests/nnfw_api/src/CircleGen.cc b/tests/nnfw_api/src/CircleGen.cc index e4e4ba1af..4f1c7f9f5 100644 --- a/tests/nnfw_api/src/CircleGen.cc +++ b/tests/nnfw_api/src/CircleGen.cc @@ -269,6 +269,20 @@ uint32_t CircleGen::addOperatorFloorDiv(const OperatorParams ¶ms) circle::BuiltinOptions_NONE, 0); } +uint32_t CircleGen::addOperatorGreater(const OperatorParams ¶ms) +{ + auto options = circle::CreateLessOptions(_fbb).Union(); + return addOperatorWithOptions(params, circle::BuiltinOperator_GREATER, + circle::BuiltinOptions_GreaterOptions, options); +} + +uint32_t CircleGen::addOperatorGreaterEqual(const OperatorParams ¶ms) +{ + auto options = circle::CreateGreaterOptions(_fbb).Union(); + return addOperatorWithOptions(params, circle::BuiltinOperator_GREATER_EQUAL, + circle::BuiltinOptions_GreaterEqualOptions, options); +} + uint32_t CircleGen::addOperatorL2Normalization(const OperatorParams ¶ms) { auto options = circle::CreateL2NormOptions(_fbb).Union(); @@ -283,6 +297,13 @@ uint32_t CircleGen::addOperatorLess(const OperatorParams ¶ms) circle::BuiltinOptions_LessOptions, options); } +uint32_t CircleGen::addOperatorLessEqual(const OperatorParams ¶ms) +{ + auto options = circle::CreateLessOptions(_fbb).Union(); + return addOperatorWithOptions(params, circle::BuiltinOperator_LESS_EQUAL, + circle::BuiltinOptions_LessEqualOptions, options); +} + uint32_t CircleGen::addOperatorLeakyRelu(const OperatorParams ¶ms, float alpha) { auto options = circle::CreateLeakyReluOptions(_fbb, alpha).Union(); @@ -319,6 +340,13 @@ uint32_t CircleGen::addOperatorNeg(const OperatorParams ¶ms) circle::BuiltinOptions_NegOptions, options); } +uint32_t CircleGen::addOperatorNotEqual(const OperatorParams ¶ms) +{ + auto options = circle::CreateEqualOptions(_fbb).Union(); + return addOperatorWithOptions(params, circle::BuiltinOperator_NOT_EQUAL, + circle::BuiltinOptions_NotEqualOptions, options); +} + uint32_t CircleGen::addOperatorOneHot(const OperatorParams ¶ms, int32_t axis) { auto options = circle::CreateOneHotOptions(_fbb, axis).Union(); |