Function arm_gather_s8

Function Documentation

arm_cmsis_nn_status arm_gather_s8(const int8_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, int8_t *output_data, const cmsis_nn_dims *output_dims)

Gather elements along an axis for int8 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