1 /*
2 * SPDX-FileCopyrightText: Copyright 2010-2022 Arm Limited and/or its affiliates <open-source-office@arm.com>
3 *
4 * SPDX-License-Identifier: Apache-2.0
5 *
6 * Licensed under the Apache License, Version 2.0 (the License); you may
7 * not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
9 *
10 * www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an AS IS BASIS, WITHOUT
14 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 */
18
19 /* ----------------------------------------------------------------------
20 * Project: CMSIS NN Library
21 * Title: arm_concatenation_s8_z.c
22 * Description: s8 version of concatenation along the Z axis
23 *
24 * $Date: 26 October 2022
25 * $Revision: V.1.0.1
26 *
27 * Target Processor: Cortex-M cores
28 *
29 * -------------------------------------------------------------------- */
30
31 #include "arm_nnfunctions.h"
32 #include "arm_nnsupportfunctions.h"
33
34 /**
35 * @ingroup Public
36 */
37
38 /**
39 * @addtogroup Concatenation
40 * @{
41 */
42
43 /*
44 * s8 version of concatenation along the Z axis
45 *
46 * Refer to header file for details.
47 *
48 */
arm_concatenation_s8_z(const int8_t * input,const uint16_t input_x,const uint16_t input_y,const uint16_t input_z,const uint16_t input_w,int8_t * output,const uint16_t output_z,const uint32_t offset_z)49 void arm_concatenation_s8_z(const int8_t *input,
50 const uint16_t input_x,
51 const uint16_t input_y,
52 const uint16_t input_z,
53 const uint16_t input_w,
54 int8_t *output,
55 const uint16_t output_z,
56 const uint32_t offset_z)
57 {
58 const uint32_t input_copy_size = input_x * input_y * input_z;
59 const uint32_t output_stride = input_x * input_y * output_z;
60
61 output += offset_z * (input_x * input_y);
62
63 uint32_t i;
64
65 for (i = 0; i < input_w; ++i)
66 {
67 arm_memcpy_s8(output, input, input_copy_size);
68 input += input_copy_size;
69 output += output_stride;
70 }
71 }
72
73 /**
74 * @} end of Concatenation group
75 */
76
