ailayer_template.h

Go to the documentation of this file.

 
#ifndef TEMPLATE_LAYER
#define TEMPLATE_LAYER
 
#include "core/aifes_core.h"
 
typedef struct ailayer_template     ailayer_template_t;
 
struct ailayer_template {
    ailayer_t base;  // The base or super "class" of the layer.
 
    // Configurations of the layer, that can be configured by the user when creating the layer
    uint32_t example_configuration;
 
    // Tensor pointer for trainable parameters [DEPRECATED]
    aitensor_t *params;
    aitensor_t *d_params;        // Gradients
 
    // If the shape of the result differs from the input shape, this array can be used for the result.
    // If not, the result shape array of the input tensor can be used for the result tensor too.
    uint16_t result_shape[2]; 
    aitensor_t *trainable_params[2]; 
    aitensor_t *gradients[2]; 
    void *optimem[2]; 
    // Make the math functions accessable for expert developers
    void (*copy_tensor)(const aitensor_t *from, aitensor_t *to); 
    void (*tensor_add)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result); 
};
 
extern const aicore_layertype_t *ailayer_template_type;
ailayer_t *ailayer_template(ailayer_template_t *layer, ailayer_t *input_layer);
 
void ailayer_template_forward(ailayer_t *self);
 
void ailayer_template_backward(ailayer_t *self);
 
void ailayer_template_calc_result_shape(ailayer_t *self);
 
uint32_t ailayer_template_sizeof_paramem(const ailayer_t *self);
 
void ailayer_template_set_paramem(ailayer_t *self, void *memory_ptr);
 
uint32_t ailayer_template_sizeof_trainmem(const ailayer_t *self);
 
void ailayer_template_set_trainmem(ailayer_t *self, void *memory_ptr);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailayer_template_print_specs(const ailayer_t *self);
#endif
 
#endif // TEMPLATE_LAYER