aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tosa_mapping.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/tosa_mapping.py')
-rw-r--r--ethosu/vela/tosa_mapping.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/ethosu/vela/tosa_mapping.py b/ethosu/vela/tosa_mapping.py
index 75ca43ef..5d0dd33d 100644
--- a/ethosu/vela/tosa_mapping.py
+++ b/ethosu/vela/tosa_mapping.py
@@ -221,11 +221,9 @@ unsupported_tosa_operators = {
TosaOp.REDUCE_SUM,
TosaOp.CONCAT,
TosaOp.PAD,
- TosaOp.RESHAPE,
TosaOp.REVERSE,
TosaOp.SLICE,
TosaOp.TILE,
- TosaOp.TRANSPOSE,
TosaOp.GATHER,
TosaOp.SCATTER,
TosaOp.RESIZE,
@@ -304,11 +302,11 @@ tosa_operator_map = {
# TODO TosaOp.REDUCE_SUM
# TODO TosaOp.CONCAT
# TODO TosaOp.PAD
- # TODO TosaOp.RESHAPE
+ TosaOp.RESHAPE: (Op.Reshape, reshape_attrs, None, TOSA_IFM_INDICES),
# TODO TosaOp.REVERSE
# TODO TosaOp.SLICE
# TODO TosaOp.TILE
- # TODO TosaOp.TRANSPOSE
+ TosaOp.TRANSPOSE: (Op.Transpose, None, None, TOSA_IFM_IFM2_INDICES),
# TODO TosaOp.GATHER
# TODO TosaOp.SCATTER
# TODO TosaOp.RESIZE