ailayer_maxpool2d.h

Go to the documentation of this file.

 
#ifndef AILAYER_MAXPOOL2D_H
#define AILAYER_MAXPOOL2D_H
 
#include "core/aifes_core.h"
 
#define AIFES_CHANNELS_FIRST    1
#define AIFES_CHANNELS_LAST     -1
 
typedef struct ailayer_maxpool2d    ailayer_maxpool2d_t;
 
struct ailayer_maxpool2d {
    ailayer_t base; 
    // Values (configurable)
    uint16_t pool_size[2];    
    uint16_t stride[2];  
    uint16_t padding[2]; // Must be 0 >= padding < pool_size; -inf padding
 
    int8_t channel_axis; 
 
    // Values (autogenerated)
    uint32_t *max_locations;    
    void *optimem[2];   
 
 
    void (*maxpool2d_fwd)(
                          const aitensor_t *input,
                          const uint16_t pool_size[2],
                          const uint16_t stride[2],
                          const uint16_t padding[2],
                          int8_t channel_axis,
                          void *work_space,
                          uint32_t *max_locations,
                          aitensor_t *output
                          );
 
    void (*maxpool2d_bwd)(
                          const aitensor_t *delta_out,
                          const uint16_t pool_size[2],
                          const uint16_t stride[2],
                          const uint16_t padding[2],
                          int8_t channel_axis,
                          void *work_space,
                          const uint32_t *max_locations,
                          aitensor_t *delta_in
                          );
 
    uint16_t result_shape[4]; 
};
 
extern const aicore_layertype_t *ailayer_maxpool2d_type;
 
ailayer_t *ailayer_maxpool2d(ailayer_maxpool2d_t *layer, ailayer_t *input_layer);
 
void ailayer_maxpool2d_forward(ailayer_t *self);
 
void ailayer_maxpool2d_backward(ailayer_t *self);
 
void ailayer_maxpool2d_calc_result_shape(ailayer_t *self);
 
uint32_t ailayer_maxpool2d_sizeof_trainmem(const ailayer_t *self);
 
void ailayer_maxpool2d_set_trainmem(ailayer_t *self, void *memory_ptr);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailayer_maxpool2d_print_specs(const ailayer_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // AILAYER_MAXPOOL2D_H