ailayer_leaky_relu.h

Go to the documentation of this file.

 
#ifndef LEAKY_RELU_LAYER
#define LEAKY_RELU_LAYER
 
#include "core/aifes_core.h"
 
typedef struct ailayer_leaky_relu   ailayer_leaky_relu_t;
 
struct ailayer_leaky_relu {
    ailayer_t base; 
    void *alpha; 
    const aimath_dtype_t *alpha_dtype; 
 
 
    void (*leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
    void (*d_leaky_relu)(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
    void (*multiply)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
};
 
extern const aicore_layertype_t *ailayer_leaky_relu_type;
 
ailayer_t *ailayer_leaky_relu(ailayer_leaky_relu_t *layer, ailayer_t *input_layer);
 
void ailayer_leaky_relu_forward(ailayer_t *self);
 
void ailayer_leaky_relu_backward(ailayer_t *self);
 
void ailayer_leaky_relu_calc_result_shape(ailayer_t *self);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailayer_leaky_relu_print_specs(const ailayer_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // LEAKY_RELU_LAYER