diff options
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 94d256c..c9a30b2 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -286,7 +286,7 @@ class Op(Enum): SplitV = OperatorInfo(indices=NNG_IFM_INDICES) Sqrt = OperatorInfo() Square = OperatorInfo() - SquaredDifference = OperatorInfo() + SquaredDifference = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) Squeeze = OperatorInfo(indices=NNG_IFM_INDICES) StridedSlice = OperatorInfo(indices=NNG_IFM_INDICES) Sub = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) |