blas.c File Reference

#include "blas.h"
#include "math.h"
#include <assert.h>
#include <float.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

Include dependency graph for blas.c:

Go to the source code of this file.

Functions
void	axpy_cpu (int N, float ALPHA, float *X, int INCX, float *Y, int INCY)
void	const_cpu (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	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	pow_cpu (int N, float ALPHA, float *X, int INCX, float *Y, int INCY)
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	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	variance_cpu (float *x, float *mean, int batch, int filters, int spatial, float *variance)
void	weighted_sum_cpu (float *a, float *b, float *s, int n, 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 const_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX
	)

Definition at line 128 of file blas.c.

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 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 pow_cpu	(	int	N,
		float	ALPHA,
		float *	X,
		int	INCX,
		float *	Y,
		int	INCY
	)

Definition at line 140 of file blas.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 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 variance_cpu	(	float *	x,
		float *	mean,
		int	batch,
		int	filters,
		int	spatial,
		float *	variance
	)

Definition at line 99 of file blas.c.

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

Definition at line 49 of file blas.c.