aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/tensor_ops.cc
diff options
context:
space:
mode:
authorKevin Cheng <kevin.cheng@arm.com>2020-10-19 12:35:05 -0700
committerKevin Cheng <kevin.cheng@arm.com>2020-10-19 12:35:05 -0700
commit99bea145a050e12f1b5f8301979713d9a9b04e12 (patch)
treebc53dd8cf4566c22b75404dd5cc4ffb849b358d8 /reference_model/src/ops/tensor_ops.cc
parente5e2676409a936431f87d31fb74d825257b20804 (diff)
downloadreference_model-99bea145a050e12f1b5f8301979713d9a9b04e12.tar.gz
Update apply_scale_32()
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com> Change-Id: Ida8e3a17d74e5d6379b2244896ddf9e295d0ecc9
Diffstat (limited to 'reference_model/src/ops/tensor_ops.cc')
-rw-r--r--reference_model/src/ops/tensor_ops.cc4
1 files changed, 2 insertions, 2 deletions
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<Dtype>::eval()
{
this->out->getTensor() = sum.binaryExpr(div_map, [](AccEigenType value, int32_t div) -> OutEigenType {
int32_t multiplier, shift;
- TosaReference::QuantUtil<AccDtype>::reciprocal_scale(div, multiplier, shift);
+ TosaReference::QuantUtil::reciprocal_scale(div, multiplier, shift);
- return (OutEigenType)TosaReference::QuantUtil<AccDtype>::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);