colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use RTNE as default rounding policy 

RTNE, Round To Nearest Even is a better rounding policy
which aligns with implementation of Tensorflow Lite.

Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-04-27 15:30:15 +08:00
parent eff71c38bf
commit 88f83019a8
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tim/vx/operation_private.h
View File

@ -38,7 +38,7 @@ class OperationImpl {
OperationImpl& BindOutput(const std::shared_ptr<Tensor>& tensor);
OperationImpl& SetRoundingPolicy(
OverflowPolicy overflow_policy = OverflowPolicy::SATURATE,
RoundingPolicy rounding_policy = RoundingPolicy::TO_ZERO,
RoundingPolicy rounding_policy = RoundingPolicy::RTNE,
DownScaleSizeRounding down_scale_size_rounding =
DownScaleSizeRounding::FLOOR,
uint32_t accumulator_bits = 0);
@ -57,4 +57,4 @@ class OperationImpl {
} // namespace vx
} // namespace tim
#endif /* TIM_VX_OPERATION_PRIVATE_H_ */
#endif /* TIM_VX_OPERATION_PRIVATE_H_ */

2
src/tim/vx/type_utils.cc
View File

@ -129,7 +129,7 @@ vsi_enum TranslateRoundingPolicy(RoundingPolicy type) {
default:
break;
}
return VX_ROUND_POLICY_TO_ZERO;
return VX_ROUND_POLICY_TO_NEAREST_EVEN;
}
vsi_enum TranslateDownScaleSizeRounding(DownScaleSizeRounding type) {