|
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::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
} |
| 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::ActivationType : int { nvinfer1::ActivationType::kRELU = 0,
nvinfer1::ActivationType::kSIGMOID = 1,
nvinfer1::ActivationType::kTANH = 2
} |
| 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 scale is 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::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
} |
| 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::CalibrationAlgoType : int { kLEGACY_CALIBRATION = 0,
kENTROPY_CALIBRATION = 1
} |
| Version of calibration algorithm to use. More...
|
|
This is the top-level API file for TensorRT.