diff options
| author | Jiacheng Liang <jiacheng.liang@arm.com> | 2023-05-17 16:49:44 +0100 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-06-02 21:29:19 +0000 |
| commit | eb52cc18b342d6329322f84b671eab4450e663fd (patch) | |
| tree | c6bc5b6fbd064b7fe0a7ae0c0c2be4d9939bd813 /reference_model/src/ops/control_flow.cc | |
| parent | 56a3a06c260714703cb531bac3417ed37aebe6ce (diff) | |
| download | reference_model-eb52cc18b342d6329322f84b671eab4450e663fd.tar.gz | |
Add support for boolean outputs in model runner
Comparison operators produce boolean outputs, which need to be written into client data
Allow subgraph traverser to use main block to look for tensors when serialization handler is missing
Signed-off-by: Jiacheng Liang <jiacheng.liang@arm.com>
Change-Id: I6f9af470185541fa6466b3f7786c48f1555fa6f6
Diffstat (limited to 'reference_model/src/ops/control_flow.cc')
| -rw-r--r-- | reference_model/src/ops/control_flow.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/reference_model/src/ops/control_flow.cc b/reference_model/src/ops/control_flow.cc index 03ad6c6..a0e1fc2 100644 --- a/reference_model/src/ops/control_flow.cc +++ b/reference_model/src/ops/control_flow.cc @@ -172,6 +172,8 @@ OpCondIf::~OpCondIf() int OpCondIf::checkTensorAttributes() { + ERROR_IF(!tsh, "OpCondIf: tosa serialization handler must not be null"); + ERROR_IF(getInputs().size() < 1, "OpCondIf: must have at least 1 operand"); ERROR_IF(inputs[0]->getDtype() != TOSA_REF_TYPE_BOOL || inputs[0]->getRank() != 0, @@ -309,6 +311,11 @@ OpWhileLoop::~OpWhileLoop() int OpWhileLoop::checkTensorAttributes() { + if (!tsh) { + WARNING("OpWhileLoop: tosa serialization handler must not be null"); + return 1; + } + if (getInputs().size() <= 0) { WARNING("OpWhileLoop: must have at least 1 operands"); |