Function arm_transpose_conv_f32

• Defined in File arm_nnfunctions_flt.h

Function Documentation

arm_cmsis_nn_status arm_transpose_conv_f32(const cmsis_nn_context *ctx, const cmsis_nn_context *output_ctx, const cmsis_nn_transpose_conv_params_f32 *transpose_conv_params, const cmsis_nn_dims *input_dims, const float32_t *input_data, const cmsis_nn_dims *filter_dims, const float32_t *filter_data, const cmsis_nn_dims *bias_dims, const float32_t *bias_data, const cmsis_nn_dims *output_dims, float32_t *output_data, arm_nn_tensor_layout layout)

Transpose convolution, dispatch by layout.

Parameters:

• ctx – [inout] Function context that may hold a temporary scratch buffer.

• output_ctx – [inout] Output accumulation context for helper implementations.

• transpose_conv_params – [in] Transpose convolution parameters.

• input_dims – [in] Input tensor dimensions.

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

• filter_dims – [in] Filter tensor dimensions.

• filter_data – [in] Pointer to the filter tensor data.

• bias_dims – [in] Bias tensor dimensions.

• bias_data – [in] Optional bias tensor data.

• output_dims – [in] Output tensor dimensions.

• output_data – [out] Pointer to the output tensor data.

• layout – [in] Tensor layout selector. Current float APIs require ARM_NN_LAYOUT_NHWC.

Returns:

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.