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 6be9dc25..67717104 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -134,7 +134,7 @@ class Op(Enum): Add = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) AddN = OperatorInfo() Any = OperatorInfo() - ArgMax = OperatorInfo() + ArgMax = OperatorInfo(indices=NNG_IFM_INDICES) ArgMin = OperatorInfo() AvgPool = OperatorInfo(block_type=NpuBlockType.Pooling, indices=NNG_IFM_INDICES) Atan2 = OperatorInfo(indices=NNG_IFM_IFM2_INDICES) |