diff options
Diffstat (limited to 'tests/validation/Reference.cpp')
| -rw-r--r-- | tests/validation/Reference.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/validation/Reference.cpp b/tests/validation/Reference.cpp index 0fca661dc4..9cdd2d74d4 100644 --- a/tests/validation/Reference.cpp +++ b/tests/validation/Reference.cpp @@ -525,6 +525,7 @@ RawTensor Reference::compute_reference_pooling_layer(const TensorShape &shape_in max = 1; break; case DataType::QS8: + case DataType::QS16: min = -(1 << fixed_point_position); max = (1 << fixed_point_position); break; @@ -535,7 +536,7 @@ RawTensor Reference::compute_reference_pooling_layer(const TensorShape &shape_in library->fill(ref_src, distribution, 0.0); // Compute reference - ReferenceCPP::pooling_layer(ref_src, ref_dst, pool_info, fixed_point_position); + ReferenceCPP::pooling_layer(ref_src, ref_dst, pool_info); return ref_dst; } |