I’m using TensorRT version 126.96.36.199.
I successfully replaced an unsupported Tensorflow operation with a TensorRT PlugIn layer using the graphsurgeon APIs.
Now, I want to add a totally new node in the middle of the model.
I saw that the graphsurgeon support these APIs:
But I couldn’t understand if I can use them for my purpose.
I know how to find the specific model location where I want to add my new node but I cannot find the right way to use the graphsurgeon APIs to do that.