aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/supported_operators.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/supported_operators.py')
-rw-r--r--ethosu/vela/supported_operators.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index 4e989124..18496c54 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 not op.type.is_startup_init_op():
+ print('Info: "{}" is not supported on the NPU. Placing on CPU instead'.format(op.type))
return False
for constraint in self.generic_constraints:
valid, extra = constraint(op)