Function arm_elementwise_mul_s16_batch_offset

Function Documentation

arm_cmsis_nn_status arm_elementwise_mul_s16_batch_offset(const int16_t *input_1_vect, const int16_t *input_2_vect, int16_t *output, const int32_t out_offset, const int32_t out_mult, const int32_t out_shift, const int32_t block_size, const int32_t batch_size, const int32_t batch_offset)

s16 elementwise multiplication with s16 output

Supported framework: TensorFlow Lite micro

Parameters:

  • input_1_vect – [in] pointer to input vector 1

  • input_2_vect – [in] pointer to input vector 2

  • output – [inout] pointer to output vector

  • out_offset – [in] output offset

  • out_mult – [in] output multiplier

  • out_shift – [in] output shift

  • block_size – [in] number of samples per batch

  • batch_size – [in] number of samples per batch

  • batch_offset – [in] Number of timesteps between consecutive batches in output, see arm_nn_lstm_step_s16. Note that it is assumed that the input is stored with sequential batches.

Returns:

The function returns ARM_CMSIS_NN_SUCCESS