Skip to content

Support for multiple signatures in exported models #1837

Open
@leandro-gracia-gil

Description

@leandro-gracia-gil

I'm finding that ONNX Runtime actually has very good performance compared to TFLite in the situations I'm using it, and I'm becoming inclined towards migrating to ONNX. However, ONNX model conversion from TF seems to be missing a key feature: support for multiple signatures.

As far as I can see, if a model has multiple signatures you are forced to select one of them using the --signature_def argument. However, in my case I need to be able to export and run multiple different functions that run in a same model. Creating separate models for each signature would be quite wasteful, as I have many functions and many of them share weights.

Would it be possible to add support for multiple signatures? If not, is this a limitation of TF to ONNX conversion, or a limitation of ONNX itself?

By the way, TensorFlow Lite has support for multiple signatures too since version 2.7, and --signature_def is currently limited to saved models. Though that's not what I'm asking here, maybe you want to consider extending this argument to work on TFLite models too.

Metadata

Metadata

Assignees

No one assigned

    Labels

    contribution welcomeCommunity contribution is welcomedenhancementNew feature or requesttfliteIssues related to TensorFlow Lite

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions