#include "helpers.h"

Macros
#define	TYPE VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE)

#define	CONST_ONE 1.f

#define	ABS_OP(a) fabs((a))

#define	ADD_OP(a, b) ((a) + (b))

#define	SUB_OP(a, b) ((a) - (b))

#define	MUL_OP(a, b) ((a) * (b))

#define	MLA_OP(a, b, c) ((b) * (c) + (a))

#define	DIV_OP(a, b) ((a) / (b))

#define	EXP_OP(a) exp((a))

#define	LOG_OP(a) log((a))

#define	SQRT_OP(a) sqrt((a))

#define	TANH_OP(a) tanh((a))

#define	ACTIVATION_OP2(op, x) op##_op(x)

#define	ACTIVATION_OP(op, x) ACTIVATION_OP2(op, x)

Functions
DATA_TYPEVEC_SIZE	logistic_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	tanh_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	relu_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	brelu_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	lu_brelu_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	lrelu_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	srelu_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	abs_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	square_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	sqrt_op (DATA_TYPEVEC_SIZE x)

DATA_TYPEVEC_SIZE	linear_op (DATA_TYPEVEC_SIZE x)

Macro Definition Documentation

#define ABS_OP ( a ) fabs((a))

Definition at line 46 of file activation_layer.cl.

Referenced by abs_op().

#define ACTIVATION_OP	(	op,
		x
	)	ACTIVATION_OP2(op, x)

Definition at line 116 of file activation_layer.cl.

#define ACTIVATION_OP2	(	op,
		x
	)	op##_op(x)

Definition at line 115 of file activation_layer.cl.

#define ADD_OP	(	a,
		b
	)	((a) + (b))

Definition at line 47 of file activation_layer.cl.

Referenced by logistic_op(), and srelu_op().

#define CONST_ONE 1.f

Definition at line 45 of file activation_layer.cl.

Referenced by logistic_op(), and srelu_op().

#define DIV_OP	(	a,
		b
	)	((a) / (b))

Definition at line 51 of file activation_layer.cl.

Referenced by logistic_op().

#define EXP_OP ( a ) exp((a))

Definition at line 52 of file activation_layer.cl.

Referenced by logistic_op(), and srelu_op().

#define LOG_OP ( a ) log((a))

Definition at line 53 of file activation_layer.cl.

Referenced by srelu_op().

#define MLA_OP	(	a,
		b,
		c
	)	((b) * (c) + (a))

Definition at line 50 of file activation_layer.cl.

Referenced by linear_op().

#define MUL_OP	(	a,
		b
	)	((a) * (b))

Definition at line 49 of file activation_layer.cl.

Referenced by lrelu_op(), square_op(), and tanh_op().

#define SQRT_OP ( a ) sqrt((a))

Definition at line 54 of file activation_layer.cl.

Referenced by sqrt_op().

#define SUB_OP	(	a,
		b
	)	((a) - (b))

Definition at line 48 of file activation_layer.cl.

#define TANH_OP ( a ) tanh((a))

Definition at line 55 of file activation_layer.cl.

Referenced by tanh_op().

#define TYPE VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE)

Definition at line 26 of file activation_layer.cl.

Referenced by brelu_op(), linear_op(), logistic_op(), lrelu_op(), lu_brelu_op(), srelu_op(), and tanh_op().

Function Documentation

DATA_TYPEVEC_SIZE abs_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 95 of file activation_layer.cl.

References ABS_OP.

 {
     return ABS_OP(x);
 }

DATA_TYPEVEC_SIZE brelu_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 75 of file activation_layer.cl.

References arm_compute::test::fixed_point_arithmetic::detail::max(), arm_compute::test::fixed_point_arithmetic::detail::min(), and TYPE.

 {
     return min((TYPE)A_VAL, max(0, x));
 }

DATA_TYPEVEC_SIZE linear_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 110 of file activation_layer.cl.

References MLA_OP, and TYPE.

 {
     return MLA_OP((TYPE)B_VAL, (TYPE)A_VAL, x);
 }

DATA_TYPEVEC_SIZE logistic_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 60 of file activation_layer.cl.

References ADD_OP, CONST_ONE, DIV_OP, EXP_OP, and TYPE.

 {
     return DIV_OP((TYPE)CONST_ONE, ADD_OP((TYPE)CONST_ONE, EXP_OP(-x)));
 }

DATA_TYPEVEC_SIZE lrelu_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 85 of file activation_layer.cl.

References MUL_OP, and TYPE.

 {
     return select(MUL_OP((TYPE)A_VAL, x), x, x > (TYPE)0);
 }

DATA_TYPEVEC_SIZE lu_brelu_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 80 of file activation_layer.cl.

References arm_compute::test::fixed_point_arithmetic::detail::max(), arm_compute::test::fixed_point_arithmetic::detail::min(), and TYPE.

 {
     return min(max(x, (TYPE)B_VAL), (TYPE)A_VAL);
 }

DATA_TYPEVEC_SIZE relu_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 70 of file activation_layer.cl.

References arm_compute::test::fixed_point_arithmetic::detail::max().

 {
     return max(0, x);
 }

DATA_TYPEVEC_SIZE sqrt_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 105 of file activation_layer.cl.

References SQRT_OP.

 {
     return SQRT_OP(x);
 }

DATA_TYPEVEC_SIZE square_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 100 of file activation_layer.cl.

References MUL_OP.

 {
     return MUL_OP(x, x);
 }

DATA_TYPEVEC_SIZE srelu_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 90 of file activation_layer.cl.

References ADD_OP, CONST_ONE, EXP_OP, LOG_OP, and TYPE.

 {
     return LOG_OP(ADD_OP((TYPE)CONST_ONE, EXP_OP(x)));
 }

DATA_TYPEVEC_SIZE tanh_op ( DATA_TYPEVEC_SIZE x )

inline

Definition at line 65 of file activation_layer.cl.

References MUL_OP, TANH_OP, and TYPE.

 {
     return MUL_OP((TYPE)A_VAL, TANH_OP(MUL_OP((TYPE)B_VAL, x)));
 }

Macros

Functions

Macro Definition Documentation

Function Documentation