Add boolean input argument support

Add support for tensors of booleans as input arguments to operators.

Signed-off-by: Jack Frankland <jack.frankland@arm.com>
Change-Id: I35bb5768cef11348c27ffbe5869c32c144485958

1 files changed, 7 insertions, 1 deletions

diff --git a/reference_model/src/model_runner_impl.cc b/reference_model/src/model_runner_impl.cc
index ca45e19..447ee26 100644
--- a/reference_model/src/model_runner_impl.cc
+++ b/reference_model/src/model_runner_impl.cc
@@ -203,6 +203,12 @@ int ModelRunnerImpl::setInput(std::string input_name, uint8_t* raw_ptr, size_t s
             status             = setInput(input_name, ArrayProxy(elements, typed_ptr));
             break;
         }
+        case TOSA_REF_TYPE_BOOL: {
+            auto typed_ptr     = reinterpret_cast<unsigned char*>(raw_ptr);
+            const int elements = size / sizeof(unsigned char);
+            status             = setInput(input_name, ArrayProxy(elements, typed_ptr));
+            break;
+        }
         default:
             status = 1;
     }
@@ -386,4 +392,4 @@ template std::vector<float> ModelRunnerImpl::getOutput<float>(std::string output
 template std::vector<half_float::half> ModelRunnerImpl::getOutput<half_float::half>(std::string output_name);
 template std::vector<int32_t> ModelRunnerImpl::getOutput<int32_t>(std::string output_name);
 template std::vector<int64_t> ModelRunnerImpl::getOutput<int64_t>(std::string output_name);
-template std::vector<unsigned char> ModelRunnerImpl::getOutput<unsigned char>(std::string output_name);
\ No newline at end of file
+template std::vector<unsigned char> ModelRunnerImpl::getOutput<unsigned char>(std::string output_name);