Support kwargs in ai_edge_torch.convert to named inputs

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.

Apache License 2.0

278 stars 36 forks source link

Support kwargs in ai_edge_torch.convert to named inputs #86

Closed chunnienc closed 1 month ago

chunnienc commented 1 month ago

torch_xla does not support exported_program with kwargs to stablehlo, so I need to flatten inputs to a tuple before the very first torch.export.export.

BUG=https://b.corp.google.com/issues/350510246