commit b081d67f3e96896ea0ca7426a18c5fe153419ee0
author Patrik Gustavsson <patrik.gustavsson@arm.com> Wed Aug 25 13:49:25 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Sep 03 14:37:43 2021 +0200
tree a62a36256836b9bebcf318fe71793ad800099267
parent df99510f04aef99d1b8e9be9bfcde8fc1738b65f

TOSA: Added support for ADD, SUB and MUL

Added support for ADD, SUB and MUL

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I52acdc126b16e2cf4096bcf7a77023ea7d204998

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 80be228..681f498 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -244,6 +244,7 @@
     ReluN = OperatorInfo(indices=NNG_IFM_INDICES)  # TOSA specific
     Rescale = OperatorInfo(indices=NNG_IFM_INDICES)  # TOSA specific
     RescaleAdd = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES)
+    RescaleMul = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES)
     Reshape = OperatorInfo(indices=NNG_IFM_INDICES)
     ResizeBilinear = OperatorInfo(block_type=NpuBlockType.Pooling, indices=NNG_IFM_INDICES)
     ResizeNearestNeighbor = OperatorInfo()