From 36ad73a0fb46d3f844845c97c56d92de2a7a9b3d Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Tue, 6 Oct 2020 13:58:24 +0200 Subject: MLBEDSW-3154 Fix issue for checking axis in concat Fix issue for checking axis in concat, now allowing 0. Signed-off-by: Patrik Gustavsson Change-Id: I85a5fc3dacdfc66dc01b0e05048dd100254fddff --- 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 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] -- cgit v1.2.1