commit e3de4e553a4d7054cf34d26ac9232d6972f5a719
author Jacob Bohlin <jacob.bohlin@arm.com> Fri Nov 27 14:52:06 2020 +0100
committer tim.hall <tim.hall@arm.com> Fri Nov 27 15:13:14 2020 +0000
tree f08f347525341dc0b4f766490424792db29b19ef
parent 333770a695b72cb4930361e00b82924db8e501fa

MLBEDSW-3633: SplitV incorrectly placed on CPU

Minor fix in SPLITV tensor indexing for supported operators check.

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: If8fa702bfbb25a4a7e5bdb136a19ef72eec7e1c2

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 9f7d544..5cb4b6a 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -235,7 +235,7 @@
     SparseToDense = OperatorInfo()
     Split = OperatorInfo(indices=SPLIT_IFM_INDICES)
     SplitSliceRead = OperatorInfo(indices=IFM_INDICES)
-    SplitV = OperatorInfo(indices=IFM_IFM2_INDICES)
+    SplitV = OperatorInfo(indices=IFM_INDICES)
     Sqrt = OperatorInfo()
     Square = OperatorInfo()
     SquaredDifference = OperatorInfo()

ethosu/vela/supported_operators.py

diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index f7dfec2..a1fcf6a 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -708,7 +708,7 @@
     @staticmethod
     def constraint_splitv_inferred(op):
         "Only one size is allowed to be inferred"
-        sizes = op.ifm2.values
+        sizes = op.inputs[1].values
         valid = np.count_nonzero(sizes == -1) <= 1
         return valid, f"Op has multiple inferred sizes (-1): {sizes}"