diff options
Diffstat (limited to 'ethosu/vela/softmax.py')
-rw-r--r-- | ethosu/vela/softmax.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |