diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-10-12 10:56:16 +0200 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-10-12 12:40:13 +0200 |
| commit | 23f87162d068b78c378f8113578cd2047ffd3a7d (patch) | |
| tree | d43ecd99c59b13b38a2aab326302e2918cb7cabc | |
| parent | 7ec09e73b5147af13990f4cac8e8e376ef538174 (diff) | |
| download | ethos-u-vela-23f87162d068b78c378f8113578cd2047ffd3a7d.tar.gz | |
MLBEDSW-3230 Remove restriction of batching 16 for FC
Vela supports batching of FC, restriction removed.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Ica56738f1b2676628644fc44f2039a24807f5ccb
| -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 |