From e3de4e553a4d7054cf34d26ac9232d6972f5a719 Mon Sep 17 00:00:00 2001 From: Jacob Bohlin Date: Fri, 27 Nov 2020 14:52:06 +0100 Subject: MLBEDSW-3633: SplitV incorrectly placed on CPU Minor fix in SPLITV tensor indexing for supported operators check. Signed-off-by: Jacob Bohlin Change-Id: If8fa702bfbb25a4a7e5bdb136a19ef72eec7e1c2 --- ethosu/vela/operation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/operation.py') diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 9f7d544b..5cb4b6ae 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -235,7 +235,7 @@ class Op(Enum): 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() -- cgit v1.2.1