|
enum | nvinfer1::ActivationType : int {
nvinfer1::ActivationType::kRELU = 0,
nvinfer1::ActivationType::kSIGMOID = 1,
nvinfer1::ActivationType::kTANH = 2,
nvinfer1::ActivationType::kLEAKY_RELU = 3,
nvinfer1::ActivationType::kELU = 4,
nvinfer1::ActivationType::kSELU = 5,
nvinfer1::ActivationType::kSOFTSIGN = 6,
nvinfer1::ActivationType::kSOFTPLUS = 7,
nvinfer1::ActivationType::kCLIP = 8,
nvinfer1::ActivationType::kHARD_SIGMOID = 9,
nvinfer1::ActivationType::kSCALED_TANH = 10,
nvinfer1::ActivationType::kTHRESHOLDED_RELU = 11
} |
| Forward declare IGpuAllocator for use in other interfaces. More...
|
|
enum | nvinfer1::DataType : int {
nvinfer1::DataType::kFLOAT = 0,
nvinfer1::DataType::kHALF = 1,
nvinfer1::DataType::kINT8 = 2,
nvinfer1::DataType::kINT32 = 3,
nvinfer1::DataType::kBOOL = 4
} |
| The type of weights and tensors. More...
|
|
enum | nvinfer1::DimensionType : int { nvinfer1::DimensionType::kSPATIAL = 0,
nvinfer1::DimensionType::kCHANNEL = 1,
nvinfer1::DimensionType::kINDEX = 2,
nvinfer1::DimensionType::kSEQUENCE = 3
} |
| The type of data encoded across this dimension. More...
|
|
enum | nvinfer1::TensorFormat : int {
nvinfer1::TensorFormat::kLINEAR = 0,
TRT_DEPRECATED_ENUM = kLINEAR,
nvinfer1::TensorFormat::kCHW2 = 1,
TRT_DEPRECATED_ENUM = kCHW2,
nvinfer1::TensorFormat::kHWC8 = 2,
TRT_DEPRECATED_ENUM = kHWC8,
nvinfer1::TensorFormat::kCHW4 = 3,
nvinfer1::TensorFormat::kCHW16 = 4,
nvinfer1::TensorFormat::kCHW32 = 5
} |
| Format of the input/output tensors. More...
|
|
enum | nvinfer1::PluginVersion : uint8_t { kV2 = 0,
nvinfer1::PluginVersion::kV2_EXT = 1,
nvinfer1::PluginVersion::kV2_IOEXT = 2,
nvinfer1::PluginVersion::kV2_DYNAMICEXT = 3
} |
|
enum | nvinfer1::PluginFieldType : int {
nvinfer1::PluginFieldType::kFLOAT16 = 0,
nvinfer1::PluginFieldType::kFLOAT32 = 1,
nvinfer1::PluginFieldType::kFLOAT64 = 2,
nvinfer1::PluginFieldType::kINT8 = 3,
nvinfer1::PluginFieldType::kINT16 = 4,
nvinfer1::PluginFieldType::kINT32 = 5,
nvinfer1::PluginFieldType::kCHAR = 6,
nvinfer1::PluginFieldType::kDIMS = 7,
kUNKNOWN = 8
} |
|
enum | nvinfer1::TensorLocation : int { nvinfer1::TensorLocation::kDEVICE = 0,
nvinfer1::TensorLocation::kHOST = 1
} |
| The location for tensor data storage, device or host. More...
|
|
enum | nvinfer1::ErrorCode : int {
nvinfer1::ErrorCode::kSUCCESS = 0,
nvinfer1::ErrorCode::kUNSPECIFIED_ERROR = 1,
nvinfer1::ErrorCode::kINTERNAL_ERROR = 2,
nvinfer1::ErrorCode::kINVALID_ARGUMENT = 3,
nvinfer1::ErrorCode::kINVALID_CONFIG = 4,
nvinfer1::ErrorCode::kFAILED_ALLOCATION = 5,
nvinfer1::ErrorCode::kFAILED_INITIALIZATION = 6,
nvinfer1::ErrorCode::kFAILED_EXECUTION = 7,
nvinfer1::ErrorCode::kFAILED_COMPUTATION = 8,
nvinfer1::ErrorCode::kINVALID_STATE = 9,
nvinfer1::ErrorCode::kUNSUPPORTED_STATE = 10
} |
| Error codes that can be returned by TensorRT during execution. More...
|
|
This is the top-level API file for TensorRT core runtime library.