diff options
Diffstat (limited to 'tests/nnfw_api/src/CircleGen.cc')
-rw-r--r-- | tests/nnfw_api/src/CircleGen.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/nnfw_api/src/CircleGen.cc b/tests/nnfw_api/src/CircleGen.cc index 19cb95f37..6ebd5a945 100644 --- a/tests/nnfw_api/src/CircleGen.cc +++ b/tests/nnfw_api/src/CircleGen.cc @@ -183,6 +183,14 @@ uint32_t CircleGen::addOperatorWhile(const OperatorParams ¶ms, uint32_t cond circle::BuiltinOptions_WhileOptions, options); } +uint32_t CircleGen::addOperatorIf(const OperatorParams ¶ms, uint32_t then_subg, + uint32_t else_subg) +{ + auto options = circle::CreateIfOptions(_fbb, then_subg, else_subg).Union(); + return addOperatorWithOptions(params, circle::BuiltinOperator_IF, + circle::BuiltinOptions_IfOptions, options); +} + // NOTE Please add addOperator functions ABOVE this lie // // % How to add a new addOperatorXXX fuction |