NVIDIA DeepStream SDK API Reference
6.4 Release
Public Member Functions
YoloLayerV3 Class Reference

Detailed Description

Definition at line 50 of file yoloPlugins.h.

Inheritance diagram for YoloLayerV3:
Inheritance graph
[legend]
Collaboration diagram for YoloLayerV3:
Collaboration graph
[legend]

Public Member Functions

 YoloLayerV3 (const void *data, size_t length)
 
 YoloLayerV3 (const uint &numBoxes, const uint &numClasses, const uint &gridSize)
 
const char * getPluginType () const noexcept override
 
const char * getPluginVersion () const noexcept override
 
int getNbOutputs () const noexcept override
 
nvinfer1::Dims getOutputDimensions (int index, const nvinfer1::Dims *inputs, int nbInputDims) noexcept override
 
bool supportsFormat (nvinfer1::DataType type, nvinfer1::PluginFormat format) const noexcept override
 
void configureWithFormat (const nvinfer1::Dims *inputDims, int nbInputs, const nvinfer1::Dims *outputDims, int nbOutputs, nvinfer1::DataType type, nvinfer1::PluginFormat format, int maxBatchSize) noexcept override
 
int initialize () noexcept override
 
void terminate () noexcept override
 
size_t getWorkspaceSize (int maxBatchSize) const noexcept override
 
int32_t enqueue (int32_t batchSize, void const *const *inputs, void *const *outputs, void *workspace, cudaStream_t stream) noexcept override
 
size_t getSerializationSize () const noexcept override
 
void serialize (void *buffer) const noexcept override
 
void destroy () noexcept override
 
nvinfer1::IPluginV2 * clone () const noexcept override
 
void setPluginNamespace (const char *pluginNamespace) noexcept override
 
virtual const char * getPluginNamespace () const noexcept override
 

Constructor & Destructor Documentation

◆ YoloLayerV3() [1/2]

YoloLayerV3::YoloLayerV3 ( const void *  data,
size_t  length 
)

◆ YoloLayerV3() [2/2]

YoloLayerV3::YoloLayerV3 ( const uint &  numBoxes,
const uint &  numClasses,
const uint &  gridSize 
)

Member Function Documentation

◆ clone()

nvinfer1::IPluginV2* YoloLayerV3::clone ( ) const
overridenoexcept

◆ configureWithFormat()

void YoloLayerV3::configureWithFormat ( const nvinfer1::Dims *  inputDims,
int  nbInputs,
const nvinfer1::Dims *  outputDims,
int  nbOutputs,
nvinfer1::DataType  type,
nvinfer1::PluginFormat  format,
int  maxBatchSize 
)
overridenoexcept

◆ destroy()

void YoloLayerV3::destroy ( )
inlineoverridenoexcept

Definition at line 79 of file yoloPlugins.h.

◆ enqueue()

int32_t YoloLayerV3::enqueue ( int32_t  batchSize,
void const *const *  inputs,
void *const *  outputs,
void *  workspace,
cudaStream_t  stream 
)
overridenoexcept

◆ getNbOutputs()

int YoloLayerV3::getNbOutputs ( ) const
inlineoverridenoexcept

Definition at line 57 of file yoloPlugins.h.

◆ getOutputDimensions()

nvinfer1::Dims YoloLayerV3::getOutputDimensions ( int  index,
const nvinfer1::Dims *  inputs,
int  nbInputDims 
)
overridenoexcept

◆ getPluginNamespace()

virtual const char* YoloLayerV3::getPluginNamespace ( ) const
inlineoverridevirtualnoexcept

Definition at line 85 of file yoloPlugins.h.

◆ getPluginType()

const char* YoloLayerV3::getPluginType ( ) const
inlineoverridenoexcept

Definition at line 55 of file yoloPlugins.h.

◆ getPluginVersion()

const char* YoloLayerV3::getPluginVersion ( ) const
inlineoverridenoexcept

Definition at line 56 of file yoloPlugins.h.

◆ getSerializationSize()

size_t YoloLayerV3::getSerializationSize ( ) const
overridenoexcept

◆ getWorkspaceSize()

size_t YoloLayerV3::getWorkspaceSize ( int  maxBatchSize) const
inlineoverridenoexcept

Definition at line 73 of file yoloPlugins.h.

◆ initialize()

int YoloLayerV3::initialize ( )
inlineoverridenoexcept

Definition at line 71 of file yoloPlugins.h.

◆ serialize()

void YoloLayerV3::serialize ( void *  buffer) const
overridenoexcept

◆ setPluginNamespace()

void YoloLayerV3::setPluginNamespace ( const char *  pluginNamespace)
inlineoverridenoexcept

Definition at line 82 of file yoloPlugins.h.

◆ supportsFormat()

bool YoloLayerV3::supportsFormat ( nvinfer1::DataType  type,
nvinfer1::PluginFormat  format 
) const
overridenoexcept

◆ terminate()

void YoloLayerV3::terminate ( )
inlineoverridenoexcept

Definition at line 72 of file yoloPlugins.h.

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