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
;
41
const
aimath_dtype_t
*
dtype
;
49
uint8_t
input_dim
;
50
uint16_t *
input_shape
;
52
};
53
59
extern
const
aicore_layertype_t
*
ailayer_input_type
;
60
71
ailayer_t
*
ailayer_input
(
ailayer_input_t
*layer);
72
83
void
ailayer_input_forward
(
ailayer_t
*
self
);
84
95
void
ailayer_input_backward
(
ailayer_t
*
self
);
96
106
void
ailayer_input_calc_result_shape
(
ailayer_t
*
self
);
107
108
#ifdef AIDEBUG_PRINT_MODULE_SPECS
113
void
ailayer_input_print_specs
(
const
ailayer_t
*
self
);
114
#endif
// AIDEBUG_PRINT_MODULE_SPECS
115
116
#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:79
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:50
ailayer_input::input_dim
uint8_t input_dim
Dimension of the input tensor.
Definition:
ailayer_input.h:49
ailayer_input::base
ailayer_t base
Inherited field members from general ailayer struct.
Definition:
ailayer_input.h:40
ailayer_input::dtype
const aimath_dtype_t * dtype
Data type of the input and inference result values.
Definition:
ailayer_input.h:41
ailayer
AIfES layer interface.
Definition:
aifes_core.h:249
aimath_dtype
Indicator for the used datatype.
Definition:
aifes_math.h:44
include
basic
base
ailayer
ailayer_input.h
Generated by
1.9.1