| Kevin Petit | a7ac313 | 2024-01-08 15:27:25 +0000 | [diff] [blame] | 1 | // |
| 2 | // This confidential and proprietary software may be used only as |
| 3 | // authorised by a licensing agreement from ARM Limited |
| 4 | // (C) COPYRIGHT 2020-2024 ARM Limited |
| 5 | // ALL RIGHTS RESERVED |
| 6 | // The entire notice above must be reproduced on all authorised |
| 7 | // copies and copies may only be made to the extent permitted |
| 8 | // by a licensing agreement from ARM Limited. |
| 9 | |
| 10 | ERROR_IF(axis < 0 || axis >= rank(shape)); |
| 11 | for_each(index in shape) { |
| Eric Kunze | 526f6c7 | 2024-01-12 17:18:42 -0800 | [diff] [blame^] | 12 | shape_t tmp_index = index; |
| Kevin Petit | a7ac313 | 2024-01-08 15:27:25 +0000 | [diff] [blame] | 13 | tmp_index[axis] = shape[axis] - 1 - index[axis]; |
| 14 | in_out_t value = tensor_read<in_out_t>(input, shape, tmp_index); |
| 15 | tensor_write<in_out_t>(output, shape, index, value); |
| 16 | } |