From 5f2ea2ff77b62d12fc46ae8d08c7815a2a9caae2 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Thu, 15 Oct 2020 08:39:44 +0200 Subject: MLBEDSW-3219: Suppress CPU info Const/Placeholder Suppress info print that Const/Placeholder/SubgraphInput are not supported on the NPU. Change-Id: I6f323b64185b01b619b584c1473ae61d010ab3a4 Signed-off-by: Louis Verhaard --- ethosu/vela/supported_operators.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 4e989124..66c74fce 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -137,7 +137,8 @@ class SupportedOperators: def is_operator_supported(self, op): if op.type not in SupportedOperators.supported_operators: - print('Info: "{}" is not supported on the NPU. Placing on CPU instead'.format(op.type)) + if op.type not in (Op.Placeholder, Op.SubgraphInput, Op.Const): + print("Info: {} '{}' is not supported on the NPU. Placing on CPU instead".format(op.type, op.name)) return False for constraint in self.generic_constraints: valid, extra = constraint(op) -- cgit v1.2.1