Function arm_gather_s16

Function Documentation

arm_cmsis_nn_status arm_gather_s16(const int16_t *input_data, const cmsis_nn_dims *input_dims, const int32_t *indices_data, const cmsis_nn_dims *indices_dims, const cmsis_nn_gather_params *params, int16_t *output_data, const cmsis_nn_dims *output_dims)

Gather elements along an axis for int16 tensors.

  1. Supported framework: TensorFlow Lite Micro

Parameters:

  • input_data – [in] Pointer to input tensor data

  • input_dims – [in] Input tensor dimensions

  • indices_data – [in] Pointer to indices tensor data (int32)

  • indices_dims – [in] Indices tensor dimensions

  • params – [in] Pointer to gather parameters

  • output_data – [out] Pointer to output tensor data

  • output_dims – [in] Output tensor dimensions

Returns:

The function returns ARM_CMSIS_NN_SUCCESS