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
{
43
ailayer_t
base
;
44
const
aimath_dtype_t
*
dtype
;
52
uint8_t
input_dim
;
53
uint16_t *
input_shape
;
55
};
56
62
extern
const
aicore_layertype_t
*
ailayer_input_type
;
63
74
ailayer_t
*
ailayer_input
(
ailayer_input_t
*layer);
75
86
void
ailayer_input_forward
(
ailayer_t
*
self
);
87
98
void
ailayer_input_backward
(
ailayer_t
*
self
);
99
109
void
ailayer_input_calc_result_shape
(
ailayer_t
*
self
);
110
111
#ifdef AIDEBUG_PRINT_MODULE_SPECS
117
void
ailayer_input_print_specs
(
const
ailayer_t
*
self
);
118
#endif
// AIDEBUG_PRINT_MODULE_SPECS
119
120
#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
ailayer_t * ailayer_input(ailayer_input_t *layer)
Initialize 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.
aicore_layertype
Type indicator of the layer.
Definition:
aifes_core.h:82
ailayer_input
General Input layer structure.
Definition:
ailayer_input.h:42
ailayer_input::input_shape
uint16_t * input_shape
Shape of the input tensor.
Definition:
ailayer_input.h:53
ailayer_input::input_dim
uint8_t input_dim
Dimension of the input tensor.
Definition:
ailayer_input.h:52
ailayer_input::base
ailayer_t base
Inherited field members from general ailayer struct.
Definition:
ailayer_input.h:43
ailayer_input::dtype
const aimath_dtype_t * dtype
Data type of the input and inference result values.
Definition:
ailayer_input.h:44
ailayer
AIfES layer interface.
Definition:
aifes_core.h:252
aimath_dtype
Indicator for the used datatype.
Definition:
aifes_math.h:47
include
basic
base
ailayer
ailayer_input.h
Generated by
1.9.2