// | |
// This confidential and proprietary software may be used only as | |
// authorised by a licensing agreement from ARM Limited | |
// (C) COPYRIGHT 2020-2024 ARM Limited | |
// ALL RIGHTS RESERVED | |
// The entire notice above must be reproduced on all authorised | |
// copies and copies may only be made to the extent permitted | |
// by a licensing agreement from ARM Limited. | |
ERROR_IF(axis >= rank(shape)); | |
output[0] = shape_dim(shape, axis); |