// // 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. for_each(0 <= n < N, 0 <= w < W, 0 <= c < C) { index_t k = tensor_read(indices, [N,W], [n,w]); REQUIRE(0 <= k && k < K); in_out_t value = tensor_read(values, [N,K,C], [n,k,c]); tensor_write(output, [N,W,C], [n,w,c], value); }