diff options
| author | Johan Alfvén <johan.alfven@arm.com> | 2022-01-27 06:47:26 +0100 |
|---|---|---|
| committer | Johan Alfvén <johan.alfven@arm.com> | 2022-01-27 07:35:23 +0100 |
| commit | 628928d32ceae5c95abd9b3a2cb7333b1e5de059 (patch) | |
| tree | 717f0081d3e268369731a398f8cf94726189e420 /ethosu/vela/extract_npu_subgraphs.py | |
| parent | 1793e2f3ebc65b56f8f1cb1bbcb377eb4913c7f7 (diff) | |
| download | ethos-u-vela-628928d32ceae5c95abd9b3a2cb7333b1e5de059.tar.gz | |
MLBEDSW-6060: Revert patch for MLBEDSW-5582
- Issue was due to a previous patch to fix MLBEDSW-5582
- Revert fix for MLBEDSW-5582
commit 849ff81f82c10a68898e5101930b92372bec5565,
- Made new fix for MLBEDSW-5582 that enforce
output tensor from NPU graphs to be in NHWC format.
This information is otherwise lost in the case when
parts of a concatenation are placed in different custom operators
resulting in mismatch bewteen NHWC and NHCWB16.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Iab3ba29d348353c854f357836e6aa7c338ae1572
Diffstat (limited to 'ethosu/vela/extract_npu_subgraphs.py')
| -rw-r--r-- | ethosu/vela/extract_npu_subgraphs.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/extract_npu_subgraphs.py b/ethosu/vela/extract_npu_subgraphs.py index d802b513..f46f031a 100644 --- a/ethosu/vela/extract_npu_subgraphs.py +++ b/ethosu/vela/extract_npu_subgraphs.py @@ -98,6 +98,9 @@ def rewrite_tensor_cpu_producer_npu_consumers( # Deal with output tensors for the NPU graph. These are special. npu_subgraph.output_tensors = [new_tens if tens == orig_tens else tens for tens in npu_subgraph.output_tensors] + for tens in npu_subgraph.output_tensors: + # Enforce output tensor from NPU graph to use normal NHWC output + tens.needs_linear_format = True def rewrite_tensor_npu_producer_cpu_consumers( |