diff options
Diffstat (limited to 'pseudocode')
-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) { |