|
enum | nvinfer1::EngineCapability : int32_t { nvinfer1::EngineCapability::kDEFAULT = 0,
nvinfer1::EngineCapability::kSAFE_GPU = 1,
nvinfer1::EngineCapability::kSAFE_DLA = 2
} |
| List of supported engine capability flows. More...
|
|
enum | nvinfer1::DimensionOperation : int32_t {
nvinfer1::DimensionOperation::kSUM = 0,
nvinfer1::DimensionOperation::kPROD = 1,
nvinfer1::DimensionOperation::kMAX = 2,
nvinfer1::DimensionOperation::kMIN = 3,
nvinfer1::DimensionOperation::kSUB = 4,
nvinfer1::DimensionOperation::kEQUAL = 5,
nvinfer1::DimensionOperation::kLESS = 6,
nvinfer1::DimensionOperation::kFLOOR_DIV = 7,
nvinfer1::DimensionOperation::kCEIL_DIV = 8
} |
| An operation on two IDimensionExpr, which represent integer expressions used in dimension computations. More...
|
|
enum | nvinfer1::TensorLocation : int32_t { nvinfer1::TensorLocation::kDEVICE = 0,
nvinfer1::TensorLocation::kHOST = 1
} |
| The location for tensor data storage, device or host. More...
|
|
enum | nvinfer1::WeightsRole : int32_t {
nvinfer1::WeightsRole::kKERNEL = 0,
nvinfer1::WeightsRole::kBIAS = 1,
nvinfer1::WeightsRole::kSHIFT = 2,
nvinfer1::WeightsRole::kSCALE = 3,
nvinfer1::WeightsRole::kCONSTANT = 4,
nvinfer1::WeightsRole::kANY = 5
} |
| How a layer uses particular Weights. More...
|
|
enum | nvinfer1::DeviceType : int32_t { nvinfer1::DeviceType::kGPU,
nvinfer1::DeviceType::kDLA
} |
| The device that this layer/network will execute on. More...
|
|
enum | nvinfer1::OptProfileSelector : int32_t { nvinfer1::OptProfileSelector::kMIN = 0,
nvinfer1::OptProfileSelector::kOPT = 1,
nvinfer1::OptProfileSelector::kMAX = 2
} |
| When setting or querying optimization profile parameters (such as shape tensor inputs or dynamic dimensions), select whether we are interested in the minimum, optimum, or maximum values for these parameters. The minimum and maximum specify the permitted range that is supported at runtime, while the optimum value is used for the kernel selection. This should be the "typical" value that is expected to occur at runtime. More...
|
|
enum | nvinfer1::TacticSource : int32_t { nvinfer1::TacticSource::kCUBLAS = 0,
nvinfer1::TacticSource::kCUBLAS_LT = 1,
nvinfer1::TacticSource::kCUDNN = 2
} |
| List of tactic sources for TensorRT. More...
|
|
This is the top-level API file for TensorRT extended runtime library.