diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2020-09-25 08:30:44 +0200 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-09-29 07:38:13 +0000 |
| commit | f03bad31c194d1a24ce808223f1b00310a7325e2 (patch) | |
| tree | 3e24836d8319522d994da51faad7ce92273f1ed0 /ethosu/vela/operation.py | |
| parent | eadb166072958febd7e222297d2a65bcf6063f01 (diff) | |
| download | ethos-u-vela-f03bad31c194d1a24ce808223f1b00310a7325e2.tar.gz | |
MLBEDSW-2031: LUT support tanh/sigmoid
Uses LUT for int8/uint8 based tanh/sigmoid.
Change-Id: Ib6ac5a5c958ab9a17e47f620b22c3e22d8d60321
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 252f03b7..14818870 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -152,7 +152,7 @@ input and output tensors, as well as an attribute dictionary.""" weight_idx = 1 ofm_idx = 0 - elif self.type in ("Squeeze", "Reshape", "QuantizedReshape", "ExpandDims"): + elif self.type in ("Squeeze", "Reshape", "QuantizedReshape", "ExpandDims", "Sigmoid", "Tanh"): ifm_idx = 0 ofm_idx = 0 |