Function arm_depth_to_space_s8¶
Defined in File arm_nnfunctions.h
Function Documentation¶
-
arm_cmsis_nn_status arm_depth_to_space_s8(const int8_t *input_data, const cmsis_nn_dims *input_dims, const int32_t block_size, int8_t *output_data, const cmsis_nn_dims *output_dims)¶
Depth to Space function for s8 data type.
Supported Framework: TensorFlow Lite
- Parameters:
input_data – [in] Pointer to the input tensor. Data type: int8
input_dims – [in] Input tensor dimensions. Format: [N, H, W, C_IN]
block_size – [in] Block size for depth to space transformation
output_data – [out] Pointer to the output tensor. Data type: int8
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_ERRORif argument constraints fail. or,ARM_CMSIS_NN_SUCCESSon successful completion.