From f03bad31c194d1a24ce808223f1b00310a7325e2 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Fri, 25 Sep 2020 08:30:44 +0200 Subject: MLBEDSW-2031: LUT support tanh/sigmoid Uses LUT for int8/uint8 based tanh/sigmoid. Change-Id: Ib6ac5a5c958ab9a17e47f620b22c3e22d8d60321 Signed-off-by: Louis Verhaard --- ethosu/vela/operation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/operation.py') 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 -- cgit v1.2.1