|
| enum | nvinfer1::DataType : int { nvinfer1::DataType::kFLOAT = 0,
nvinfer1::DataType::kHALF = 1,
nvinfer1::DataType::kINT8 = 2,
nvinfer1::DataType::kINT32 = 3
} |
| | The type of weights and tensors. More...
|
| |
| enum | nvinfer1::DeviceType : int { nvinfer1::DeviceType::kGPU,
nvinfer1::DeviceType::kDLA
} |
| | The device that this layer/network will execute on. 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::LayerType : int {
nvinfer1::LayerType::kCONVOLUTION = 0,
nvinfer1::LayerType::kFULLY_CONNECTED = 1,
nvinfer1::LayerType::kACTIVATION = 2,
nvinfer1::LayerType::kPOOLING = 3,
nvinfer1::LayerType::kLRN = 4,
nvinfer1::LayerType::kSCALE = 5,
nvinfer1::LayerType::kSOFTMAX = 6,
nvinfer1::LayerType::kDECONVOLUTION = 7,
nvinfer1::LayerType::kCONCATENATION = 8,
nvinfer1::LayerType::kELEMENTWISE = 9,
nvinfer1::LayerType::kPLUGIN = 10,
nvinfer1::LayerType::kRNN = 11,
nvinfer1::LayerType::kUNARY = 12,
nvinfer1::LayerType::kPADDING = 13,
nvinfer1::LayerType::kSHUFFLE = 14,
nvinfer1::LayerType::kREDUCE = 15,
nvinfer1::LayerType::kTOPK = 16,
nvinfer1::LayerType::kGATHER = 17,
nvinfer1::LayerType::kMATRIX_MULTIPLY = 18,
nvinfer1::LayerType::kRAGGED_SOFTMAX = 19,
nvinfer1::LayerType::kCONSTANT = 20,
nvinfer1::LayerType::kRNN_V2 = 21,
nvinfer1::LayerType::kIDENTITY = 22,
nvinfer1::LayerType::kPLUGIN_V2 = 23,
nvinfer1::LayerType::kSLICE = 24
} |
| | The type values of layer classes. More...
|
| |
| enum | nvinfer1::TensorLocation : int { nvinfer1::TensorLocation::kDEVICE = 0,
nvinfer1::TensorLocation::kHOST = 1
} |
| | The location for tensor data storage, device or host. More...
|
| |
| enum | nvinfer1::PaddingMode : int {
nvinfer1::PaddingMode::kEXPLICIT_ROUND_DOWN = 0,
nvinfer1::PaddingMode::kEXPLICIT_ROUND_UP = 1,
nvinfer1::PaddingMode::kSAME_UPPER = 2,
nvinfer1::PaddingMode::kSAME_LOWER = 3,
nvinfer1::PaddingMode::kCAFFE_ROUND_DOWN = 4,
nvinfer1::PaddingMode::kCAFFE_ROUND_UP = 5
} |
| | Enumerates the modes of padding to perform in convolution, deconvolution and pooling layer, padding mode gets precedence if setPaddingMode() and setPrePadding() are also used. More...
|
| |
| 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
} |
| | Enumerates the types of activation to perform in an activation layer. More...
|
| |
| enum | nvinfer1::PoolingType : int { kMAX = 0,
kAVERAGE = 1,
kMAX_AVERAGE_BLEND = 2
} |
| | The type of pooling to perform in a pooling layer.
|
| |
| enum | nvinfer1::ScaleMode : int { nvinfer1::ScaleMode::kUNIFORM = 0,
nvinfer1::ScaleMode::kCHANNEL = 1,
nvinfer1::ScaleMode::kELEMENTWISE = 2
} |
| | Controls how shift, scale and power are applied in a Scale layer. More...
|
| |
| enum | nvinfer1::ElementWiseOperation : int {
nvinfer1::ElementWiseOperation::kSUM = 0,
nvinfer1::ElementWiseOperation::kPROD = 1,
nvinfer1::ElementWiseOperation::kMAX = 2,
nvinfer1::ElementWiseOperation::kMIN = 3,
nvinfer1::ElementWiseOperation::kSUB = 4,
nvinfer1::ElementWiseOperation::kDIV = 5,
nvinfer1::ElementWiseOperation::kPOW = 6
} |
| | Enumerates the binary operations that may be performed by an ElementWise layer. More...
|
| |
| enum | nvinfer1::RNNOperation : int { nvinfer1::RNNOperation::kRELU = 0,
nvinfer1::RNNOperation::kTANH = 1,
nvinfer1::RNNOperation::kLSTM = 2,
nvinfer1::RNNOperation::kGRU = 3
} |
| | Enumerates the RNN operations that may be performed by an RNN layer. More...
|
| |
| enum | nvinfer1::RNNDirection : int { nvinfer1::RNNDirection::kUNIDIRECTION = 0,
nvinfer1::RNNDirection::kBIDIRECTION = 1
} |
| | Enumerates the RNN direction that may be performed by an RNN layer. More...
|
| |
| enum | nvinfer1::RNNInputMode : int { nvinfer1::RNNInputMode::kLINEAR = 0,
nvinfer1::RNNInputMode::kSKIP = 1
} |
| | Enumerates the RNN input modes that may occur with an RNN layer. More...
|
| |
| enum | nvinfer1::RNNGateType : int {
nvinfer1::RNNGateType::kINPUT = 0,
nvinfer1::RNNGateType::kOUTPUT = 1,
nvinfer1::RNNGateType::kFORGET = 2,
nvinfer1::RNNGateType::kUPDATE = 3,
nvinfer1::RNNGateType::kRESET = 4,
nvinfer1::RNNGateType::kCELL = 5,
nvinfer1::RNNGateType::kHIDDEN = 6
} |
| | Identifies an individual gate within an RNN cell. More...
|
| |
| enum | nvinfer1::PluginFormat : uint8_t { nvinfer1::PluginFormat::kNCHW = 0,
nvinfer1::PluginFormat::kNC2HW2 = 1,
nvinfer1::PluginFormat::kNHWC8 = 2
} |
| |
| 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::UnaryOperation : int {
nvinfer1::UnaryOperation::kEXP = 0,
nvinfer1::UnaryOperation::kLOG = 1,
nvinfer1::UnaryOperation::kSQRT = 2,
nvinfer1::UnaryOperation::kRECIP = 3,
nvinfer1::UnaryOperation::kABS = 4,
nvinfer1::UnaryOperation::kNEG = 5,
nvinfer1::UnaryOperation::kSIN = 6,
nvinfer1::UnaryOperation::kCOS = 7,
nvinfer1::UnaryOperation::kTAN = 8,
nvinfer1::UnaryOperation::kSINH = 9,
nvinfer1::UnaryOperation::kCOSH = 10,
nvinfer1::UnaryOperation::kASIN = 11,
nvinfer1::UnaryOperation::kACOS = 12,
nvinfer1::UnaryOperation::kATAN = 13,
nvinfer1::UnaryOperation::kASINH = 14,
nvinfer1::UnaryOperation::kACOSH = 15,
nvinfer1::UnaryOperation::kATANH = 16,
nvinfer1::UnaryOperation::kCEIL = 17,
nvinfer1::UnaryOperation::kFLOOR = 18
} |
| | Enumerates the unary operations that may be performed by a Unary layer. More...
|
| |
| enum | nvinfer1::ReduceOperation : int {
kSUM = 0,
kPROD = 1,
kMAX = 2,
kMIN = 3,
kAVG = 4
} |
| | Enumerates the reduce operations that may be performed by a Reduce layer.
|
| |
| enum | nvinfer1::TopKOperation : int { nvinfer1::TopKOperation::kMAX = 0,
nvinfer1::TopKOperation::kMIN = 1
} |
| | Enumerates the operations that may be performed by a TopK layer. More...
|
| |
| enum | nvinfer1::MatrixOperation : int { nvinfer1::MatrixOperation::kNONE,
nvinfer1::MatrixOperation::kTRANSPOSE,
nvinfer1::MatrixOperation::kVECTOR
} |
| | Enumerates the operations that may be performed on a tensor by IMatrixMultiplyLayer before multiplication. More...
|
| |
| enum | nvinfer1::CalibrationAlgoType : int { kLEGACY_CALIBRATION = 0,
kENTROPY_CALIBRATION = 1,
kENTROPY_CALIBRATION_2 = 2
} |
| | Version of calibration algorithm to use. More...
|
| |
| enum | nvinfer1::EngineCapability { nvinfer1::EngineCapability::kDEFAULT = 0,
nvinfer1::EngineCapability::kSAFE_GPU = 1,
nvinfer1::EngineCapability::kSAFE_DLA = 2
} |
| | List of supported engine capability flows. More...
|
| |
| enum | nvinfer1::WeightsRole : int {
nvinfer1::WeightsRole::kKERNEL = 0,
nvinfer1::WeightsRole::kBIAS = 1,
nvinfer1::WeightsRole::kSHIFT = 2,
nvinfer1::WeightsRole::kSCALE = 3,
nvinfer1::WeightsRole::kCONSTANT = 4
} |
| | How a layer uses particular Weights. More...
|
| |
This is the top-level API file for TensorRT.
1.8.6