diff options
| -rw-r--r-- | ethosu/vela/supported_operators.py | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 357e7fe8..3517175e 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -458,12 +458,6 @@ class SupportedOperators: print("Warning: only 8-bit datatypes supported for {}, placing on CPU".format(op.type)) return False - # check batch size - batch_sizes = {1, 2, 4, 8} - if ifm_tensor.shape[0] not in batch_sizes: - print("Warning: only batch sizes {} supported for {}, placing on CPU".format(batch_sizes, op.type)) - return False - if not cls.check_bias_restrictions(bias_tensor): return False |