diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-10-04 10:43:14 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2021-10-05 20:33:41 +0000 |
commit | 478101bebd3058a1917d9a9d87ca6d030af71c47 (patch) | |
tree | da9babc14408efd5416e772e3aceb07c3a95fd9b /reference_model/src/subgraph_traverser.cc | |
parent | 848efb46db2d407a9bb4fba1940d06e143a5dbad (diff) | |
download | reference_model-478101bebd3058a1917d9a9d87ca6d030af71c47.tar.gz |
Couple of reference model fixes
- comparison ops could have different type of input/output
- add SUBGRAPH_ERROR_IF() when operator doesn't have any output tensor
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I10f2c10f92de1c7a979221a421fa8e86b26fcc72
Diffstat (limited to 'reference_model/src/subgraph_traverser.cc')
-rw-r--r-- | reference_model/src/subgraph_traverser.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/reference_model/src/subgraph_traverser.cc b/reference_model/src/subgraph_traverser.cc index 3597314..82de69c 100644 --- a/reference_model/src/subgraph_traverser.cc +++ b/reference_model/src/subgraph_traverser.cc @@ -186,6 +186,9 @@ int SubgraphTraverser::initializeGraph() weight_rank = weight_tensor->GetShape().size(); } + SUBGRAPH_ERROR_IF(op->GetOutputTensorNames().size() == 0, + "SubgraphTraverser::initializeGraph(): Op=%s must have at least one output tensor.", + EnumNamesOp()[op->GetOp()]); std::string output_name = op->GetOutputTensorNames()[0]; TosaSerializationTensor* output_tensor = block->GetTensorByName(output_name); SUBGRAPH_ERROR_IF( |