From 99bea145a050e12f1b5f8301979713d9a9b04e12 Mon Sep 17 00:00:00 2001 From: Kevin Cheng Date: Mon, 19 Oct 2020 12:35:05 -0700 Subject: Update apply_scale_32() Signed-off-by: Kevin Cheng Change-Id: Ida8e3a17d74e5d6379b2244896ddf9e295d0ecc9 --- reference_model/src/ops/tensor_ops.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'reference_model/src/ops/tensor_ops.cc') diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index a735334..82ce3d2 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -268,9 +268,9 @@ int OpAvgPool2d::eval() { this->out->getTensor() = sum.binaryExpr(div_map, [](AccEigenType value, int32_t div) -> OutEigenType { int32_t multiplier, shift; - TosaReference::QuantUtil::reciprocal_scale(div, multiplier, shift); + TosaReference::QuantUtil::reciprocal_scale(div, multiplier, shift); - return (OutEigenType)TosaReference::QuantUtil::apply_scale(value, multiplier, shift, false); + return (OutEigenType)TosaReference::QuantUtil::apply_scale_32(value, multiplier, shift, false); }); this->out->getTensor() = this->out->getTensor() + (OutEigenType)(this->qinfo->output_zp()); this->out->getTensor() = this->out->getTensor().cwiseMax((OutEigenType)QMin); -- cgit v1.2.1