diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-04-08 09:04:00 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-04-08 11:21:35 +0200 |
commit | ee99bb124b088430b97d205df9fc90a1e9412e0c (patch) | |
tree | bfb1a6799f9e9ff9d0f387471277b8a26edbab70 /ethosu/vela/test/test_graph_optimiser.py | |
parent | 95b279f1454d58a93238851cb5ff394c7782ad32 (diff) | |
download | ethos-u-vela-ee99bb124b088430b97d205df9fc90a1e9412e0c.tar.gz |
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 <patrik.gustavsson@arm.com>
Change-Id: Iec5c7996a1a6039cad052197f1ae56f7c0290440
Diffstat (limited to 'ethosu/vela/test/test_graph_optimiser.py')
-rw-r--r-- | ethosu/vela/test/test_graph_optimiser.py | 1 |
1 files changed, 0 insertions, 1 deletions
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]) |