问题:已经有了一个onnx格式的模型,转换成trt格式的时候想替换onnx模型格式里的某个算子,如何替换?(比如有五个算子,其数据流关系0->1->2->3->4,我想替换或者优化第2个算子,这时候我自己实现了一个plugin,那我怎么可以准确的把自己实现的plugin替换原先位置为2的算子呢?),多谢。
问题:已经有了一个onnx格式的模型,转换成trt格式的时候想替换onnx模型格式里的某个算子,如何替换?(比如有五个算子,其数据流关系0->1->2->3->4,我想替换或者优化第2个算子,这时候我自己实现了一个plugin,那我怎么可以准确的把自己实现的plugin替换原先位置为2的算子呢?),多谢。