Closed
Description
(此 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推理框架