diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-11-27 14:52:06 +0100 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-11-27 15:13:14 +0000 |
| commit | e3de4e553a4d7054cf34d26ac9232d6972f5a719 (patch) | |
| tree | f08f347525341dc0b4f766490424792db29b19ef /ethosu/vela/operation.py | |
| parent | 333770a695b72cb4930361e00b82924db8e501fa (diff) | |
| download | ethos-u-vela-e3de4e553a4d7054cf34d26ac9232d6972f5a719.tar.gz | |
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
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
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() |