Skip to content

[ONNX] Support complex types #86746

@justinchuby

Description

@justinchuby

Complex types are currently not tracible in the onnx export process. In ops like STFT, torch uses complex for the aten operators. Since ONNX has no plans for now to expand its complex support, we can consider creating a pass to convert complex inputs to real (and later cancel the nodes out like we do for quantized tensors) and consume them as real values.

This is blocking for STFT and other complex operator support.

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: complexRelated to complex number support in PyTorchmodule: onnxRelated to torch.onnxneeds designWe want to add this feature but we need to figure out how firstonnx-needs-infoneeds information from the author / reporter before ONNX team can take actiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    Status

    Done

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions