| AIfES 2 2.0.0

ailayer_input.h
Go to the documentation of this file.
1
29 #ifndef INPUT_LAYER
30 #define INPUT_LAYER
31
32 #include "core/aifes_core.h"
33
34 typedef struct ailayer_input ailayer_input_t;
35
39 struct ailayer_input {
40 ailayer_t base;
48 uint8_t input_dim;
49 uint16_t *input_shape;
51 };
52
58 extern const aicore_layertype_t *ailayer_input_type;
59
70 ailayer_t *ailayer_input(ailayer_input_t *layer);
71
82 void ailayer_input_forward(ailayer_t *self);
83
94 void ailayer_input_backward(ailayer_t *self);
95
105 void ailayer_input_calc_result_shape(ailayer_t *self);
106
107 #ifdef AIDEBUG_PRINT_MODULE_SPECS
112 void ailayer_input_print_specs(const ailayer_t *self);
113 #endif // AIDEBUG_PRINT_MODULE_SPECS
114
115 #endif // INPUT_LAYER
aifes_core.h
AIfES 2 core interface.
ailayer_input_type
const aicore_layertype_t * ailayer_input_type
Input layer type.
ailayer_input_backward
void ailayer_input_backward(ailayer_t *self)
Calculate the backward pass for the given Input layer.
ailayer_input_forward
void ailayer_input_forward(ailayer_t *self)
Calculate the forward pass for given Input layer.
ailayer_input_print_specs
void ailayer_input_print_specs(const ailayer_t *self)
Print the layer specification.
ailayer_input_calc_result_shape
void ailayer_input_calc_result_shape(ailayer_t *self)
Calculate the shape of the result tensor.
ailayer_input
ailayer_t * ailayer_input(ailayer_input_t *layer)
Initialize the given Input layer.
aicore_layertype
Type indicator of the layer.
Definition: aifes_core.h:82
ailayer_input
General Input layer structure.
Definition: ailayer_input.h:39
ailayer_input::input_shape
uint16_t * input_shape
Shape of the input tensor.
Definition: ailayer_input.h:49
ailayer_input::input_dim
uint8_t input_dim
Dimension of the input tensor.
Definition: ailayer_input.h:48
ailayer_input::base
ailayer_t base
Inherited field members from general ailayer struct.
Definition: ailayer_input.h:40
ailayer
AIfES layer interface.
Definition: aifes_core.h:252