TensorRT：NvInfer.h

版本：5.1.5.0

工欲善其事必先利其器，后面需要使用RT插件，所以还是得学习一下RT的相关代码

NvInfer.h中声明和定义了RT中的大部分变量、函数、类等

RT中代码格式较为统一，大体看了设定规则

枚举变量：k+全大写， kFLOAT， kGPU，kCHANNEL

类内变量：k+首字母大写，nbDims

类内函数：小写+首字母大写，setNames(), setInput()

不能派生或者继承的：用大写的“I”进行了标识， ITensor,ILayer

0：命名空间：nvidia + infer  --- > nvinfer1

namespace nvinfer1

1：枚举变量

数据类型：float32, float16, int8, int32  

enum class DataType : int
{
    kFLOAT = 0, //!< FP32 format.
    kHALF = 1,  //!< FP16 format.
    kINT8 = 2,  //!< quantized INT8 format.
    kINT32 = 3  //!< INT32 format.
};

 设备类型：GPU, DLA

enum class DeviceType : int
{
    kGPU, //!< GPU Device
    kDLA, //!< DLA Core
};

 

来源：oschina

链接：https://my.oschina.net/u/4407314/blog/4496957