48 #ifndef __STOCHASTIC_POOLING2D_LAYER_TYPES_H__ 49 #define __STOCHASTIC_POOLING2D_LAYER_TYPES_H__ 51 #include "algorithms/algorithm.h" 52 #include "data_management/data/tensor.h" 53 #include "data_management/data/homogen_tensor.h" 54 #include "services/daal_defines.h" 55 #include "algorithms/neural_networks/layers/pooling2d/pooling2d_layer_types.h" 61 namespace neural_networks
71 namespace stochastic_pooling2d
89 lastLayerDataId = auxSelectedIndices
96 enum LayerDataNumericTableId
98 auxInputDimensions = lastLayerDataId + 1,
100 lastLayerDataNumericTableId = auxInputDimensions
115 struct DAAL_EXPORT Parameter:
public pooling2d::Parameter
130 Parameter(
size_t firstIndex,
size_t secondIndex,
size_t firstKernelSize = 2,
size_t secondKernelSize = 2,
131 size_t firstStride = 2,
size_t secondStride = 2,
size_t firstPadding = 0,
size_t secondPadding = 0);
134 engines::EnginePtr engine;
139 using interface1::Parameter;
daal::algorithms::neural_networks::layers::stochastic_pooling2d::Method
Method
Computation methods for the stochastic 2D pooling layer.
Definition: stochastic_pooling2d_layer_types.h:77
daal
Definition: algorithm_base_common.h:57
daal::algorithms::neural_networks::layers::stochastic_pooling2d::auxInputDimensions
Definition: stochastic_pooling2d_layer_types.h:98
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter::engine
engines::EnginePtr engine
Definition: stochastic_pooling2d_layer_types.h:134
daal::algorithms::neural_networks::layers::stochastic_pooling2d::auxSelectedIndices
Definition: stochastic_pooling2d_layer_types.h:88
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter::seed
size_t seed
Definition: stochastic_pooling2d_layer_types.h:133
daal::algorithms::neural_networks::layers::pooling2d::interface1::Parameter
Parameters for the forward and backward two-dimensional pooling layers.
Definition: pooling2d_layer_types.h:152
daal::algorithms::neural_networks::layers::stochastic_pooling2d::defaultDense
Definition: stochastic_pooling2d_layer_types.h:79
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:96
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:86
daal::algorithms::neural_networks::layers::stochastic_pooling2d::interface1::Parameter
Parameters for the stochastic 2D pooling layer.
Definition: stochastic_pooling2d_layer_types.h:115