diff options
Diffstat (limited to 'ethosu/vela/npu_performance.py')
-rw-r--r-- | ethosu/vela/npu_performance.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index e315f1f1..c83f8f52 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -389,7 +389,8 @@ def estimate_memory_transfer_efficiency( elem_size = tensor.dtype.size_in_bytes() is_ifm = direction == BandwidthDirection.Read tens = tensor.clone() - if not tens.avoid_NHCWB16: + + if not tensor.needs_linear_format: tens.set_format(TensorFormat.NHCWB16, arch) strides = tens.get_strides(shape4D=shape4D) |