核心类
1. 基础类
类名 | 职责说明 | 典型生命周期 |
---|---|---|
ILogger | 日志输出控制(错误/警告/信息) | 全局单例 |
IBuilder | 引擎构建器(创建网络、配置参数、生成引擎) | build阶段创建 |
INetworkDefinition | 定义计算图结构(添加层、设置输入输出) | 由Builder创建 |
IBuilderConfig | 构建配置(设置精度、工作空间大小、优化策略等) | 与Builder关联 |
2. 模型解析
类名 | 功能说明 |
---|---|
OnnxParser | 解析ONNX模型到TensorRT网络结构 |
UffParser |