diff options
| author | William Isaksson <william.isaksson@arm.com> | 2023-06-15 15:31:44 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2023-07-10 16:37:04 +0000 |
| commit | 52afbbbe7bdc1e501069166df881fef1357f74d7 (patch) | |
| tree | 5fcfd8373512466fa63e4f852850b11ac4055508 | |
| parent | 762d3acb93c597b5299864f9b3ee551107978ee2 (diff) | |
| download | ethos-u-vela-52afbbbe7bdc1e501069166df881fef1357f74d7.tar.gz | |
MLBEDSW-7752: setting query shapes to Shape4D(0)
Changes query initialization shapes to Shape4D(0,0,0,0) = [0,0,0,0]
instead of Shape4D(0) = [0,1,1,1]. The [0,1,1,1] tensors would affect
performance estimates and are not real.
Change-Id: Ic83b6f6a70c0c904b500f62756e1e125c99856c6
Signed-off-by: William Isaksson <william.isaksson@arm.com>
| -rw-r--r-- | ethosu/vela/npu_performance.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index 5b54538..2325a9c 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -104,7 +104,7 @@ class PassCycles(IntEnum): class PerformanceQuery: def __init__(self, npu_block_type=0): self.npu_block_type = npu_block_type - self.ifm_shape = Shape4D(0) + self.ifm_shape = Shape4D(0, 0, 0, 0) self.ifm_format = TensorFormat.NHWC self.ifm_memory_area = MemArea.Unknown self.ifm2_memory_area = MemArea.Unknown @@ -112,11 +112,11 @@ class PerformanceQuery: self.ifm2_bits = 0 self.ifm2_shape = None self.ifm2_format = TensorFormat.NHWC - self.ofm_shape = Shape4D(0) + self.ofm_shape = Shape4D(0, 0, 0, 0) self.ofm_format = TensorFormat.NHWC self.ofm_memory_area = MemArea.Unknown self.ofm_bits = 0 - self.const_shape = Shape4D(0) + self.const_shape = Shape4D(0, 0, 0, 0) self.const_memory_area = MemArea.Unknown self.kernel = Kernel(1, 1) self.config = ArchitectureBlockConfig() |