ailayer_reshape.h

Go to the documentation of this file.

 
#ifndef AILAYER_RESHAPE_H
#define AILAYER_RESHAPE_H
 
#include "core/aifes_core.h"
 
typedef struct ailayer_reshape  ailayer_reshape_t;
typedef struct ailayer_flatten  ailayer_flatten_t;
 
struct ailayer_reshape {
    ailayer_t base; 
    uint8_t output_dim; 
    uint8_t infer_axis; 
    uint16_t *output_shape; 
 
 
    void (*reshape)(const aitensor_t *x, aitensor_t *result);
 
};
 
struct ailayer_flatten {
    ailayer_reshape_t base; 
    uint16_t output_shape[2]; 
};
 
extern const aicore_layertype_t *ailayer_reshape_type;
 
ailayer_t *ailayer_reshape(ailayer_reshape_t *layer, ailayer_t *input_layer);
 
void ailayer_reshape_forward(ailayer_t *self);
 
void ailayer_reshape_backward(ailayer_t *self);
 
void ailayer_reshape_calc_result_shape(ailayer_t *self);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailayer_reshape_print_specs(const ailayer_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // AILAYER_RESHAPE_H