|
virtual void | setPoolingType (PoolingType type)=0 |
| Set the type of activation to be performed. More...
|
|
virtual PoolingType | getPoolingType () const =0 |
| Get the type of activation to be performed. More...
|
|
virtual void | setWindowSize (DimsHW windowSize)=0 |
| Set the window size for pooling. More...
|
|
virtual DimsHW | getWindowSize () const =0 |
| Get the window size for pooling. More...
|
|
virtual void | setStride (DimsHW stride)=0 |
| Set the stride for pooling. More...
|
|
virtual DimsHW | getStride () const =0 |
| Get the stride for pooling. More...
|
|
virtual void | setPadding (DimsHW padding)=0 |
| Set the padding for pooling. More...
|
|
virtual DimsHW | getPadding () const =0 |
| Get the padding for pooling. More...
|
|
virtual void | setBlendFactor (float blendFactor)=0 |
| Set the blending factor for the max_average_blend mode: max_average_blendPool = (1-blendFactor)*maxPool + blendFactor*avgPool blendFactor is a user value in [0,1] with the default value of 0.0 This value only applies for the kMAX_AVERAGE_BLEND mode. More...
|
|
virtual float | getBlendFactor () const =0 |
| Get the blending factor for the max_average_blend mode: max_average_blendPool = (1-blendFactor)*maxPool + blendFactor*avgPool blendFactor is a user value in [0,1] with the default value of 0.0 In modes other than kMAX_AVERAGE_BLEND, blendFactor is ignored. More...
|
|
virtual void | setAverageCountExcludesPadding (bool exclusive)=0 |
| Set whether average pooling uses as a denominator the overlap area between the window and the unpadded input. If this is not set, the denominator is the overlap between the pooling window and the padded input. More...
|
|
virtual bool | getAverageCountExcludesPadding () const =0 |
| Get whether exclusive pooling uses as a denominator the overlap area betwen the window and the unpadded input. More...
|
|
virtual LayerType | getType () const =0 |
| Return the type of a layer. More...
|
|
virtual void | setName (const char *name)=0 |
| Set the name of a layer. More...
|
|
virtual const char * | getName () const =0 |
| Return the name of a layer. More...
|
|
virtual int | getNbInputs () const =0 |
| Get the number of inputs of a layer.
|
|
virtual ITensor * | getInput (int index) const =0 |
| Get the layer input corresponding to the given index. More...
|
|
virtual int | getNbOutputs () const =0 |
| Get the number of outputs of a layer.
|
|
virtual ITensor * | getOutput (int index) const =0 |
| Get the layer output corresponding to the given index. More...
|
|
A Pooling layer in a network definition.
The layer applies a reduction operation within a window over the input.
The output size is determined from the input size using the formula set by INetworkDefinition::setCustomPoolingDimensions().