TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
nvinfer1::Weights Class Reference

An array of weights used as a layer parameter. More...

Collaboration diagram for nvinfer1::Weights:

Public Attributes

DataType type
 The type of the weights. More...
 
const void * values
 The weight values, in a contiguous array. More...
 
int64_t count
 The number of weights in the array. More...
 

Detailed Description

An array of weights used as a layer parameter.

When using the DLA, the cumulative size of all Weights used in a network must be less than 512MB in size. If the build option kGPU_FALLBACK is specified, then multiple DLA sub-networks may be generated from the single original network.

The weights are held by reference until the engine has been built. Therefore the data referenced by values field should be preserved until the build is complete.

Member Data Documentation

◆ type

DataType nvinfer1::Weights::type

The type of the weights.

◆ values

const void* nvinfer1::Weights::values

The weight values, in a contiguous array.

◆ count

int64_t nvinfer1::Weights::count

The number of weights in the array.


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