|
enum class | nvinfer1::DataType : int32_t {
nvinfer1::kFLOAT = 0
, nvinfer1::kHALF = 1
, nvinfer1::kINT8 = 2
, nvinfer1::kINT32 = 3
,
nvinfer1::kBOOL = 4
, nvinfer1::kUINT8 = 5
, nvinfer1::kFP8 = 6
, nvinfer1::kBF16 = 7
,
nvinfer1::kINT64 = 8
, nvinfer1::kINT4 = 9
} |
| The type of weights and tensors. More...
|
|
enum class | nvinfer1::TensorFormat : int32_t {
nvinfer1::kLINEAR = 0
, nvinfer1::kCHW2 = 1
, nvinfer1::kHWC8 = 2
, nvinfer1::kCHW4 = 3
,
nvinfer1::kCHW16 = 4
, nvinfer1::kCHW32 = 5
, nvinfer1::kDHWC8 = 6
, nvinfer1::kCDHW32 = 7
,
nvinfer1::kHWC = 8
, nvinfer1::kDLA_LINEAR = 9
, nvinfer1::kDLA_HWC4 = 10
, nvinfer1::kHWC16 = 11
,
nvinfer1::kDHWC = 12
} |
| Format of the input/output tensors. More...
|
|
enum class | nvinfer1::APILanguage : int32_t { nvinfer1::kCPP = 0
, nvinfer1::kPYTHON = 1
} |
| Programming language used in the implementation of a TRT interface. More...
|
|
enum class | nvinfer1::AllocatorFlag : int32_t { nvinfer1::kRESIZABLE = 0
} |
| Allowed type of memory allocation. More...
|
|
enum class | nvinfer1::ErrorCode : int32_t {
nvinfer1::kSUCCESS = 0
, nvinfer1::kUNSPECIFIED_ERROR = 1
, nvinfer1::kINTERNAL_ERROR = 2
, nvinfer1::kINVALID_ARGUMENT = 3
,
nvinfer1::kINVALID_CONFIG = 4
, nvinfer1::kFAILED_ALLOCATION = 5
, nvinfer1::kFAILED_INITIALIZATION = 6
, nvinfer1::kFAILED_EXECUTION = 7
,
nvinfer1::kFAILED_COMPUTATION = 8
, nvinfer1::kINVALID_STATE = 9
, nvinfer1::kUNSUPPORTED_STATE = 10
} |
| Error codes that can be returned by TensorRT during execution. More...
|
|
enum class | nvinfer1::TensorIOMode : int32_t { nvinfer1::kNONE = 0
, nvinfer1::kINPUT = 1
, nvinfer1::kOUTPUT = 2
} |
| Definition of tensor IO Mode. More...
|
|
This file contains common definitions, data structures and interfaces shared between the standard and safe runtime.
- Warning
- Do not directly include this file. Instead include either NvInferRuntime.h (for the standard runtime) or NvInferSafeRuntime.h (for the safety runtime).