diff options
| -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 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) |