diff options
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 19258b52..160cf630 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -21,6 +21,7 @@ import uuid import numpy as np from . import numeric_util +from .ethos_u55_regs.ethos_u55_regs import resampling_mode from .numeric_util import round_up_divide from .range_set import MemoryRangeSet @@ -230,6 +231,7 @@ class Tensor: "cpu_tensor", "npu_tensor", "equivalence_id", + "resampling_mode", ) AllocationQuantum = 16 @@ -274,6 +276,7 @@ class Tensor: self.reshaped = False self.block_traversal = TensorBlockTraversal.Default + self.resampling_mode = resampling_mode.NONE def element_size(self): if self.element_size_bytes == 0: @@ -312,6 +315,8 @@ class Tensor: else: res.quantization = None + res.resampling_mode = self.resampling_mode + return res def clone_into_fast_storage(self, arch): |