diff options
author | Johan Alfvén <johan.alfven@arm.com> | 2022-05-11 13:10:50 +0200 |
---|---|---|
committer | Johan Alfvén <johan.alfven@arm.com> | 2022-05-11 15:38:49 +0200 |
commit | e51a05c4ab4493a6745dd15d6d6a41d0f1663552 (patch) | |
tree | da24eaf94f277365de03363784c64667a23e97a4 /ethosu/vela/tflite_model_semantic.py | |
parent | f018a6831e9b3267436b9e352f08d7b64df628b8 (diff) | |
download | ethos-u-vela-e51a05c4ab4493a6745dd15d6d6a41d0f1663552.tar.gz |
MLBEDSW-6454: Enable ReLu with negative alpha value
Removing constraint for negative alpha value in ReLu
for int8 and uint8.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Id7a3a30bf5d1f0a591f990bd04cd0dbbad5819c6
Diffstat (limited to 'ethosu/vela/tflite_model_semantic.py')
-rw-r--r-- | ethosu/vela/tflite_model_semantic.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/ethosu/vela/tflite_model_semantic.py b/ethosu/vela/tflite_model_semantic.py index c811a0d4..e0541df5 100644 --- a/ethosu/vela/tflite_model_semantic.py +++ b/ethosu/vela/tflite_model_semantic.py @@ -532,10 +532,11 @@ class TFLiteSemantic: @staticmethod def constraint_alpha_valid(op): - "Alpha must not be negative" + "Alpha only allowed to be negative if IFM is int8 or uint8" alpha = op.attrs["alpha"] - valid = alpha >= 0 - return valid, f"Op has alpha={alpha}" + ifm_dtype = op.ifm.dtype + valid = ifm_dtype == DataType.int8 or ifm_dtype == DataType.uint8 or alpha >= 0 + return valid, f"Op has alpha={alpha} and ifm_dtype={ifm_dtype} " @staticmethod def constraint_keep_dim_ifm_ofm(op): |