From f1580f0167d7e9539a17ac8e33b0b595300f8090 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Wed, 1 Sep 2021 12:43:02 +0200 Subject: 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 Change-Id: I487ac494b6506a2a6ba947ee758aa193194dd796 --- ethosu/vela/tosa_reader.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'ethosu/vela/tosa_reader.py') 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 -- cgit v1.2.1