Function arm_batch_to_space_nd_s16

Function Documentation

arm_cmsis_nn_status arm_batch_to_space_nd_s16(const int16_t *input_data, const cmsis_nn_dims *input_dims, const cmsis_nn_tile *block_shape, const cmsis_nn_dims *crop, int16_t *output_data, const cmsis_nn_dims *output_dims)

Batch to Space ND function for s16 data type.

  • Supported Framework: TensorFlow Lite

Parameters:

  • input_data – [in] Pointer to the input tensor. Data type: int16

  • input_dims – [in] Input tensor dimensions. Format: [N, H, W, C_IN]

  • block_shape – [in] Block shape for batch to space transformation

  • crop – [in] Cropping for height and width. Format: [n->top, h->left, w->bottom, c->right]

  • output_data – [out] Pointer to the output tensor. Data type: int16

  • output_dims – [in] Output tensor dimensions. Format: [N/(block_shape[0]*block_shape[1]), H*block_shape[0] - crop_top - crop_bottom, W*block_shape[1] - crop_left - crop_right, C_IN]

Returns:

The function returns either ARM_CMSIS_NN_ARG_ERROR if argument constraints fail. or, ARM_CMSIS_NN_SUCCESS on successful completion.