torch-mlir/python/torch_mlir/extras
Stella Laurenzo 282e9b0e64
[fx] Fix type determination for multi-return ops and static `None` returns. (#3081)
In practice, this was caught by the way that AOT autograd traces
`convolution_backward`. For the unit test, we just repro it with a
custom op.
2024-04-01 09:39:38 -07:00
..
fx_decomp_util.py add support for decomposition (#2879) 2024-02-14 21:00:52 -08:00
fx_importer.py [fx] Fix type determination for multi-return ops and static `None` returns. (#3081) 2024-04-01 09:39:38 -07:00
onnx_importer.py Normalize type hints to be compatible with multiple Python versions (#3028) 2024-03-15 08:29:48 -07:00