diff options
| author | Thomas Preud'homme <thomas.preudhomme@arm.com> | 2024-01-12 19:49:51 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-19 15:56:21 +0000 |
| commit | e340233191a34285856f5bcb313df397982c95ef (patch) | |
| tree | 73569e1b7c18ce61ad4e2989a1cf8814719732af | |
| parent | fe2fd1e0e2fe9e9b9b6c73d57840187faf899966 (diff) | |
| download | specification-e340233191a34285856f5bcb313df397982c95ef.tar.gz | |
Fix typo in argmax pseudocode
Axis is a dimension index of the input tensor, but is used to index the
output tensor.
Signed-off-by: Thomas Preud'homme <thomas.preudhomme@arm.com>
Change-Id: Ie50f3921153e01437fe763b5d2b6cfaa06049f25
| -rw-r--r-- | pseudocode/operators/ARGMAX.tosac | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/pseudocode/operators/ARGMAX.tosac b/pseudocode/operators/ARGMAX.tosac index bbe301f..982957a 100644 --- a/pseudocode/operators/ARGMAX.tosac +++ b/pseudocode/operators/ARGMAX.tosac @@ -23,7 +23,7 @@ for_each(left_index in left_shape) { for_each(right_index in right_shape) { in_t max_value = minimum_s<in_t>; out_t max_index = 0; - for (i = 0; i < shape[axis]; i++) { + for (i = 0; i < shape1[axis]; i++) { dim_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) { |