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 链接: