aboutsummaryrefslogtreecommitdiff
path: root/pseudocode/operators/ARGMAX.tosac
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2024-01-12 17:18:42 -0800
committerEric Kunze <eric.kunze@arm.com>2024-01-31 06:00:03 +0000
commit526f6c7b5d20e967109ca92c8fc54c26c0438135 (patch)
tree7186a2dec336db8389cdf0cf8bacd60f9da4cf64 /pseudocode/operators/ARGMAX.tosac
parent7e5d187c612fcc715ea3f7f0c900eb13af75a660 (diff)
downloadspecification-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.tosac4
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);
}
}