超级会员免费看
本文档详细介绍了TensorRT中的NvInfer.h头文件，涵盖枚举变量、Dims类、weights类、IHostMemory类、ITensor类、ILayer类、Plugin类和PluginExt类等内容。重点讨论了Dims类的结构及其子类，如Dims2、DimsHW，以及权重和张量的表示。此外，还提到了自定义插件开发所需的Plugin类。
版本：5.1.5.0
工欲善其事必先利其器，后面需要使用RT插件，所以还是得学习一下RT的相关代码
NvInfer.h中声明和定义了RT中的大部分变量、函数、类等
RT中代码格式较为统一，大体看了设定规则
枚举变量：k+全大写， kFLOAT， kGPU，kCHANNEL
类内变量：k+首字母大写，nbDims
类内函数：小写+首字母大写，setNames(), setInput()
不能派生或者继承的：用大写的“I”进行了标识， ITensor,ILayer具体的接口：https://docs.nvidia.com/deeplearning/tensorrt/api/c_api 可以在这里查看
0：命名空间：nvidia + infer --- > nvinfer1
namespace nvinfer11：枚举变量
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
