diff options
author | Raul Farkas <raul.farkas@arm.com> | 2023-05-19 15:25:08 +0100 |
---|---|---|
committer | Raul Farkas <raul.farkas@arm.com> | 2023-06-15 10:56:07 +0100 |
commit | 5d24821355ea5c3af1d069fd50864c5f2f0effd3 (patch) | |
tree | eb4894a4f0af1542a8776e2654886ac760ee8eb5 | |
parent | 8da4f954d32b6395c93ff72123211f40f52cc136 (diff) | |
download | ethos-u-vela-5d24821355ea5c3af1d069fd50864c5f2f0effd3.tar.gz |
MLBEDSW-7531: Remove npu_block_type on unsupported ops
Change-Id: I4f466a7bac77d8bb6fa7243ea2e7c9f3be6d0585
Signed-off-by: Raul Farkas <raul.farkas@arm.com>
-rw-r--r-- | ethosu/vela/operation.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index da923119..3685c5ae 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -148,8 +148,8 @@ class Op(Enum): Atan2 = OperatorInfo(indices=NNG_IFM_IFM2_INDICES) BatchMatMul = OperatorInfo() BatchToSpaceND = OperatorInfo() - BidirectionalSequenceLstm = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) - BidirectionalSequenceRnn = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) + BidirectionalSequenceLstm = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) + BidirectionalSequenceRnn = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) CLZ = OperatorInfo( block_type=NpuBlockType.ElementWise, indices=NNG_IFM_INDICES, is_unary=True ) # NPU specific operation @@ -214,9 +214,9 @@ class Op(Enum): LogicalAnd = OperatorInfo() LogicalNot = OperatorInfo() LogicalOr = OperatorInfo() - Lstm = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) + Lstm = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) LUT = OperatorInfo() # NPU specific, operator has LUT, only used in fused activation functions - MatMul = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) + MatMul = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) MatrixDiag = OperatorInfo() MatrixSetDiag = OperatorInfo() Max = OperatorInfo() @@ -261,7 +261,7 @@ class Op(Enum): ResizeNearestNeighbor = OperatorInfo(block_type=NpuBlockType.Pooling, indices=NNG_IFM_INDICES) ReverseSequence = OperatorInfo() ReverseV2 = OperatorInfo() - Rnn = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) + Rnn = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) Round = OperatorInfo() Rsqrt = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_INDICES, is_unary=True) SHL = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) # NPU specific operation @@ -298,8 +298,8 @@ class Op(Enum): Tile = OperatorInfo() TopKV2 = OperatorInfo() 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) + UnidirectionalSequenceLstm = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) + UnidirectionalSequenceRnn = OperatorInfo(indices=NNG_IFM_WEIGHTS_INDICES) Unique = OperatorInfo() Unpack = OperatorInfo(indices=NNG_IFM_INDICES) UnpackReshaped = OperatorInfo(indices=NNG_IFM_INDICES) |