From 23f87162d068b78c378f8113578cd2047ffd3a7d Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Mon, 12 Oct 2020 10:56:16 +0200 Subject: MLBEDSW-3230 Remove restriction of batching 16 for FC Vela supports batching of FC, restriction removed. Signed-off-by: Patrik Gustavsson Change-Id: Ica56738f1b2676628644fc44f2039a24807f5ccb --- ethosu/vela/supported_operators.py | 6 ------ 1 file changed, 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 -- cgit v1.2.1