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 47f4fe0f..54e823a8 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -229,7 +229,7 @@ class Op(Enum): PadV2 = OperatorInfo() Placeholder = OperatorInfo() # Only used in CPU subgraphs Pow = OperatorInfo() - Prelu = OperatorInfo() + Prelu = OperatorInfo(indices=NNG_IFM_IFM2_INDICES) Prod = OperatorInfo() Quantize = OperatorInfo(indices=NNG_IFM_INDICES) QuantizedAvgPool = OperatorInfo(block_type=NpuBlockType.Pooling, indices=NNG_IFM_INDICES) |