diff options
author | Jack Frankland <jack.frankland@arm.com> | 2023-10-17 17:01:07 +0100 |
---|---|---|
committer | Jack Frankland <jack.frankland@arm.com> | 2023-10-17 17:07:02 +0100 |
commit | c48590e3097308b420b57b5816e3e21bf47c88b0 (patch) | |
tree | 8618709b004b9c15a56f872b3a432b1be77adcd2 /reference_model/src/model_runner_impl.cc | |
parent | 65ba809d7a8b4ddd0a51f6c76ad0afc5f417de07 (diff) | |
download | reference_model-c48590e3097308b420b57b5816e3e21bf47c88b0.tar.gz |
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
Diffstat (limited to 'reference_model/src/model_runner_impl.cc')
-rw-r--r-- | reference_model/src/model_runner_impl.cc | 8 |
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); |