NNSupport

Functions
arm_cmsis_nn_status	arm_transpose_f32 (const cmsis_nn_context *ctx, const cmsis_nn_transpose_params_f32 *params, const cmsis_nn_dims *input_dims, const float32_t *input, const cmsis_nn_dims *output_dims, float32_t *output)
	Transpose a floating-point tensor.
void	arm_concatenation_f32_x (const float32_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float32_t *output, int32_t output_x, uint32_t offset_x)
	Concatenate tensors along the X axis.
void	arm_concatenation_f32_y (const float32_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float32_t *output, int32_t output_y, uint32_t offset_y)
	Concatenate tensors along the Y axis.
void	arm_concatenation_f32_z (const float32_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float32_t *output, int32_t output_z, uint32_t offset_z)
	Concatenate tensors along the Z axis.
void	arm_concatenation_f32_w (const float32_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float32_t *output, uint32_t offset_w)
	Concatenate tensors along the W axis.
arm_cmsis_nn_status	arm_batch_norm_f32 (const float32_t *input, float32_t *output, const float32_t *scale, const float32_t *bias, const cmsis_nn_dims *input_dims, arm_nn_tensor_layout layout)
	Apply batch normalization.
void	arm_reshape_f32 (const float32_t *input, float32_t *output, uint32_t total_size)
	Reshape by copying data without changing element order.
arm_cmsis_nn_status	arm_transpose_f16 (const cmsis_nn_context *ctx, const cmsis_nn_transpose_params_f16 *params, const cmsis_nn_dims *input_dims, const float16_t *input, const cmsis_nn_dims *output_dims, float16_t *output)
	Transpose a floating-point tensor.
void	arm_concatenation_f16_x (const float16_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float16_t *output, int32_t output_x, uint32_t offset_x)
	arm concatenation f16 x
void	arm_concatenation_f16_y (const float16_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float16_t *output, int32_t output_y, uint32_t offset_y)
	arm concatenation f16 y
void	arm_concatenation_f16_z (const float16_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float16_t *output, int32_t output_z, uint32_t offset_z)
	arm concatenation f16 z
void	arm_concatenation_f16_w (const float16_t *input, int32_t input_x, int32_t input_y, int32_t input_z, int32_t input_w, float16_t *output, uint32_t offset_w)
	arm concatenation f16 w
arm_cmsis_nn_status	arm_batch_norm_f16 (const float16_t *input, float16_t *output, const float16_t *scale, const float16_t *bias, const cmsis_nn_dims *input_dims, arm_nn_tensor_layout layout)
	Apply batch normalization.
void	arm_reshape_f16 (const float16_t *input, float16_t *output, uint32_t total_size)
	Reshape by copying data without changing element order.

Description

Function Documentation

arm_batch_norm_f16()

arm_cmsis_nn_status arm_batch_norm_f16	(	const float16_t *	input,
		float16_t *	output,
		const float16_t *	scale,
		const float16_t *	bias,
		const cmsis_nn_dims *	input_dims,
		arm_nn_tensor_layout	layout
	)

Apply batch normalization.

arm_batch_norm_f32()

arm_cmsis_nn_status arm_batch_norm_f32	(	const float32_t *	input,
		float32_t *	output,
		const float32_t *	scale,
		const float32_t *	bias,
		const cmsis_nn_dims *	input_dims,
		arm_nn_tensor_layout	layout
	)

Apply batch normalization.

arm_concatenation_f16_w()

void arm_concatenation_f16_w	(	const float16_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float16_t *	output,
		uint32_t	offset_w
	)

arm concatenation f16 w

arm_concatenation_f16_x()

void arm_concatenation_f16_x	(	const float16_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float16_t *	output,
		int32_t	output_x,
		uint32_t	offset_x
	)

arm concatenation f16 x

arm_concatenation_f16_y()

void arm_concatenation_f16_y	(	const float16_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float16_t *	output,
		int32_t	output_y,
		uint32_t	offset_y
	)

arm concatenation f16 y

arm_concatenation_f16_z()

void arm_concatenation_f16_z	(	const float16_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float16_t *	output,
		int32_t	output_z,
		uint32_t	offset_z
	)

arm concatenation f16 z

arm_concatenation_f32_w()

void arm_concatenation_f32_w	(	const float32_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float32_t *	output,
		uint32_t	offset_w
	)

Concatenate tensors along the W axis.

arm_concatenation_f32_x()

void arm_concatenation_f32_x	(	const float32_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float32_t *	output,
		int32_t	output_x,
		uint32_t	offset_x
	)

Concatenate tensors along the X axis.

arm_concatenation_f32_y()

void arm_concatenation_f32_y	(	const float32_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float32_t *	output,
		int32_t	output_y,
		uint32_t	offset_y
	)

Concatenate tensors along the Y axis.

arm_concatenation_f32_z()

void arm_concatenation_f32_z	(	const float32_t *	input,
		int32_t	input_x,
		int32_t	input_y,
		int32_t	input_z,
		int32_t	input_w,
		float32_t *	output,
		int32_t	output_z,
		uint32_t	offset_z
	)

Concatenate tensors along the Z axis.

arm_reshape_f16()

void arm_reshape_f16	(	const float16_t *	input,
		float16_t *	output,
		uint32_t	total_size
	)

Reshape by copying data without changing element order.

arm_reshape_f32()

void arm_reshape_f32	(	const float32_t *	input,
		float32_t *	output,
		uint32_t	total_size
	)

Reshape by copying data without changing element order.

arm_transpose_f16()

arm_cmsis_nn_status arm_transpose_f16	(	const cmsis_nn_context *	ctx,
		const cmsis_nn_transpose_params_f16 *	params,
		const cmsis_nn_dims *	input_dims,
		const float16_t *	input,
		const cmsis_nn_dims *	output_dims,
		float16_t *	output
	)

Transpose a floating-point tensor.

Parameters

[in,out]	ctx	Function context that may hold a temporary scratch buffer.
[in]	params	Transpose parameters, including permutation and layout information.
[in]	input_dims	Input tensor dimensions.
[in]	input	Pointer to the input tensor data.
[in]	output_dims	Output tensor dimensions.
[out]	output	Pointer to the output tensor data.

Returns

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.

arm_transpose_f32()

arm_cmsis_nn_status arm_transpose_f32	(	const cmsis_nn_context *	ctx,
		const cmsis_nn_transpose_params_f32 *	params,
		const cmsis_nn_dims *	input_dims,
		const float32_t *	input,
		const cmsis_nn_dims *	output_dims,
		float32_t *	output
	)

Transpose a floating-point tensor.

Parameters

[in,out]	ctx	Function context that may hold a temporary scratch buffer.
[in]	params	Transpose parameters, including permutation and layout information.
[in]	input_dims	Input tensor dimensions.
[in]	input	Pointer to the input tensor data.
[in]	output_dims	Output tensor dimensions.
[out]	output	Pointer to the output tensor data.

Returns

ARM_CMSIS_NN_SUCCESS on success or ARM_CMSIS_NN_ARG_ERROR on invalid arguments.