TensorRT
7.0.0.11
|
Application-implemented interface to compute layer output sizes. More...
#include <NvInfer.h>
Public Member Functions | |
virtual DimsHW | compute (DimsHW inputDims, DimsHW kernelSize, DimsHW stride, DimsHW padding, DimsHW dilation, const char *layerName) const =0 |
Application-implemented interface to compute the HW output dimensions of a layer from the layer input and parameters. More... | |
Application-implemented interface to compute layer output sizes.
|
pure virtual |
Application-implemented interface to compute the HW output dimensions of a layer from the layer input and parameters.
inputDims | The input dimensions of the layer. |
kernelSize | The kernel size (or window size, for a pooling layer) parameter of the layer operation. |
stride | The stride parameter for the layer. |
padding | The padding parameter of the layer. |
dilation | The dilation parameter of the layer (only applicable to convolutions). |
layerName | The name of the layer. |
Note that for dilated convolutions, the dilation is applied to the kernel size before this routine is called.