Transpose Functions

static riscv_nmsis_nn_status riscv_transpose_s8_nhcw(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const int32_t *const in_strides, const int32_t *const out_strides)
static riscv_nmsis_nn_status riscv_transpose_s8_default(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const int32_t *const in_strides, const int32_t *const out_strides)
riscv_nmsis_nn_status riscv_transpose_s8(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const nmsis_nn_dims *const output_dims, const nmsis_nn_transpose_params *const transpose_params)
group Transpose Functions

Functions

static riscv_nmsis_nn_status riscv_transpose_s8_nhcw(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const int32_t *const in_strides, const int32_t *const out_strides)
static riscv_nmsis_nn_status riscv_transpose_s8_default(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const int32_t *const in_strides, const int32_t *const out_strides)
riscv_nmsis_nn_status riscv_transpose_s8(const int8_t *input, int8_t *const output, const nmsis_nn_dims *const input_dims, const nmsis_nn_dims *const output_dims, const nmsis_nn_transpose_params *const transpose_params)

Basic transpose function.

Parameters:

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

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

  • 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.

  • transpose_params – [in] Transpose parameters. Contains permutation dimensions.

Returns:

The function returns either RISCV_NMSIS_NN_ARG_ERROR if argument constraints fail. or, RISCV_NMSIS_NN_SUCCESS on successful completion.