22 #ifndef __FULLYCONNECTED_LAYER_H__
23 #define __FULLYCONNECTED_LAYER_H__
25 #include "algorithms/algorithm.h"
26 #include "data_management/data/tensor.h"
27 #include "services/daal_defines.h"
28 #include "algorithms/neural_networks/layers/layer.h"
29 #include "algorithms/neural_networks/layers/fullyconnected/fullyconnected_layer_types.h"
30 #include "algorithms/neural_networks/layers/fullyconnected/fullyconnected_layer_forward.h"
31 #include "algorithms/neural_networks/layers/fullyconnected/fullyconnected_layer_backward.h"
37 namespace neural_networks
44 namespace fullyconnected
68 template<
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
69 class Batch :
public LayerIface
76 Batch(
size_t nOutputs): parameter(nOutputs)
78 forward::Batch<algorithmFPType, method> *forwardLayerObject =
new forward::Batch<algorithmFPType, method>(parameter);
79 backward::Batch<algorithmFPType, method> *backwardLayerObject =
new backward::Batch<algorithmFPType, method>(parameter);
81 LayerIface::forwardLayer = services::SharedPtr<forward::Batch<algorithmFPType, method> >(forwardLayerObject);
82 LayerIface::backwardLayer = services::SharedPtr<backward::Batch<algorithmFPType, method> >(backwardLayerObject);
87 using interface1::Batch;
daal::algorithms::neural_networks::layers::fullyconnected::interface1::Parameter
Fully-connected layer parameters.
Definition: fullyconnected_layer_types.h:73
daal
Definition: algorithm_base_common.h:31
daal::algorithms::neural_networks::layers::fullyconnected::interface1::Batch
Computes the result of the forward and backward fully-connected layer of neural network in the batch ...
Definition: fullyconnected_layer.h:69
daal::algorithms::neural_networks::layers::fullyconnected::interface1::Batch::Batch
Batch(size_t nOutputs)
Definition: fullyconnected_layer.h:76
daal::algorithms::neural_networks::layers::fullyconnected::interface1::Batch::parameter
Parameter parameter
Definition: fullyconnected_layer.h:72