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/ARGMAX.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/ARGMAX.tosac')
-rw-r--r-- | pseudocode/operators/ARGMAX.tosac | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pseudocode/operators/ARGMAX.tosac b/pseudocode/operators/ARGMAX.tosac index 982957a..73b74a0 100644 --- a/pseudocode/operators/ARGMAX.tosac +++ b/pseudocode/operators/ARGMAX.tosac @@ -24,14 +24,14 @@ for_each(left_index in left_shape) { in_t max_value = minimum_s<in_t>; out_t max_index = 0; for (i = 0; i < shape1[axis]; i++) { - dim_t index = flatten(left_index, [i], right_index); + shape_t index = flatten(left_index, [i], right_index); in_t value = tensor_read<in_t>(input, shape1, index); if (apply_max_s<in_t>(value, max_value) != max_value) { max_value = value; max_index = i; } } - dim_t index = flatten(left_index, right_index); + shape_t index = flatten(left_index, right_index); tensor_write<out_t>(output, shape, index, max_index); } } |