diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-09-21 11:05:58 +0100 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-09-25 14:28:43 +0100 |
commit | c8fdccfa51a631a4d6df725a88d38d4679b2a439 (patch) | |
tree | c9e072a48089343ee3a8260f3ed2ec892edb8d1e /reference_model/test | |
parent | 99f8f9fb4cc93f089209772d7d1c3b45ca399437 (diff) | |
download | reference_model-c8fdccfa51a631a4d6df725a88d38d4679b2a439.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')
-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 |