diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-06-04 15:51:24 +0200 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | b2fb212216eaa29b96ddf270a0392172265ff02c (patch) | |
tree | e2ed4f847c267fd6d01765cab5faadc263593bf9 /ethosu/vela/tensor.py | |
parent | 3c07c97e0202c1cf01eba06c24b37a8f15ff7a7c (diff) | |
download | ethos-u-vela-b2fb212216eaa29b96ddf270a0392172265ff02c.tar.gz |
MLBEDSW-2420: Improved support for dilated convolution
- Dilation added to SET_KERNEL_STRIDE instruction
- Kernel height/width adjusted for dilation
- Updated padding calculation
- Updated weight compression
Change-Id: I0c8190223e223b039a305aba0f37896ae1de2b80
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 2f91f61c..426a710b 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -521,7 +521,7 @@ class Tensor: strides[4] = stride strides[3] = 16 * stride # STRIDE_X strides[1] = strides[3] * augmented_shape[2] # STRIDE_C - strides[2] = augmented_shape[2] * augmented_shape[3] * stride # STRIDE_Y + strides[2] = augmented_shape[2] * augmented_shape[3] * stride # STRIDE_Y strides[0] = strides[2] * augmented_shape[1] # STRIDE_N return strides, augmented_coord @@ -539,6 +539,15 @@ class Tensor: # Note: for DMA ops, Pass.weight_tensor is referring to the SRAM weight tensor return self.ops[0].inputs[0] if self.needs_dma() else None + def find_npu_op(self): + # Returns the NPU operator that uses this tensor, excluding DMA operators. + for op in self.consumers(): + if op.type == "DMA": + return op.outputs[0].find_npu_op() + if "npu_block_type" in op.attrs: + return op + return None + def compressed_stream_index_from_coord(self, coord): assert self.format == TensorFormat.WeightsCompressed assert len(self.compressed_values) > 0 |