diff options
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index a2b67dfb..710511c6 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -103,7 +103,7 @@ class Op(Enum): Densify = OperatorInfo() DepthToSpace = OperatorInfo() DepthwiseConv2DBias = OperatorInfo(block_type=NpuBlockType.ConvolutionDepthWise, indices=IFM_WEIGHTS_BIAS_INDICES) - Dequantize = OperatorInfo() + Dequantize = OperatorInfo(indices=IFM_INDICES) Div = OperatorInfo() Elu = OperatorInfo() EmbeddingLookup = OperatorInfo() @@ -163,7 +163,7 @@ class Op(Enum): Pow = OperatorInfo() Prelu = OperatorInfo() Prod = OperatorInfo() - Quantize = OperatorInfo() + Quantize = OperatorInfo(indices=IFM_INDICES) QuantizedAvgPool = OperatorInfo(block_type=NpuBlockType.Pooling, indices=IFM_INDICES) QuantizedConv2D = OperatorInfo(block_type=NpuBlockType.ConvolutionMxN, indices=IFM_WEIGHTS_INDICES) QuantizedMatMul = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=IFM_WEIGHTS_INDICES) @@ -207,7 +207,6 @@ class Op(Enum): SquaredDifference = OperatorInfo() Squeeze = OperatorInfo(indices=IFM_INDICES) StridedSlice = OperatorInfo(indices=IFM_INDICES) - StridedSliceOptions = OperatorInfo() Sub = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=IFM_IFM2_INDICES) SubgraphInput = OperatorInfo() # Only used in CPU subgraphs Sum = OperatorInfo() |