|
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,
nvinfer1::LayerType::kSHAPE = 25,
nvinfer1::LayerType::kPARAMETRIC_RELU = 26,
nvinfer1::LayerType::kRESIZE = 27,
nvinfer1::LayerType::kTRIP_LIMIT = 28,
nvinfer1::LayerType::kRECURRENCE = 29,
nvinfer1::LayerType::kITERATOR = 30,
nvinfer1::LayerType::kLOOP_OUTPUT = 31,
nvinfer1::LayerType::kSELECT = 32,
nvinfer1::LayerType::kFILL = 33
} |
| The type values of layer classes. 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 takes precedence if setPaddingMode() and setPrePadding() are also used. 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,
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::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::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,
nvinfer1::UnaryOperation::kERF = 19,
nvinfer1::UnaryOperation::kNOT = 20
} |
| 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. More...
|
|
enum | nvinfer1::SliceMode : int { nvinfer1::SliceMode::kDEFAULT = 0,
nvinfer1::SliceMode::kWRAP = 1
} |
| Controls how ISliceLayer handles out of bounds coordinates. More...
|
|
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::ResizeMode : int { kNEAREST = 0,
kLINEAR = 1
} |
| Enumerates various modes of resize in the resize layer. Resize mode set using setResizeMode().
|
|
enum | nvinfer1::LoopOutput : int { nvinfer1::LoopOutput::kLAST_VALUE = 0,
nvinfer1::LoopOutput::kCONCATENATE = 1,
nvinfer1::LoopOutput::kREVERSE = 2
} |
| Enum that describes kinds of loop outputs. More...
|
|
enum | nvinfer1::TripLimit : int { kCOUNT = 0,
kWHILE = 1
} |
| Enum that describes kinds of trip limits.
|
|
enum | nvinfer1::FillOperation : int { nvinfer1::FillOperation::kLINSPACE = 0,
nvinfer1::FillOperation::kRANDOM_UNIFORM = 1
} |
| Enumerates the tensor fill operations that may performed by a fill layer. More...
|
|
enum | nvinfer1::CalibrationAlgoType : int { 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
} |
| List of valid modes that the builder can enable when creating an engine from a network definition. More...
|
|
enum | nvinfer1::ProfilingVerbosity : int32_t { nvinfer1::ProfilingVerbosity::kDEFAULT = 0,
nvinfer1::ProfilingVerbosity::kNONE = 1,
nvinfer1::ProfilingVerbosity::kVERBOSE = 2
} |
| List of verbosity levels of layer information exposed in NVTX annotations. More...
|
|
enum | nvinfer1::NetworkDefinitionCreationFlag : int { 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.