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 1e733d56..1707d318 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -285,7 +285,7 @@ class Op(Enum): Tanh = OperatorInfo(indices=NNG_IFM_INDICES) Tile = OperatorInfo() TopKV2 = OperatorInfo() - Transpose = OperatorInfo(indices=NNG_IFM_INDICES) + Transpose = OperatorInfo(indices=NNG_IFM_IFM2_INDICES) UnidirectionalSequenceLstm = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) UnidirectionalSequenceRnn = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) Unique = OperatorInfo() |