arch_cpu.h File Reference

#include "stdlib.h"
#include "stdint.h"
#include "tinymaix.h"

Include dependency graph for arch_cpu.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
TM_INLINE void	tm_dot_prod (mtype_t *sptr, mtype_t *kptr, uint32_t size, sumtype_t *result)
TM_INLINE void	tm_dot_prod_pack2 (mtype_t *sptr, mtype_t *kptr, uint32_t size, sumtype_t *result)
TM_INLINE void	tm_dot_prod_gap_3x3x1 (mtype_t *sptr, mtype_t *kptr, uint32_t *k_oft, sumtype_t *result)
TM_INLINE void	tm_dot_prod_3x3x1 (mtype_t *sptr, mtype_t *kptr, sumtype_t *result)
TM_INLINE void	tm_postprocess_sum (int n, sumtype_t *sums, btype_t *bs, int act, mtype_t *outp, sctype_t *scales, sctype_t out_s_inv, zptype_t out_zp)

Function Documentation

tm_dot_prod()

TM_INLINE void tm_dot_prod	(	mtype_t *	sptr,
		mtype_t *	kptr,
		uint32_t	size,
		sumtype_t *	result
	)

tm_dot_prod_3x3x1()

TM_INLINE void tm_dot_prod_3x3x1	(	mtype_t *	sptr,
		mtype_t *	kptr,
		sumtype_t *	result
	)

tm_dot_prod_gap_3x3x1()

TM_INLINE void tm_dot_prod_gap_3x3x1	(	mtype_t *	sptr,
		mtype_t *	kptr,
		uint32_t *	k_oft,
		sumtype_t *	result
	)

tm_dot_prod_pack2()

TM_INLINE void tm_dot_prod_pack2	(	mtype_t *	sptr,
		mtype_t *	kptr,
		uint32_t	size,
		sumtype_t *	result
	)

tm_postprocess_sum()

TM_INLINE void tm_postprocess_sum	(	int	n,
		sumtype_t *	sums,
		btype_t *	bs,
		int	act,
		mtype_t *	outp,
		sctype_t *	scales,
		sctype_t	out_s_inv,
		zptype_t	out_zp
	)