22 #ifndef __STOCHASTIC_POOLING2D_LAYER_TYPES_H__
23 #define __STOCHASTIC_POOLING2D_LAYER_TYPES_H__
25 #include "algorithms/algorithm.h"
26 #include "data_management/data/tensor.h"
27 #include "data_management/data/homogen_tensor.h"
28 #include "services/daal_defines.h"
29 #include "algorithms/neural_networks/layers/pooling2d/pooling2d_layer_types.h"
35 namespace neural_networks
45 namespace stochastic_pooling2d
63 lastLayerDataId = auxSelectedIndices
70 enum LayerDataNumericTableId
72 auxInputDimensions = lastLayerDataId + 1,
74 lastLayerDataNumericTableId = auxInputDimensions
89 struct DAAL_EXPORT Parameter:
public pooling2d::Parameter
104 Parameter(
size_t firstIndex,
size_t secondIndex,
size_t firstKernelSize = 2,
size_t secondKernelSize = 2,
105 size_t firstStride = 2,
size_t secondStride = 2,
size_t firstPadding = 0,
size_t secondPadding = 0);
108 engines::EnginePtr engine;
113 using interface1::Parameter;
daal
Definition: algorithm_base_common.h:31
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter::engine
engines::EnginePtr engine
Definition: stochastic_pooling2d_layer_types.h:108
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter::seed
size_t seed
Definition: stochastic_pooling2d_layer_types.h:107
daal::algorithms::neural_networks::layers::stochastic_pooling2d::auxSelectedIndices
Definition: stochastic_pooling2d_layer_types.h:62
daal::algorithms::neural_networks::layers::stochastic_pooling2d::Method
Method
Computation methods for the stochastic 2D pooling layer.
Definition: stochastic_pooling2d_layer_types.h:51
daal::algorithms::neural_networks::layers::stochastic_pooling2d::auxInputDimensions
Definition: stochastic_pooling2d_layer_types.h:72
daal::algorithms::neural_networks::layers::stochastic_pooling2d::LayerDataId
LayerDataId
Identifiers of input tensors for the backward stochastic 2D pooling layer and results for the forward...
Definition: stochastic_pooling2d_layer_types.h:60
daal::algorithms::neural_networks::layers::stochastic_pooling2d::defaultDense
Definition: stochastic_pooling2d_layer_types.h:53
daal::algorithms::neural_networks::layers::stochastic_pooling2d::LayerDataNumericTableId
LayerDataNumericTableId
Identifiers of input numeric tables for the backward stochastic 2D pooling layer and results for the ...
Definition: stochastic_pooling2d_layer_types.h:70
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter
Parameters for the stochastic 2D pooling layer.
Definition: stochastic_pooling2d_layer_types.h:89