diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-12-01 12:18:15 +0000 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-12-07 17:19:33 +0000 |
commit | 0de60f38e828e13359acbcfac51b6c179a34d042 (patch) | |
tree | 532cda49e0c9101440715b17a3e18ddaebc75858 /reference_model/include | |
parent | e9059775c0486de4a96d42b41104496f4aefe8e8 (diff) | |
download | reference_model-0de60f38e828e13359acbcfac51b6c179a34d042.tar.gz |
Add support for list of tensors as input parameter
Some operators (e.g. Concat) expect list of tensor as an input
parameter. Currently operators API does not support passing
such parameters from the client code.
In order to enable it:
- Add new type tensor_list_t
- Update operators API generation script to support new type
- Add unit test for operator Concat
Signed-off-by: Dmitrii Agibov <dmitrii.agibov@arm.com>
Change-Id: Ib2f61bcea5e5ecabf56ce031d905cb46a4cc68ea
Diffstat (limited to 'reference_model/include')
-rw-r--r-- | reference_model/include/operators.h | 2 | ||||
-rw-r--r-- | reference_model/include/types.h | 6 |
2 files changed, 7 insertions, 1 deletions
diff --git a/reference_model/include/operators.h b/reference_model/include/operators.h index d037631..62e6db1 100644 --- a/reference_model/include/operators.h +++ b/reference_model/include/operators.h @@ -320,7 +320,7 @@ extern "C" tosa_tensor_t client_output, const func_ctx_t& func_ctx); - tosa_status_t tosa_run_concat(tosa_tensor_t client_input1, + tosa_status_t tosa_run_concat(const tosa_tensor_list_t client_input1, const int32_t client_axis, tosa_tensor_t client_output, const func_ctx_t& func_ctx); diff --git a/reference_model/include/types.h b/reference_model/include/types.h index a371d04..2be884d 100644 --- a/reference_model/include/types.h +++ b/reference_model/include/types.h @@ -75,6 +75,12 @@ extern "C" size_t size; }; + struct tosa_tensor_list_t + { + tosa_tensor_t* tensors; + size_t size; + }; + #ifdef __cplusplus } #endif /* __cplusplus */ |