[Paddle-TRT] replace slice plugin with TRT ISliceLayer

oiopk7p5  于 2022-04-21  发布在  Java
关注(0)|答案(1)|浏览(195)

@cryoco ,
The PR #25956 loses to add serialize/deserialize to slice plugin. The slice plugin will be included by ERNIE inference when static shape of slice plugin is ready. I have made a reference: commit-d07febd2 at my repo. Maybe you can use cherry-pick to merge my commit.

deikduxw

deikduxw1#

@NHZlX, @cryoco,
There are 2 pull requests regarding slice plugin - #25602 and #25276.
商智洲 told me that what Paddle needs is actually as same as nvinfer1::ISliceLayer, so we can replace slice plugin with nvinfer1::ISliceLayer, then we don't need the 2 pull requests.

相关问题