当前位置:   article > 正文

tensorRT的使用笔记!_tensorrt 输入输出

tensorrt 输入输出

几个重要概念

binding:

描述engine的输入输出端口。其实就是对输入输出端口的tensor描述其大小等信息。我们用cudaMalloc()函数为engine申请输入输出buff的时候需要查询使用这些信息。

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号