Function arm_batch_matmul_f16

• Defined in File arm_nnfunctions_flt.h

Function Documentation

arm_cmsis_nn_status arm_batch_matmul_f16(const cmsis_nn_context *ctx, const cmsis_nn_bmm_params_f16 *bmm_params, const cmsis_nn_dims *input_lhs_dims, const float16_t *input_lhs, const cmsis_nn_dims *input_rhs_dims, const float16_t *input_rhs, const cmsis_nn_dims *output_dims, float16_t *output)

Batched matrix multiplication.

Parameters:

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

• bmm_params – [in] Batch matmul parameters and activation clamp.

• input_lhs_dims – [in] Left-hand-side input tensor dimensions.

• input_lhs – [in] Pointer to the left-hand-side input tensor.

• input_rhs_dims – [in] Right-hand-side input tensor dimensions.

• input_rhs – [in] Pointer to the right-hand-side input tensor.

• output_dims – [in] Output tensor dimensions.

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

Returns:

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.