|
enum | nvinfer1::LayerType : int32_t {
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::kUNARY = 11,
nvinfer1::LayerType::kPADDING = 12,
nvinfer1::LayerType::kSHUFFLE = 13,
nvinfer1::LayerType::kREDUCE = 14,
nvinfer1::LayerType::kTOPK = 15,
nvinfer1::LayerType::kGATHER = 16,
nvinfer1::LayerType::kMATRIX_MULTIPLY = 17,
nvinfer1::LayerType::kRAGGED_SOFTMAX = 18,
nvinfer1::LayerType::kCONSTANT = 19,
nvinfer1::LayerType::kRNN_V2 = 20,
nvinfer1::LayerType::kIDENTITY = 21,
nvinfer1::LayerType::kPLUGIN_V2 = 22,
nvinfer1::LayerType::kSLICE = 23,
nvinfer1::LayerType::kSHAPE = 24,
nvinfer1::LayerType::kPARAMETRIC_RELU = 25,
nvinfer1::LayerType::kRESIZE = 26,
nvinfer1::LayerType::kTRIP_LIMIT = 27,
nvinfer1::LayerType::kRECURRENCE = 28,
nvinfer1::LayerType::kITERATOR = 29,
nvinfer1::LayerType::kLOOP_OUTPUT = 30,
nvinfer1::LayerType::kSELECT = 31,
nvinfer1::LayerType::kFILL = 32,
nvinfer1::LayerType::kQUANTIZE = 33,
nvinfer1::LayerType::kDEQUANTIZE = 34,
nvinfer1::LayerType::kCONDITION = 35,
nvinfer1::LayerType::kCONDITIONAL_INPUT = 36,
nvinfer1::LayerType::kCONDITIONAL_OUTPUT = 37,
nvinfer1::LayerType::kSCATTER = 38,
nvinfer1::LayerType::kEINSUM = 39,
nvinfer1::LayerType::kASSERTION = 40
} |
| The type values of layer classes. More...
|
|
enum | nvinfer1::ActivationType : int32_t {
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::PaddingMode : int32_t {
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 takes precedence if setPaddingMode() and setPrePadding() are also used. More...
|
|
enum | nvinfer1::PoolingType : int32_t { kMAX = 0,
kAVERAGE = 1,
kMAX_AVERAGE_BLEND = 2
} |
| The type of pooling to perform in a pooling layer.
|
|
enum | nvinfer1::ScaleMode : int32_t { 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 : int32_t {
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,
nvinfer1::ElementWiseOperation::kFLOOR_DIV = 7,
nvinfer1::ElementWiseOperation::kAND = 8,
nvinfer1::ElementWiseOperation::kOR = 9,
nvinfer1::ElementWiseOperation::kXOR = 10,
nvinfer1::ElementWiseOperation::kEQUAL = 11,
nvinfer1::ElementWiseOperation::kGREATER = 12,
nvinfer1::ElementWiseOperation::kLESS = 13
} |
| Enumerates the binary operations that may be performed by an ElementWise layer. More...
|
|
enum | nvinfer1::GatherMode : int32_t { nvinfer1::GatherMode::kDEFAULT = 0,
nvinfer1::GatherMode::kELEMENT = 1,
nvinfer1::GatherMode::kND = 2
} |
| Control form of IGatherLayer. More...
|
|
enum | nvinfer1::RNNOperation : int32_t { 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 : int32_t { nvinfer1::RNNDirection::kUNIDIRECTION = 0,
nvinfer1::RNNDirection::kBIDIRECTION = 1
} |
| Enumerates the RNN direction that may be performed by an RNN layer. More...
|
|
enum | nvinfer1::RNNInputMode : int32_t { nvinfer1::RNNInputMode::kLINEAR = 0,
nvinfer1::RNNInputMode::kSKIP = 1
} |
| Enumerates the RNN input modes that may occur with an RNN layer. More...
|
|
enum | nvinfer1::RNNGateType : int32_t {
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::UnaryOperation : int32_t {
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,
nvinfer1::UnaryOperation::kERF = 19,
nvinfer1::UnaryOperation::kNOT = 20,
nvinfer1::UnaryOperation::kSIGN = 21,
nvinfer1::UnaryOperation::kROUND = 22
} |
| Enumerates the unary operations that may be performed by a Unary layer. More...
|
|
enum | nvinfer1::ReduceOperation : int32_t {
kSUM = 0,
kPROD = 1,
kMAX = 2,
kMIN = 3,
kAVG = 4
} |
| Enumerates the reduce operations that may be performed by a Reduce layer. More...
|
|
enum | nvinfer1::SliceMode : int32_t {
nvinfer1::SliceMode::kDEFAULT = 0,
nvinfer1::SliceMode::kWRAP = 1,
nvinfer1::SliceMode::kCLAMP = 2,
nvinfer1::SliceMode::kFILL = 3,
nvinfer1::SliceMode::kREFLECT = 4
} |
| Controls how ISliceLayer handles out of bounds coordinates. More...
|
|
enum | nvinfer1::TopKOperation : int32_t { nvinfer1::TopKOperation::kMAX = 0,
nvinfer1::TopKOperation::kMIN = 1
} |
| Enumerates the operations that may be performed by a TopK layer. More...
|
|
enum | nvinfer1::MatrixOperation : int32_t { 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::ResizeMode : int32_t { nvinfer1::ResizeMode::kNEAREST = 0,
nvinfer1::ResizeMode::kLINEAR = 1
} |
| Enumerates various modes of resize in the resize layer. Resize mode set using setResizeMode(). More...
|
|
enum | nvinfer1::ResizeCoordinateTransformation : int32_t { nvinfer1::ResizeCoordinateTransformation::kALIGN_CORNERS = 0,
nvinfer1::ResizeCoordinateTransformation::kASYMMETRIC = 1,
nvinfer1::ResizeCoordinateTransformation::kHALF_PIXEL = 2
} |
| The resize coordinate transformation function. More...
|
|
enum | nvinfer1::ResizeSelector : int32_t { nvinfer1::ResizeSelector::kFORMULA = 0,
nvinfer1::ResizeSelector::kUPPER = 1
} |
| The coordinate selector when resize to single pixel output. More...
|
|
enum | nvinfer1::ResizeRoundMode : int32_t { nvinfer1::ResizeRoundMode::kHALF_UP = 0,
nvinfer1::ResizeRoundMode::kHALF_DOWN = 1,
nvinfer1::ResizeRoundMode::kFLOOR = 2,
nvinfer1::ResizeRoundMode::kCEIL = 3
} |
| The rounding mode for nearest neighbor resize. More...
|
|
enum | nvinfer1::LoopOutput : int32_t { nvinfer1::LoopOutput::kLAST_VALUE = 0,
nvinfer1::LoopOutput::kCONCATENATE = 1,
nvinfer1::LoopOutput::kREVERSE = 2
} |
| Enum that describes kinds of loop outputs. More...
|
|
enum | nvinfer1::TripLimit : int32_t { nvinfer1::TripLimit::kCOUNT = 0,
nvinfer1::TripLimit::kWHILE = 1
} |
| Enum that describes kinds of trip limits. More...
|
|
enum | nvinfer1::FillOperation : int32_t { nvinfer1::FillOperation::kLINSPACE = 0,
nvinfer1::FillOperation::kRANDOM_UNIFORM = 1
} |
| Enumerates the tensor fill operations that may performed by a fill layer. More...
|
|
enum | nvinfer1::ScatterMode : int32_t { nvinfer1::ScatterMode::kELEMENT = 0,
nvinfer1::ScatterMode::kND = 1
} |
| Control form of IScatterLayer. More...
|
|
enum | nvinfer1::CalibrationAlgoType : int32_t { kLEGACY_CALIBRATION = 0,
kENTROPY_CALIBRATION = 1,
kENTROPY_CALIBRATION_2 = 2,
kMINMAX_CALIBRATION = 3
} |
| Version of calibration algorithm to use. More...
|
|
enum | nvinfer1::QuantizationFlag : int32_t { nvinfer1::QuantizationFlag::kCALIBRATE_BEFORE_FUSION = 0
} |
| List of valid flags for quantizing the network to int8. More...
|
|
enum | nvinfer1::BuilderFlag : int32_t {
nvinfer1::BuilderFlag::kFP16 = 0,
nvinfer1::BuilderFlag::kINT8 = 1,
nvinfer1::BuilderFlag::kDEBUG = 2,
nvinfer1::BuilderFlag::kGPU_FALLBACK = 3,
nvinfer1::BuilderFlag::kSTRICT_TYPES = 4,
nvinfer1::BuilderFlag::kREFIT = 5,
nvinfer1::BuilderFlag::kDISABLE_TIMING_CACHE = 6,
nvinfer1::BuilderFlag::kTF32 = 7,
nvinfer1::BuilderFlag::kSPARSE_WEIGHTS = 8,
nvinfer1::BuilderFlag::kSAFETY_SCOPE = 9
} |
| List of valid modes that the builder can enable when creating an engine from a network definition. More...
|
|
enum | nvinfer1::NetworkDefinitionCreationFlag : int32_t { nvinfer1::NetworkDefinitionCreationFlag::kEXPLICIT_BATCH = 0,
nvinfer1::NetworkDefinitionCreationFlag::kEXPLICIT_PRECISION = 1
} |
| List of immutable network properties expressed at network creation time. NetworkDefinitionCreationFlag is used with createNetworkV2 to specify immutable properties of the network. The createNetwork() function always had an implicit batch dimension being specified by the maxBatchSize builder parameter. createNetworkV2 with kDEFAULT flag mimics that behaviour. More...
|
|
TensorRT Versioning follows Semantic Versioning Guidelines specified here: https://semver.org/
This is the top-level API file for TensorRT.