Function arm_split_f16

Function Documentation

arm_cmsis_nn_status arm_split_f16(const float16_t *input_data, const int32_t input_dims, const int32_t *input_shape, const int32_t axis, const int32_t num_splits, const int32_t *split_dims, float16_t *const *output_data)

float16 split of a tensor into multiple tensors along the target axis.

Data-layout independent pure copy; no arithmetic is performed.

Parameters:

  • input_data – [in] Pointer to the flattened input tensor data.

  • input_dims – [in] Number of dimensions in input_shape.

  • input_shape – [in] Array of length input_dims describing the input shape.

  • axis – [in] Axis along which to split (0 <= axis < input_dims).

  • num_splits – [in] Number of output tensors to produce.

  • split_dims – [in] Array of length num_splits giving each slice size along axis.

  • output_data – [out] Array of pointers to per-output storage.

Returns:

ARM_CMSIS_NN_SUCCESS on success.