AIfES 2  2.0.0
ailayer_input.h
Go to the documentation of this file.
1 
32 #ifndef INPUT_LAYER
33 #define INPUT_LAYER
34 
35 #include "core/aifes_core.h"
36 
37 typedef struct ailayer_input ailayer_input_t;
38 
42 struct ailayer_input {
52  uint8_t input_dim;
53  uint16_t *input_shape;
55 };
56 
63 
75 
87 
99 
110 
111 #ifdef AIDEBUG_PRINT_MODULE_SPECS
117 void ailayer_input_print_specs(const ailayer_t *self, int (*print)(const char *format, ...));
118 #endif // AIDEBUG_PRINT_MODULE_SPECS
119 
120 #endif // INPUT_LAYER
AIfES 2 core interface.
const aicore_layertype_t * ailayer_input_type
Input layer type.
void ailayer_input_print_specs(const ailayer_t *self, int(*print)(const char *format,...))
Print the layer specification.
void ailayer_input_backward(ailayer_t *self)
Calculate the backward pass for the given Input layer.
void ailayer_input_forward(ailayer_t *self)
Calculate the forward pass for given Input layer.
void ailayer_input_calc_result_shape(ailayer_t *self)
Calculate the shape of the result tensor.
ailayer_t * ailayer_input(ailayer_input_t *layer)
Initialize the given Input layer.
Type indicator of the layer.
Definition: aifes_core.h:81
General Input layer structure.
Definition: ailayer_input.h:42
uint16_t * input_shape
Shape of the input tensor.
Definition: ailayer_input.h:53
uint8_t input_dim
Dimension of the input tensor.
Definition: ailayer_input.h:52
ailayer_t base
Inherited field members from general ailayer struct.
Definition: ailayer_input.h:43
const aimath_dtype_t * dtype
Data type of the input and inference result values.
Definition: ailayer_input.h:44
AIfES layer interface.
Definition: aifes_core.h:251
Indicator for the used datatype.
Definition: aifes_math.h:53