Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【PaddlePaddle Hackathon】28、Paddle TensorRT 增加gru的支持 #35979

Closed
TCChenlong opened this issue Sep 23, 2021 · 0 comments
Closed

【PaddlePaddle Hackathon】28、Paddle TensorRT 增加gru的支持 #35979

TCChenlong opened this issue Sep 23, 2021 · 0 comments

Comments

@TCChenlong
Copy link
Contributor

(此 ISSUE 为 PaddlePaddle Hackathon 活动的任务 ISSUE,更多详见PaddlePaddle Hackathon

【任务说明】

  • 任务标题:Paddle TensorRT 增加gru的支持

  • 技术标签:预测,C++,TensorRT

  • 任务难度:中等

  • 详细描述:Paddle Inference通过子图接入的方式,在SimpleOpTypeSetTeller里判断一个OP是否可以运行到TensorRT中,支持的OP经过对应的OpConverter转变成TensorRT的原生Layer或者Plugin。通过增加该OP的支持,可以助于熟悉Paddle Inference、TensorRT等深度学习推理框架的典型执行流程。加入方式可以参考reduce_sum的实现,TensorRT中GRU的实现参考TensorRT work-with-loops

【提交内容】

  • 任务 PR 到 Paddle

  • 相关技术文档

  • 任务单测文件(静态shape/动态shape/FP32/FP16)

【技术要求】

  • 熟练掌握 C++

  • 良好的编程规范

  • 熟悉TensorRT的使用和paddle推理框架

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants