diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2021-02-03 08:43:54 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2021-02-04 11:41:09 +0100 |
commit | 3d22f3c4cdff3ed1e392048265f84d07c01080d5 (patch) | |
tree | b2bd550e062b002f0ce0cfc34cc3e17bbaf32fc4 /ethosu/vela/supported_operators.py | |
parent | fcb1a00cfd4216782f4fc4429ce66c592a0b8030 (diff) | |
download | ethos-u-vela-3d22f3c4cdff3ed1e392048265f84d07c01080d5.tar.gz |
MLBEDSW-3932: Remove Squeeze from supported operators
- Squeeze is no longer listed as supported operator
- Added missing doc-string for a Pad constraint
Change-Id: Ifd5e493acb0eb28bc4f104df74b3491589db8c29
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/supported_operators.py')
-rw-r--r-- | ethosu/vela/supported_operators.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 8bb9c581..8446ec29 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -98,7 +98,7 @@ class SupportedOperators: ) split_ops = set((Op.Split, Op.SplitV, Op.StridedSlice, Op.Slice, Op.UnpackReshaped, Op.Unpack,)) concat_ops = set((Op.Concat, Op.ConcatTFLite, Op.PackReshaped, Op.Pack,)) - memory_only_ops = set((Op.Squeeze, Op.Reshape, Op.QuantizedReshape,)) | concat_ops | split_ops + memory_only_ops = set((Op.Reshape, Op.QuantizedReshape,)) | concat_ops | split_ops shapeless_input_ops = binary_elem_wise_main_ops | set((Op.Split, Op.SplitV,)) per_axis_quant_ops = convolution_like_ops # per-axis/channel quantization only currently supported for conv ops supported_fused_activations = relu_ops | set((Op.Tanh, Op.Sigmoid, Op.LUT,)) @@ -839,6 +839,7 @@ class SupportedOperators: @staticmethod def constraint_pad_constant(op): + "The padding tensor must be constant" pad_tensor = op.inputs[1].values valid = pad_tensor is not None return valid, f"Op has non-constant padding tensor: {op.inputs[1].values}" |