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/test/test_graph_optimiser.py | 1 - 1 file changed, 1 deletion(-) (limited to 'ethosu/vela/test/test_graph_optimiser.py') diff --git a/ethosu/vela/test/test_graph_optimiser.py b/ethosu/vela/test/test_graph_optimiser.py index d9e171d6..83a3dda6 100644 --- a/ethosu/vela/test/test_graph_optimiser.py +++ b/ethosu/vela/test/test_graph_optimiser.py @@ -296,7 +296,6 @@ def test_pad_followed_by_avg_pool(k_size, padding, expect_pad_removed): "dilation_h_factor": 1, } pool_op = testutil.create_op(Op.AvgPool, [out], pool_out_tens, attrs) - pool_op.add_input_tensor(out) pad_op.run_on_npu = True pool_op.run_on_npu = True nng = testutil.create_graph([pad_op, pool_op]) -- cgit v1.2.1