Function arm_nn_depthwise_conv_nt_t_s16

Function Documentation

int16_t *arm_nn_depthwise_conv_nt_t_s16(const int16_t *lhs, const int8_t *rhs, const uint16_t num_ch, const int32_t *out_shift, const int32_t *out_mult, const int32_t activation_min, const int32_t activation_max, const uint16_t row_x_col, const int64_t *const output_bias, int16_t *out)

Depthwise convolution of transposed rhs matrix with 4 lhs matrices. To be used in non-padded cases. Dimensions are the same for lhs and rhs.

Note

If number of channels is not a multiple of 4, upto 3 elements outside the boundary will be read out for the following.

  • Output shift

  • Output multiplier

  • Output bias

  • rhs

Parameters:

  • lhs – [in] Input left-hand side matrix

  • rhs – [in] Input right-hand side matrix (transposed)

  • num_ch – [in] Number of channels in LHS/RHS

  • out_shift – [in] Per channel output shift. Length of vector is equal to number of channels.

  • out_mult – [in] Per channel output multiplier. Length of vector is equal to number of channels.

  • activation_min – [in] Minimum value to clamp the output to. Range: int8

  • activation_max – [in] Maximum value to clamp the output to. Range: int8

  • row_x_col – [in] (row_dimension * col_dimension) of LHS/RHS matrix

  • output_bias – [in] Per channel output bias. Length of vector is equal to number of channels.

  • out – [in] Output pointer

Returns:

The function returns one of the two

  • Updated output pointer if an implementation is available

  • NULL if no implementation is available.