TensorRT:NvInfer.h

2020/08/15 15:33
阅读数 414

版本: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
};

 

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部