2019年3月15日
TensorRT&Sample&Python[uff_custom_plugin]
摘要： 本文是基于TensorRT 5.0.2基础上，关于其内部的uff_custom_plugin例子的分析和介绍。 本例子展示如何使用cpp基于tensorrt python绑定和UFF解析器进行编写plugin。该例子实现一个clip层（以CUDA kernel实现），然后封装成一个tensorrt 阅读全文
posted @ 2019-03-15 17:57 仙守 阅读(4300) 评论(0) 推荐(0) 编辑