diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-09-21 11:05:58 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-28 16:42:24 -0800 |
commit | 0c0a263bf6742e943bebd42ccf97dcdbd8f4e1c8 (patch) | |
tree | 19c1d48d61b1774ccaf94b566900d17e9dbc0fb7 /reference_model/test/model_runner_tests.cpp | |
parent | 4762564da970eb1883a54aa66582e05c0dbd2b81 (diff) | |
download | reference_model-0c0a263bf6742e943bebd42ccf97dcdbd8f4e1c8.tar.gz |
Pass parameter acc_size to AvgPool2d operator
Signed-off-by: Dmitrii Agibov <dmitrii.agibov@arm.com>
Change-Id: I4cd818af0db5e6e8a96641246cd3263ba4878f56
Diffstat (limited to 'reference_model/test/model_runner_tests.cpp')
-rw-r--r-- | reference_model/test/model_runner_tests.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/reference_model/test/model_runner_tests.cpp b/reference_model/test/model_runner_tests.cpp index 820ed63..7cf9d68 100644 --- a/reference_model/test/model_runner_tests.cpp +++ b/reference_model/test/model_runner_tests.cpp @@ -111,8 +111,10 @@ TEST_SUITE("model_runner") output.data = reinterpret_cast<uint8_t*>(dstData.data()); output.size = dstData.size() * sizeof(float); + tosa_acc_size_t acc_size = tosa_acc_size_fp32_t; + // Execution - auto status = tosa_run_avg_pool2d(input, kernel, stride, pad, 0, 0, output, {}); + auto status = tosa_run_avg_pool2d(input, kernel, stride, pad, acc_size, 0, 0, output, {}); CHECK((status == tosa_status_valid)); // Compare results |