diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-11-01 13:49:37 +0000 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-11-01 14:51:59 +0000 |
commit | b0b9e33c3500bd8dc9b12ef012d4234b1245247a (patch) | |
tree | 9d7579558126028f48374dac507fa8f145cfbf5a /reference_model/include | |
parent | ce53cd103cc2ac09b43b4fdf586249e626bd5627 (diff) | |
download | reference_model-b0b9e33c3500bd8dc9b12ef012d4234b1245247a.tar.gz |
Fix TransposeConv2d in operator API
- Change name of the TransposeConv2d attribute output_shape to out_shape
in generate_api.py to match with TOSA specification
- Fix serialization attributes mapping for operator TransposeConv2d
- Add a unit test for TransposeConv2d operator
Signed-off-by: Dmitrii Agibov <dmitrii.agibov@arm.com>
Change-Id: I6613c0d093aeea0af30012bcc1c8e5d26dec746c
Diffstat (limited to 'reference_model/include')
-rw-r--r-- | reference_model/include/operators.h | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/reference_model/include/operators.h b/reference_model/include/operators.h index e56b882..1650ea4 100644 --- a/reference_model/include/operators.h +++ b/reference_model/include/operators.h @@ -112,13 +112,11 @@ extern "C" tosa_status_t tosa_run_transpose_conv2d(tosa_tensor_t client_input, tosa_tensor_t client_weight, tosa_tensor_t client_bias, + const int32_t client_out_pad[4], const int32_t client_stride[2], + const int32_t client_out_shape[4], const int32_t client_input_zp, const int32_t client_weight_zp, - const int32_t client_pad_len, - const int32_t client_pad[], - const int32_t client_dilation_len, - const int32_t client_dilation[], tosa_tensor_t client_output, const func_ctx_t& func_ctx); |