Function arm_gather_nd_s16

Function Documentation

arm_cmsis_nn_status arm_gather_nd_s16(const int16_t *params_data, const cmsis_nn_dims *params_dims, const int32_t *indices_data, const cmsis_nn_dims *indices_dims, const cmsis_nn_gather_nd_params *params, int16_t *output_data, const cmsis_nn_dims *output_dims)

Gather_nd slices for int16 tensors.

  1. Supported framework: TensorFlow Lite Micro

Parameters:

  • params_data – [in] Pointer to params tensor data

  • params_dims – [in] Params tensor dimensions

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

  • indices_dims – [in] Indices tensor dimensions

  • params – [in] Pointer to gather_nd parameters

  • output_data – [out] Pointer to output tensor data

  • output_dims – [in] Output tensor dimensions

Returns:

The function returns ARM_CMSIS_NN_SUCCESS