From a648aa9928ba6bd0cca5a6de33c2a3cbbee9a0ed Mon Sep 17 00:00:00 2001 From: Michael McGeagh Date: Wed, 18 Nov 2020 15:44:05 +0000 Subject: vela: Remove ExpandDims from supported ops list EXPAND_DIMS is not yet supported by vela, and so should not be in the list of supported ops. Signed-off-by: Michael McGeagh Change-Id: I5eca13eb52eb9b40ecc6592cda978614c71db99d --- ethosu/vela/supported_operators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index ccf61042..44292386 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -92,7 +92,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, Op.ExpandDims,)) | concat_ops | split_ops + memory_only_ops = set((Op.Squeeze, Op.Reshape, Op.QuantizedReshape,)) | concat_ops | split_ops shapeless_input_ops = binary_elem_wise_main_ops | set((Op.Split, Op.SplitV,)) supported_fused_activations = relu_ops | set((Op.Tanh, Op.Sigmoid, Op.LUT,)) supported_operators = npu_pre_ops | mac_main_ops | elem_wise_main_ops | npu_post_ops | memory_only_ops -- cgit v1.2.1