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)=0
	Application-implemented interface to compute the HW output dimensions of a layer from the layer input and parameters. More...

Detailed Description

Application-implemented interface to compute layer output sizes.

Member Function Documentation

virtual DimsHW nvinfer1::IOutputDimensionsFormula::compute	(	DimsHW	inputDims,
		DimsHW	kernelSize,
		DimsHW	stride,
		DimsHW	padding,
		DimsHW	dilation,
		const char *	layerName
	)

pure virtual

Application-implemented interface to compute the HW output dimensions of a layer from the layer input and parameters.

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.

The documentation for this class was generated from the following file: