An array of weights used as a layer parameter. More...
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... | |
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.
DataType nvinfer1::Weights::type |
The type of the weights.
const void* nvinfer1::Weights::values |
The weight values, in a contiguous array.
int64_t nvinfer1::Weights::count |
The number of weights in the array.