From 189f748e1a79ed88044efbe7137963bca830cbb5 Mon Sep 17 00:00:00 2001 From: Diqing Zhong Date: Tue, 26 Jan 2021 12:12:51 +0100 Subject: MLBEDSW-3224: Support HardSwish Change-Id: If49abc31f093f1bd3393bee86f821fd35972086f Signed-off-by: Diqing Zhong --- ethosu/vela/softmax.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/softmax.py') diff --git a/ethosu/vela/softmax.py b/ethosu/vela/softmax.py index 656a7e69..c3b0611a 100644 --- a/ethosu/vela/softmax.py +++ b/ethosu/vela/softmax.py @@ -203,7 +203,7 @@ class SoftMax: for x in range(256): input_diff = x - 255 if input_diff >= diff_min: - rescale = fp_math.saturating_rounding_mul(input_diff * (1 << shift), scale) + rescale = fp_math.saturating_rounding_mul32(input_diff * (1 << shift), scale) lut.append(fp_math.exp_on_negative_values(rescale)) else: lut.append(0) -- cgit v1.2.1