2025年3月5日
100 TensorRT Puglin开发
摘要： 1. 核心原理 定义计算逻辑：告诉 GPU 如何执行这个特殊层的计算。 集成到 TensorRT：让 TensorRT 在推理时调用你的计算逻辑。 2. 开发步骤 1. 创建一个插件类，继承TensorRT的 IPluginV2 接口，实现关键函数 #include <NvInfer.h> clas 阅读全文
posted @ 2025-03-05 11:45 ldfm 阅读(80) 评论(0) 推荐(0)