Function arm_nn_depthwise_conv_s8_core

Function Documentation

int8_t *arm_nn_depthwise_conv_s8_core(const int8_t *row, const int16_t *col, const uint16_t num_ch, const int32_t *out_shift, const int32_t *out_mult, const int32_t out_offset, const int32_t activation_min, const int32_t activation_max, const uint16_t kernel_size, const int32_t *const output_bias, int8_t *out)

Depthwise conv on an im2col buffer where the input channel equals output channel.

Parameters:

  • row – [in] pointer to row

  • col – [in] pointer to im2col buffer, always consists of 2 columns.

  • num_ch – [in] number of channels

  • out_shift – [in] pointer to per output channel requantization shift parameter.

  • out_mult – [in] pointer to per output channel requantization multiplier parameter.

  • out_offset – [in] output tensor offset.

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

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

  • kernel_size – [in] number of elements in one column.

  • output_bias – [in] per output channel bias. Range : int32

  • out – [out] pointer to output

Returns:

The function returns one of the two

  1. The incremented output pointer for a successful operation or

  2. NULL if implementation is not available.

    Supported framework: TensorFlow Lite micro.