From 94457b175b8646bce089c9924e99686587de8992 Mon Sep 17 00:00:00 2001 From: Diqing Zhong Date: Wed, 9 Dec 2020 15:22:40 +0100 Subject: Revert "Revert "MLMBED-3450: Do not convert batched fully connected to conv"" - We have combined estimates for conv and fc, add the fix back Change-Id: I49a29c716189b37b387df4b46efab5f4e6125994 Signed-off-by: Diqing Zhong --- ethosu/vela/insert_dma.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'ethosu/vela/insert_dma.py') diff --git a/ethosu/vela/insert_dma.py b/ethosu/vela/insert_dma.py index f02039cb..fc1e7986 100644 --- a/ethosu/vela/insert_dma.py +++ b/ethosu/vela/insert_dma.py @@ -77,9 +77,7 @@ def insert_dma_cmd(op, arch, nng): ): only_vector_product_consumers = True for oper in tens.consumers(): - if oper is None or not ( - oper.type.npu_block_type == NpuBlockType.VectorProduct or "is_converted_fc" in oper.attrs - ): + if oper is None or oper.type.npu_block_type != NpuBlockType.VectorProduct: only_vector_product_consumers = False break -- cgit v1.2.1