Function arm_transpose_s16

• Defined in File arm_nnfunctions.h

Function Documentation

arm_cmsis_nn_status arm_transpose_s16(const int16_t *input_data, int16_t *const output_data, const cmsis_nn_dims *const input_dims, const cmsis_nn_dims *const output_dims, const cmsis_nn_transpose_params *const transpose_params)

Basic s16 transpose function.

Parameters:

• input_data – [in] Input (activation) data pointer. Data type: int16

• output_data – [out] Output data pointer. Data type: int16

• input_dims – [in] Input (activation) tensor dimensions. Format: [N, H, W, C_IN]

• output_dims – [in] Output tensor dimensions. Format may be arbitrary relative to input format. The output dimension will depend on the permutation dimensions. In other words the out dimensions are the result of applying the permutation to the input dimensions. The first transpose_params->num_dims fields, taken in the order [N, H, W, C], must satisfy output[i] == input[permutations[i]]; the function returns ARM_CMSIS_NN_ARG_ERROR and writes nothing if they do not.

• transpose_params – [in] Transpose parameters. Contains permutation dimensions. num_dims must be in [1, 4] and permutations must be a bijection over [0, num_dims - 1].

Returns:

The function returns either ARM_CMSIS_NN_ARG_ERROR if argument constraints fail. or, ARM_CMSIS_NN_SUCCESS on successful completion.