Function arm_space_to_batch_nd_s16

Function Documentation

arm_cmsis_nn_status arm_space_to_batch_nd_s16(const int16_t *input_data, const cmsis_nn_dims *input_dims, const cmsis_nn_tile *block_shape, const cmsis_nn_dims *pad, int16_t *output_data, const cmsis_nn_dims *output_dims, const int32_t output_offset)

Space to Batch 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 space to batch transformation

  • pad – [in] Padding 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 + pad_top + pad_bottom)/block_shape[0], (W + pad_left + pad_right)/block_shape[1], C_IN]

  • output_offset – [in] Zero offset for the output tensor. NOT USED. Assume symmetric quantization for s16.

Returns:

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