Function arm_space_to_depth_s16

Function Documentation

arm_cmsis_nn_status arm_space_to_depth_s16(const int16_t *input_data, const cmsis_nn_dims *input_dims, const int32_t block_size, int16_t *output_data, const cmsis_nn_dims *output_dims)

Space to Depth 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_size – [in] Block size for space to depth transformation

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

  • output_dims – [in] Output tensor dimensions. Format: [N, H/block_size, W/block_size, C_IN*block_size*block_size]

Returns:

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