Pass parameter acc_size to AvgPool2d operator

Signed-off-by: Dmitrii Agibov <dmitrii.agibov@arm.com>
Change-Id: I4cd818af0db5e6e8a96641246cd3263ba4878f56

2 files changed, 8 insertions, 0 deletions

diff --git a/reference_model/include/operators.h b/reference_model/include/operators.h
index 08da277..1399233 100644
--- a/reference_model/include/operators.h
+++ b/reference_model/include/operators.h
@@ -46,6 +46,7 @@ extern "C"
                                       const int32_t client_kernel[2],
                                       const int32_t client_stride[2],
                                       const int32_t client_pad[4],
+                                      const tosa_acc_size_t client_acc_size,
                                       const int32_t client_input_zp,
                                       const int32_t client_output_zp,
                                       tosa_tensor_t client_output,
diff --git a/reference_model/include/types.h b/reference_model/include/types.h
index 42040bf..a371d04 100644
--- a/reference_model/include/types.h
+++ b/reference_model/include/types.h
@@ -58,6 +58,13 @@ extern "C"
         tosa_datatype_fp64_t   = 99
     };
 
+    enum tosa_acc_size_t
+    {
+        tosa_acc_size_int32_t = 0,
+        tosa_acc_size_fp16_t  = 1,
+        tosa_acc_size_fp32_t  = 2
+    };
+
     struct tosa_tensor_t
     {
         const char* name;