diff options
| -rw-r--r-- | ethosu/vela/operation.py | 2 | ||||
| -rw-r--r-- | ethosu/vela/supported_operators.py | 2 |
2 files changed, 2 insertions, 2 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() diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index f7dfec27..a1fcf6ab 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -708,7 +708,7 @@ class SupportedOperators: @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}" |