[FEATURE REQUEST] Support onnx's local function as custom plugin

[nan2088]

Functions is a feature in onnx, which can be thought of as an operator combined with an implementation of the operator using other, more primitive, ops, referred to as the function body.

Right now TensorRT support this feature. However, onnxparser will first use the LocalFunctionImporter before FallbackPluginImporter :

https://github.com/onnx/onnx-tensorrt/blob/706f02e74366b2cbaacf87be61de95df051a2788/ModelImporter.cpp#L159

So it is impossible to write a custom plugin to re-implement these local function.

I export an onnx with local function node, Is there any way I can use custom plugin for this function?

[moraxu]

cc @kevinch-nv if this issue was ever discussed internally?

[nan2088]

Hi, is there any update?

[kevinch-nv]

Yes, apologies for the late reply. It makes sense for TensorRT plugins with the same name as a local function to have high priority when parsing the ONNX model. An internal bug has been filed and we'll notify when it has been updated.