diff options
Diffstat (limited to 'ethosu/vela/graph_optimiser_util.py')
-rw-r--r-- | ethosu/vela/graph_optimiser_util.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/ethosu/vela/graph_optimiser_util.py b/ethosu/vela/graph_optimiser_util.py index 5e7e1127..b33851a8 100644 --- a/ethosu/vela/graph_optimiser_util.py +++ b/ethosu/vela/graph_optimiser_util.py @@ -1,4 +1,4 @@ -# Copyright (C) 2021 Arm Limited or its affiliates. All rights reserved. +# Copyright (C) 2021-2022 Arm Limited or its affiliates. All rights reserved. # # SPDX-License-Identifier: Apache-2.0 # @@ -111,6 +111,12 @@ def check_format_restrictions(tens, arch): if _avoid_nhcwb16_for_shapes(tens): return + # Resize bilinear half pixel center implementation requires OFM with linear format to + # allow stride modification in H/W dimensions. + for op in tens.ops: + if op.original_type == Op.ResizeBilinear and op.type == Op.DepthwiseConv2DBias: + return + for op in tens.consumer_list: if op.type == Op.ReduceSum and ( tens.dtype == DataType.int32 or arch.accelerator_config == Accelerator.Ethos_U65_512 |