diff options
| author | Eric Kunze <eric.kunze@arm.com> | 2024-01-12 17:18:42 -0800 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-31 06:00:03 +0000 |
| commit | 526f6c7b5d20e967109ca92c8fc54c26c0438135 (patch) | |
| tree | 7186a2dec336db8389cdf0cf8bacd60f9da4cf64 /pseudocode/operators/CONCAT.tosac | |
| parent | 7e5d187c612fcc715ea3f7f0c900eb13af75a660 (diff) | |
| download | specification-526f6c7b5d20e967109ca92c8fc54c26c0438135.tar.gz | |
Add section of shape operatorsv0.90.0
Rework of the shape operations. Shape operations are now done in shape specific
operators rather than being based on type.
shape_t is reworked to a list of size_t values.
Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I2fca0728f9caa6a6fc34a8ce9e389bb581eea959
Diffstat (limited to 'pseudocode/operators/CONCAT.tosac')
| -rw-r--r-- | pseudocode/operators/CONCAT.tosac | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pseudocode/operators/CONCAT.tosac b/pseudocode/operators/CONCAT.tosac index f9329af..ca26f26 100644 --- a/pseudocode/operators/CONCAT.tosac +++ b/pseudocode/operators/CONCAT.tosac @@ -9,7 +9,6 @@ ERROR_IF(axis < 0 || axis >= max(1,rank(shapes1[0]))); ERROR_IF(shape[axis] != sum(shape_dim(shapes1[k], axis) for all k)) -ERROR_IF(in_out_t == shape_t && rank(shape) > 1); // The following checks ensure all inputs are compatible for concatenation for_each(input_shape in shapes1) { ERROR_IF(rank(input_shape) != rank(shapes1[0])); @@ -18,7 +17,7 @@ for_each(input_shape in shapes1) { } } for_each(index1 in shape) { - dim_t index2 = index1; + shape_t index2 = index1; for (tensor t = 0; t < length(input1); t++) { // Continue to concatenate along axis from each tensor // For each output location, we are looking for the @@ -30,3 +29,4 @@ for_each(index1 in shape) { index2[axis] = index2[axis] - shape_dim(shapes1[t], axis); } } + |