IPoolingLayer is an optional interface implemented by pooling layers.
More...
#include <nvneural/layers/IPoolingLayer.h>
|
virtual NeuralResult | getKernelSize (std::size_t *pKernelW, std::size_t *pKernelH) const noexcept=0 |
| Retrieves the kernel size for the W and H axes. More...
|
|
virtual NeuralResult | getPadding (std::size_t *pStartW, std::size_t *pStartH, std::size_t *pEndW, std::size_t *pEndH) const noexcept=0 |
| Retrieves the start and end padding for the W and H axes. More...
|
|
virtual NeuralResult | getStride (std::size_t *pStrideW, std::size_t *pStrideH) const noexcept=0 |
| Retrieves the operation stride for the W and H axes. More...
|
|
virtual PoolType | poolingType () const noexcept=0 |
| Returns the type of pooling operation performed by this layer.
|
|
virtual NeuralResult | setEndPadding (std::size_t endW, std::size_t endH) noexcept=0 |
| Sets the end padding along the W and H axes. More...
|
|
virtual NeuralResult | setKernelSize (std::size_t kernelW, std::size_t kernelH) noexcept=0 |
| Sets the kernel size for the W and H axes. More...
|
|
virtual NeuralResult | setPoolingType (PoolType poolType) noexcept=0 |
| Sets the type of pooling operation. More...
|
|
virtual NeuralResult | setStartPadding (std::size_t startW, std::size_t startH) noexcept=0 |
| Sets the start padding along the W and H axes. More...
|
|
virtual NeuralResult | setStride (std::size_t strideW, std::size_t strideH) noexcept=0 |
| Sets the operation stride for the W and H axes. More...
|
|
virtual RefCount | addRef () const noexcept=0 |
| Increments the object's reference count. More...
|
|
virtual const void * | queryInterface (TypeId interface) const noexcept=0 |
| This is an overloaded member function, provided for convenience. It differs from the above function only in what argument(s) it accepts.
|
|
virtual void * | queryInterface (TypeId interface) noexcept=0 |
| Retrieves a new object interface pointer. More...
|
|
virtual RefCount | release () const noexcept=0 |
| Decrements the object's reference count and destroy the object if the reference count reaches zero. More...
|
|
|
using | RefCount = std::uint32_t |
| Typedef used to track the number of active references to an object.
|
|
using | TypeId = std::uint64_t |
| Every interface must define a unique TypeId. This should be randomized.
|
|
virtual | ~IRefObject ()=default |
| A protected destructor prevents accidental stack-allocation of IRefObjects or use with other smart pointer classes like std::unique_ptr.
|
|
IPoolingLayer is an optional interface implemented by pooling layers.
◆ getKernelSize()
virtual NeuralResult nvneural::IPoolingLayer::getKernelSize |
( |
std::size_t * |
pKernelW, |
|
|
std::size_t * |
pKernelH |
|
) |
| const |
|
pure virtualnoexcept |
Retrieves the kernel size for the W and H axes.
- Parameters
-
pKernelW | Optional pointer to a variable receiving the current kernel size for W |
pKernelH | Optional pointer to a variable receiving the current kernel size for H |
◆ getPadding()
virtual NeuralResult nvneural::IPoolingLayer::getPadding |
( |
std::size_t * |
pStartW, |
|
|
std::size_t * |
pStartH, |
|
|
std::size_t * |
pEndW, |
|
|
std::size_t * |
pEndH |
|
) |
| const |
|
pure virtualnoexcept |
Retrieves the start and end padding for the W and H axes.
Zero values indicate no padding for that component.
- Parameters
-
pStartW | Optional pointer to a variable receiving the current start padding for W |
pStartH | Optional pointer to a variable receiving the current start padding for H |
pEndW | Optional pointer to a variable receiving the current end padding for W |
pEndH | Optional pointer to a variable receiving the current end padding for H |
◆ getStride()
virtual NeuralResult nvneural::IPoolingLayer::getStride |
( |
std::size_t * |
pStrideW, |
|
|
std::size_t * |
pStrideH |
|
) |
| const |
|
pure virtualnoexcept |
Retrieves the operation stride for the W and H axes.
- Parameters
-
pStrideW | Optional pointer to a variable receiving the current stride for W |
pStrideH | Optional pointer to a variable receiving the current stride for H |
◆ setEndPadding()
virtual NeuralResult nvneural::IPoolingLayer::setEndPadding |
( |
std::size_t |
endW, |
|
|
std::size_t |
endH |
|
) |
| |
|
pure virtualnoexcept |
Sets the end padding along the W and H axes.
Zero values indicate no padding for that component.
- Parameters
-
endW | End padding for the W axis |
endH | End padding for the H axis |
◆ setKernelSize()
virtual NeuralResult nvneural::IPoolingLayer::setKernelSize |
( |
std::size_t |
kernelW, |
|
|
std::size_t |
kernelH |
|
) |
| |
|
pure virtualnoexcept |
Sets the kernel size for the W and H axes.
- Parameters
-
kernelW | Kernel size for the W axis |
kernelH | Kernel size for the H axis |
◆ setPoolingType()
Sets the type of pooling operation.
- Parameters
-
poolType | Pooling operation to perform |
◆ setStartPadding()
virtual NeuralResult nvneural::IPoolingLayer::setStartPadding |
( |
std::size_t |
startW, |
|
|
std::size_t |
startH |
|
) |
| |
|
pure virtualnoexcept |
Sets the start padding along the W and H axes.
Zero values indicate no padding for that component.
- Parameters
-
startW | Start padding for the W axis |
startH | Start padding for the H axis |
◆ setStride()
virtual NeuralResult nvneural::IPoolingLayer::setStride |
( |
std::size_t |
strideW, |
|
|
std::size_t |
strideH |
|
) |
| |
|
pure virtualnoexcept |
Sets the operation stride for the W and H axes.
- Parameters
-
strideW | Stride for the W axis |
strideH | Stride for the H axis |
The documentation for this class was generated from the following file: