tm_layers.c File Reference

#include "limits.h"
#include "tinymaix.h"
#include "arch_cpu.h"

Include dependency graph for tm_layers.c:

Macros
#define	OUTSCALE   outscale_inv
#define	SUMSCALE   (sumscale + c)
#define	BATCH_SIZE   2

Functions
	TM_PERF_REG (t_sbuf)
	TM_PERF_REG (t_dotp)
	TM_PERF_REG (t_post)
	TM_PERF_REG (t_valid)
	TM_PERF_REG (t_pad)
	TM_PERF_REG (t_conv)
	TM_PERF_REG (t_pwconv)
	TM_PERF_REG (t_dwconv)
tm_err_t TM_WEAK	tml_conv2d_dwconv2d (tm_mat_t *in, tm_mat_t *out, wtype_t *w, btype_t *b, int kw, int kh, int sx, int sy, int dx, int dy, int act, int pad_top, int pad_bottom, int pad_left, int pad_right, int dmul, sctype_t *ws, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp)
tm_err_t TM_WEAK	tml_gap (tm_mat_t *in, tm_mat_t *out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp)
tm_err_t TM_WEAK	tml_fc (tm_mat_t *in, tm_mat_t *out, wtype_t *w, btype_t *b, sctype_t *ws, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp)
tm_err_t TM_WEAK	tml_softmax (tm_mat_t *in, tm_mat_t *out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp)
tm_err_t TM_WEAK	tml_reshape (tm_mat_t *in, tm_mat_t *out, sctype_t in_s, zptype_t in_zp, sctype_t out_s, zptype_t out_zp)
tm_err_t TM_WEAK	tml_add (tm_mat_t *in0, tm_mat_t *in1, tm_mat_t *out, sctype_t in_s0, zptype_t in_zp0, sctype_t in_s1, zptype_t in_zp1, sctype_t out_s, zptype_t out_zp)

Variables
static uint32_t	k_oft [TM_MAX_KSIZE]
static mtype_t	sbuf [TM_MAX_KCSIZE]
static float	sumscale [TM_MAX_CSIZE]

Macro Definition Documentation

BATCH_SIZE

#define BATCH_SIZE   2

OUTSCALE

#define OUTSCALE   outscale_inv

SUMSCALE

#define SUMSCALE   (sumscale + c)

Function Documentation

TM_PERF_REG() [1/8]

TM_PERF_REG

(

t_conv

)

TM_PERF_REG() [2/8]

TM_PERF_REG

(

t_dotp

)

TM_PERF_REG() [3/8]

TM_PERF_REG

(

t_dwconv

)

TM_PERF_REG() [4/8]

TM_PERF_REG

(

t_pad

)

TM_PERF_REG() [5/8]

TM_PERF_REG

(

t_post

)

TM_PERF_REG() [6/8]

TM_PERF_REG

(

t_pwconv

)

TM_PERF_REG() [7/8]

TM_PERF_REG

(

t_sbuf

)

TM_PERF_REG() [8/8]

TM_PERF_REG

(

t_valid

)

tml_add()

tm_err_t TM_WEAK tml_add	(	tm_mat_t *	in0,
		tm_mat_t *	in1,
		tm_mat_t *	out,
		sctype_t	in_s0,
		zptype_t	in_zp0,
		sctype_t	in_s1,
		zptype_t	in_zp1,
		sctype_t	out_s,
		zptype_t	out_zp
	)

tml_conv2d_dwconv2d()

tm_err_t TM_WEAK tml_conv2d_dwconv2d	(	tm_mat_t *	in,
		tm_mat_t *	out,
		wtype_t *	w,
		btype_t *	b,
		int	kw,
		int	kh,
		int	sx,
		int	sy,
		int	dx,
		int	dy,
		int	act,
		int	pad_top,
		int	pad_bottom,
		int	pad_left,
		int	pad_right,
		int	dmul,
		sctype_t *	ws,
		sctype_t	in_s,
		zptype_t	in_zp,
		sctype_t	out_s,
		zptype_t	out_zp
	)

tml_fc()

tm_err_t TM_WEAK tml_fc	(	tm_mat_t *	in,
		tm_mat_t *	out,
		wtype_t *	w,
		btype_t *	b,
		sctype_t *	ws,
		sctype_t	in_s,
		zptype_t	in_zp,
		sctype_t	out_s,
		zptype_t	out_zp
	)

tml_gap()

tm_err_t TM_WEAK tml_gap	(	tm_mat_t *	in,
		tm_mat_t *	out,
		sctype_t	in_s,
		zptype_t	in_zp,
		sctype_t	out_s,
		zptype_t	out_zp
	)

tml_reshape()

tm_err_t TM_WEAK tml_reshape	(	tm_mat_t *	in,
		tm_mat_t *	out,
		sctype_t	in_s,
		zptype_t	in_zp,
		sctype_t	out_s,
		zptype_t	out_zp
	)

tml_softmax()

tm_err_t TM_WEAK tml_softmax	(	tm_mat_t *	in,
		tm_mat_t *	out,
		sctype_t	in_s,
		zptype_t	in_zp,
		sctype_t	out_s,
		zptype_t	out_zp
	)

Variable Documentation

k_oft

uint32_t k_oft[TM_MAX_KSIZE]

static

sbuf

mtype_t sbuf[TM_MAX_KCSIZE]

static

sumscale

float sumscale[TM_MAX_CSIZE]

static