From ef3ebddb6b4fd951046cbe7799721f6f0ed2fc87 Mon Sep 17 00:00:00 2001
From: Patrik Gustavsson <patrik.gustavsson@arm.com>
Date: Fri, 1 Oct 2021 11:10:25 +0200
Subject: TOSA: Add support for Identity operation

Added support for Identity operation.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4
---
 ethosu/vela/tosa_mapping.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'ethosu/vela/tosa_mapping.py')

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
-- 
cgit v1.2.1