diff options
author | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-12-14 15:51:20 +0000 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-12-17 17:16:02 +0000 |
commit | 168954814fb6a1cc5e7b2d44784b24402ef30199 (patch) | |
tree | 35693aeee7c291695ba83f27db7f8d81272b787c /ethosu/vela/operation.py | |
parent | f842b69d007e70d70fc5cef3b6f1f50b4cabbd90 (diff) | |
download | ethos-u-vela-168954814fb6a1cc5e7b2d44784b24402ef30199.tar.gz |
MLBEDSW-3694 Replace padding with enum
Use an Enum instead of a bytestring to specify VALID or SAME padding
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: I4e87f8c32b3bfac176d822a68de061e85a558fce
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 32cba365..afc02d41 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -326,6 +326,11 @@ class Op(Enum): return self.value.id < other.value.id +class Padding(Enum): + SAME = 0 + VALID = 1 + + class ActivationFunction: """Fused activation function""" |