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

Functions
void	axpy_cpu (int N, float ALPHA, float X, int INCX, float Y, int INCY)
void	backward_scale_cpu (float x_norm, float delta, int batch, int n, int size, float *scale_updates)
void	const_cpu (int N, float ALPHA, float *X, int INCX)
void	constrain_ongpu (int N, float ALPHA, float *X, int INCX)
void	copy_cpu (int N, float X, int INCX, float Y, int INCY)
float	dot_cpu (int N, float X, int INCX, float Y, int INCY)
void	fill_cpu (int N, float ALPHA, float *X, int INCX)
void	flatten (float *x, int size, int layers, int batch, int forward)
void	l2_cpu (int n, float pred, float truth, float delta, float error)
void	mean_cpu (float x, int batch, int filters, int spatial, float mean)
void	mean_delta_cpu (float delta, float variance, int batch, int filters, int spatial, float *mean_delta)
void	mul_cpu (int N, float X, int INCX, float Y, int INCY)
void	normalize_cpu (float x, float mean, float *variance, int batch, int filters, int spatial)
void	normalize_delta_cpu (float x, float mean, float variance, float mean_delta, float variance_delta, int batch, int filters, int spatial, float delta)
void	pm (int M, int N, float *A)
void	pow_cpu (int N, float ALPHA, float X, int INCX, float Y, int INCY)
float *	random_matrix (int rows, int cols)
void	reorg_cpu (float x, int w, int h, int c, int batch, int stride, int forward, float out)
void	scal_cpu (int N, float ALPHA, float *X, int INCX)
void	scale_bias (float output, float scales, int batch, int n, int size)
void	shortcut_cpu (int batch, int w1, int h1, int c1, float add, int w2, int h2, int c2, float out)
void	smooth_l1_cpu (int n, float pred, float truth, float delta, float error)
void	softmax (float input, int n, float temp, float output)
void	test_blas ()
void	test_gpu_blas ()
void	time_random_matrix (int TA, int TB, int m, int k, int n)
void	variance_cpu (float x, float mean, int batch, int filters, int spatial, float *variance)
void	variance_delta_cpu (float x, float delta, float mean, float variance, int batch, int filters, int spatial, float *variance_delta)
void	weighted_sum_cpu (float a, float b, float s, int num, float c)

Function Documentation

void axpy_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 146 of file blas.c.

void backward_scale_cpu	(	float *	x_norm,
		float *	delta,
		int	batch,
		int	n,
		int	size,
		float *	scale_updates
	)

Definition at line 59 of file batchnorm_layer.c.

void const_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX
	)

Definition at line 128 of file blas.c.

void constrain_ongpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX
	)

void copy_cpu	(	int	N,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 164 of file blas.c.

float dot_cpu	(	int	N,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 197 of file blas.c.

void fill_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX
	)

Definition at line 158 of file blas.c.

void flatten	(	float *	x,
		int	size,
		int	layers,
		int	batch,
		int	forward
	)

Definition at line 31 of file blas.c.

void l2_cpu	(	int	n,
		float *	pred,
		float *	truth,
		float *	delta,
		float *	error
	)

Definition at line 187 of file blas.c.

void mean_cpu	(	float *	x,
		int	batch,
		int	filters,
		int	spatial,
		float *	mean
	)

Definition at line 83 of file blas.c.

void mean_delta_cpu	(	float *	delta,
		float *	variance,
		int	batch,
		int	filters,
		int	spatial,
		float *	mean_delta
	)

Definition at line 74 of file batchnorm_layer.c.

void mul_cpu	(	int	N,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 134 of file blas.c.

void normalize_cpu	(	float *	x,
		float *	mean,
		float *	variance,
		int	batch,
		int	filters,
		int	spatial
	)

Definition at line 115 of file blas.c.

void normalize_delta_cpu	(	float *	x,
		float *	mean,
		float *	variance,
		float *	mean_delta,
		float *	variance_delta,
		int	batch,
		int	filters,
		int	spatial,
		float *	delta
	)

Definition at line 104 of file batchnorm_layer.c.

void pm	(	int	M,
		int	N,
		float *	A
	)

Definition at line 140 of file utils.c.

void pow_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 140 of file blas.c.

float* random_matrix	(	int	rows,
		int	cols
	)

Definition at line 30 of file gemm.c.

void reorg_cpu	(	float *	x,
		int	w,
		int	h,
		int	c,
		int	batch,
		int	stride,
		int	forward,
		float *	out
	)

Definition at line 8 of file blas.c.

void scal_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX
	)

Definition at line 152 of file blas.c.

void scale_bias	(	float *	output,
		float *	scales,
		int	batch,
		int	n,
		int	size
	)

Definition at line 405 of file convolutional_layer.c.

void shortcut_cpu	(	int	batch,
		int	w1,
		int	h1,
		int	c1,
		float *	add,
		int	w2,
		int	h2,
		int	c2,
		float *	out
	)

Definition at line 57 of file blas.c.

void smooth_l1_cpu	(	int	n,
		float *	pred,
		float *	truth,
		float *	delta,
		float *	error
	)

Definition at line 170 of file blas.c.

void softmax	(	float *	input,
		int	n,
		float	temp,
		float *	output
	)

Definition at line 205 of file blas.c.

void test_blas ( )

void test_gpu_blas ( )

void time_random_matrix	(	int	TA,
		int	TB,
		int	m,
		int	k,
		int	n
	)

Definition at line 40 of file gemm.c.

void variance_cpu	(	float *	x,
		float *	mean,
		int	batch,
		int	filters,
		int	spatial,
		float *	variance
	)

Definition at line 99 of file blas.c.

void variance_delta_cpu	(	float *	x,
		float *	delta,
		float *	mean,
		float *	variance,
		int	batch,
		int	filters,
		int	spatial,
		float *	variance_delta
	)

Definition at line 89 of file batchnorm_layer.c.

void weighted_sum_cpu	(	float *	a,
		float *	b,
		float *	s,
		int	num,
		float *	c
	)

Definition at line 49 of file blas.c.

Functions

Function Documentation