diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-01 12:43:02 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-07 15:35:49 +0200 |
commit | f1580f0167d7e9539a17ac8e33b0b595300f8090 (patch) | |
tree | 2fe1d9d4715ac38be9cfdc5fbe049c07ab5b9563 /ethosu/vela/tosa_reader.py | |
parent | 94292fe32c34357a8935a42c77b759a499eb0db9 (diff) | |
download | ethos-u-vela-f1580f0167d7e9539a17ac8e33b0b595300f8090.tar.gz |
TOSA: Added RESHAPE, SLICE and CONCAT
Added support for Data layout ops
RESHAPE, SLICE and CONCAT.
-No support for bool_t
-Support limited to Rank <= 4 and N = 1
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I487ac494b6506a2a6ba947ee758aa193194dd796
Diffstat (limited to 'ethosu/vela/tosa_reader.py')
-rw-r--r-- | ethosu/vela/tosa_reader.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/tosa_reader.py b/ethosu/vela/tosa_reader.py index 2925ab43..94ba350e 100644 --- a/ethosu/vela/tosa_reader.py +++ b/ethosu/vela/tosa_reader.py @@ -29,6 +29,7 @@ from .reader_util import align_tensor_indices_to_nng from .reader_util import clone_and_reshape_tensor from .reader_util import decode_str from .reader_util import fixup_tensors +from .shape4d import Shape4D from .tensor import QuantizationParameters from .tensor import shape_num_elements from .tensor import Tensor @@ -186,6 +187,9 @@ class TosaSubgraph: op.rescale = [1, shift] if op.type.is_depthwise_conv2d_op(): op.attrs["depth_multiplier"] = op.weights.shape[3] + if op.type == Op.SplitSliceRead: + op.read_offsets[0] = Shape4D.from_list(list(op.attrs["begin"]), 0) + op.read_shapes[0] = op.attrs["size"] elif op.type == Op.Transpose: op.attrs["perms"] = perms.values |