diff options
| -rw-r--r-- | ethosu/vela/supported_operators.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index fbb306e8..6ae072f0 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -511,7 +511,7 @@ class SupportedOperators: return False if axis < 0: axis += len(op.inputs[0].shape) - if not 0 < axis < len(op.inputs[0].shape): + if not 0 <= axis < len(op.inputs[0].shape): print("Warning:", op.type, "invalid axis", axis, ", placing on CPU") return False ofm = op.outputs[0] |