I am trying to convert a Caffe network which includes PReLU activation unit. I was wondering how can I use network with PReLU in tensorRT. I am using python.
Do I need to implemented the plugin from scratch or it has been implemented before or TensorRT6 supports it?
Currently PReLU is not supported in TensorRT. But you can implement custom layers using the IPluginV2Ext class.
Please refer to below support matrix for list of supported plugins: