From eb52cc18b342d6329322f84b671eab4450e663fd Mon Sep 17 00:00:00 2001 From: Jiacheng Liang Date: Wed, 17 May 2023 16:49:44 +0100 Subject: 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 Change-Id: I6f9af470185541fa6466b3f7786c48f1555fa6f6 --- reference_model/src/model_runner_impl.cc | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'reference_model/src/model_runner_impl.cc') diff --git a/reference_model/src/model_runner_impl.cc b/reference_model/src/model_runner_impl.cc index 8089a1a..31e100a 100644 --- a/reference_model/src/model_runner_impl.cc +++ b/reference_model/src/model_runner_impl.cc @@ -264,6 +264,12 @@ int ModelRunnerImpl::getOutput(std::string output_name, uint8_t* raw_ptr, size_t status = tensor->writeToVector(ArrayProxy(elements, typed_ptr)); break; } + case TOSA_REF_TYPE_BOOL: { + auto typed_ptr = reinterpret_cast(raw_ptr); + const int elements = size / sizeof(unsigned char); + status = tensor->writeToVector(ArrayProxy(elements, typed_ptr)); + break; + } default: status = 1; } -- cgit v1.2.1