48 #ifndef __STOCHASTIC_POOLING2D_LAYER_H__ 49 #define __STOCHASTIC_POOLING2D_LAYER_H__ 51 #include "algorithms/algorithm.h" 52 #include "data_management/data/tensor.h" 53 #include "services/daal_defines.h" 54 #include "algorithms/neural_networks/layers/layer.h" 55 #include "algorithms/neural_networks/layers/pooling2d/stochastic_pooling2d_layer_types.h" 56 #include "algorithms/neural_networks/layers/pooling2d/stochastic_pooling2d_layer_forward.h" 57 #include "algorithms/neural_networks/layers/pooling2d/stochastic_pooling2d_layer_backward.h" 64 namespace neural_networks
71 namespace stochastic_pooling2d
99 template<
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
100 class Batch :
public LayerIface
109 Batch(
size_t nDimensions) : parameter(nDimensions - 2, nDimensions - 1)
111 forward::Batch<algorithmFPType, method> *forwardLayerObject =
new forward::Batch<algorithmFPType, method>(parameter);
112 backward::Batch<algorithmFPType, method> *backwardLayerObject =
new backward::Batch<algorithmFPType, method>(parameter);
114 LayerIface::forwardLayer = services::SharedPtr<forward::Batch<algorithmFPType, method> >(forwardLayerObject);
115 LayerIface::backwardLayer = services::SharedPtr<backward::Batch<algorithmFPType, method> >(backwardLayerObject);
120 using interface1::Batch;
daal
Definition: algorithm_base_common.h:57
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Batch
Provides methods for the stochastic 2D pooling layer in the batch processing mode.
Definition: stochastic_pooling2d_layer.h:100
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Batch::Batch
Batch(size_t nDimensions)
Definition: stochastic_pooling2d_layer.h:109
daal::algorithms::neural_networks::layers::interface1::LayerIface
Abstract class that specifies the interface of layer.
Definition: layer.h:83
daal::services::interface1::SharedPtr
Shared pointer that retains shared ownership of an object through a pointer. Several SharedPtr object...
Definition: daal_shared_ptr.h:187
daal::algorithms::neural_networks::layers::interface1::LayerIface::backwardLayer
backward::LayerIfacePtr backwardLayer
Definition: layer.h:87
daal::algorithms::neural_networks::layers::stochastic_pooling2d::backward::interface1::Batch
Provides methods for the backward stochastic 2D pooling layer in the batch processing mode...
Definition: stochastic_pooling2d_layer_backward.h:137
daal::algorithms::neural_networks::layers::interface1::LayerIface::forwardLayer
forward::LayerIfacePtr forwardLayer
Definition: layer.h:86
daal::algorithms::neural_networks::layers::stochastic_pooling2d::forward::interface1::Batch
Provides methods for the forward stochastic 2D pooling layer in the batch processing mode...
Definition: stochastic_pooling2d_layer_forward.h:130
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Batch::parameter
Parameter parameter
Definition: stochastic_pooling2d_layer.h:103
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter
Parameters for the stochastic 2D pooling layer.
Definition: stochastic_pooling2d_layer_types.h:115