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
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()