aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tosa_mapping.py
diff options
context:
space:
mode:
authorPatrik Gustavsson <patrik.gustavsson@arm.com>2021-10-01 11:10:25 +0200
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2021-10-01 11:03:38 +0000
commitef3ebddb6b4fd951046cbe7799721f6f0ed2fc87 (patch)
tree9019c0cb4bb1a513ee1b4b00932e16031ab7da65 /ethosu/vela/tosa_mapping.py
parent3ff7a4aed60fefababdced93a5aba91409a4050c (diff)
downloadethos-u-vela-ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87.tar.gz
TOSA: Add support for Identity operation
Added support for Identity operation. Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com> Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4
Diffstat (limited to 'ethosu/vela/tosa_mapping.py')
-rw-r--r--ethosu/vela/tosa_mapping.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/tosa_mapping.py b/ethosu/vela/tosa_mapping.py
index f80a9156..75f1c9c5 100644
--- a/ethosu/vela/tosa_mapping.py
+++ b/ethosu/vela/tosa_mapping.py
@@ -223,7 +223,6 @@ unsupported_tosa_operators = {
TosaOp.SCATTER,
TosaOp.RESIZE,
TosaOp.CAST,
- TosaOp.IDENTITY,
TosaOp.CUSTOM,
TosaOp.COND_IF,
TosaOp.WHILE_LOOP,
@@ -316,7 +315,7 @@ tosa_operator_map = {
# TODO TosaOp.CAST
TosaOp.RESCALE: (Op.Rescale, rescale_attrs, None, TOSA_IFM_INDICES),
TosaOp.CONST: (Op.Const, None, None, TOSA_NO_INDICES),
- # TODO TosaOp.IDENTITY
+ TosaOp.IDENTITY: (Op.Identity, None, None, TOSA_IFM_INDICES),
# TODO TosaOp.CUSTOM
# TODO TosaOp.COND_IF
# TODO TosaOp.WHILE_LOOP