From 5d24821355ea5c3af1d069fd50864c5f2f0effd3 Mon Sep 17 00:00:00 2001 From: Raul Farkas Date: Fri, 19 May 2023 15:25:08 +0100 Subject: MLBEDSW-7531: Remove npu_block_type on unsupported ops Change-Id: I4f466a7bac77d8bb6fa7243ea2e7c9f3be6d0585 Signed-off-by: Raul Farkas --- ethosu/vela/operation.py | 14 +++++++------- 1 file 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) -- cgit v1.2.1