AIfES 2  2.0.0
ailayer_leaky_relu.h
Go to the documentation of this file.
1 
43 #ifndef LEAKY_RELU_LAYER
44 #define LEAKY_RELU_LAYER
45 
46 #include "core/aifes_core.h"
47 
49 
63  void *alpha;
65 
70 
84  void (*leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result);
85 
99  void (*d_leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result);
100 
108  void (*multiply)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
109 
111 };
112 
119 
132 
154 
179 
190 
191 #ifdef AIDEBUG_PRINT_MODULE_SPECS
197 void ailayer_leaky_relu_print_specs(const ailayer_t *self, int (*print)(const char *format, ...));
198 #endif // AIDEBUG_PRINT_MODULE_SPECS
199 
200 #endif // LEAKY_RELU_LAYER
AIfES 2 core interface.
void ailayer_leaky_relu_calc_result_shape(ailayer_t *self)
Calculate the shape of the result tensor.
ailayer_t * ailayer_leaky_relu(ailayer_leaky_relu_t *layer, ailayer_t *input_layer)
Initialize and connect the given Leaky ReLU layer.
void ailayer_leaky_relu_print_specs(const ailayer_t *self, int(*print)(const char *format,...))
Print the layer specification.
void ailayer_leaky_relu_forward(ailayer_t *self)
Calculate the forward pass for given Leaky ReLU layer.
void ailayer_leaky_relu_backward(ailayer_t *self)
Calculate the backward pass for the given Leaky ReLU layer.
const aicore_layertype_t * ailayer_leaky_relu_type
Leaky ReLU layer type.
Type indicator of the layer.
Definition: aifes_core.h:81
General Leaky ReLU layer struct.
Definition: ailayer_leaky_relu.h:53
void(* multiply)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result)
Required math function: Element wise tensor multiplication.
Definition: ailayer_leaky_relu.h:108
void * alpha
Parameter used to calculate Leaky ReLU function for input values < 0.
Definition: ailayer_leaky_relu.h:63
ailayer_t base
Inherited field members from general ailayer struct.
Definition: ailayer_leaky_relu.h:54
const aimath_dtype_t * dtype
Data type of the input and inference result values.
Definition: ailayer_leaky_relu.h:55
void(* leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result)
Required math function: Leaky ReLU.
Definition: ailayer_leaky_relu.h:84
void(* d_leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result)
Required math function: Derivative of Leaky ReLU.
Definition: ailayer_leaky_relu.h:99
AIfES layer interface.
Definition: aifes_core.h:251
Indicator for the used datatype.
Definition: aifes_math.h:53
A tensor in AIfES.
Definition: aifes_math.h:98