当前位置:   article > 正文

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

tensorrt 输入输出

几个重要概念

binding:

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/334854
推荐阅读
相关标签
  

闽ICP备14008679号