aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/operation.py
diff options
context:
space:
mode:
authorJohan Alfven <johan.alfven@arm.com>2023-04-24 13:35:40 +0200
committerRickard Bolin <rickard.bolin@arm.com>2023-05-02 11:03:37 +0000
commitce5027328d2330d33bdfc5d5b016d171e4f8a2fc (patch)
tree16528c451c430d14d3221bfe968e94a54c6af33a /ethosu/vela/operation.py
parent463f74b2ed65753811ddf7460e760aaf983ef5bb (diff)
downloadethos-u-vela-ce5027328d2330d33bdfc5d5b016d171e4f8a2fc.tar.gz
MLBEDSW-2082: Add Exp support
- Added int8 and int16 Exp support, implemented as LUT. - Added generic 8bit and 16bit LUT table functions following the implementation in the latest reference. If new ops are added by the reference, they can easily be implemented in Vela using the generic functions. - Moved convert_to_lut to lut.py to have all LUT related code in one file. - Updated SUPPORTED_OPS.md Change-Id: I388e76ea4b39162313599a5341cfb9bad71a782c Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r--ethosu/vela/operation.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index eafe3bd1..69596522 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -179,7 +179,7 @@ class Op(Enum):
EmbeddingLookup = OperatorInfo()
EmbeddingLookupSparse = OperatorInfo()
Equal = OperatorInfo()
- Exp = OperatorInfo()
+ Exp = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_INDICES, is_unary=True)
ExpandDims = OperatorInfo(indices=NNG_IFM_INDICES)
FakeQuantWithMinMaxArgs = OperatorInfo()
Fill = OperatorInfo()