Skip to content

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

@nan2088

Description

@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?

Metadata

Metadata

Assignees

Labels

Module:ONNXIssues relating to ONNX usage and importinternal-bug-trackedTracked internally, will be fixed in a future release.triagedIssue has been triaged by maintainers

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions