From ee99bb124b088430b97d205df9fc90a1e9412e0c Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Thu, 8 Apr 2021 09:04:00 +0200 Subject: MLBEDSW-4334 Non-linear format decision in graph opt. Check if non linear tensor format can be used is refactored. -Flag avoid_NHCWB16 replaced with needs_linear_format -Checking restrictions located to one function in graph optimiser. Signed-off-by: Patrik Gustavsson Change-Id: Iec5c7996a1a6039cad052197f1ae56f7c0290440 --- ethosu/vela/operation_util.py | 2 -- 1 file changed, 2 deletions(-) (limited to 'ethosu/vela/operation_util.py') diff --git a/ethosu/vela/operation_util.py b/ethosu/vela/operation_util.py index 417f27ef..c51a6b58 100644 --- a/ethosu/vela/operation_util.py +++ b/ethosu/vela/operation_util.py @@ -70,8 +70,6 @@ def create_depthwise_maxpool( op.set_output_tensor(ofm) op.ifm_shapes.append(ifm_shape) op.ofm_shapes.append(Shape4D(ofm.shape)) - op.ifm.avoid_NHCWB16 = True - op.ofm.avoid_NHCWB16 = True return op -- cgit v1.2.1