torch-mlir/python/torch_mlir/extras
Matthew Francis-Landau fe9db78120
Allow custom ops to return an array of tensors (#3531)
This PR adds support to `fx_importer.py` for handling custom ops that
return an array of tensors. As long as the length of the array is
consistent across runs (determined statically), then this patch will
work. This does not require that the number of tensors returned is
determined by the op's definition.

CC @sjain-stanford
2024-07-14 11:54:23 -07:00
..
fx_decomp_util.py [torch-mlir][sparse] add decomposition features to sparse compiler (#3505) 2024-06-28 10:18:36 -07:00
fx_importer.py Allow custom ops to return an array of tensors (#3531) 2024-07-14 11:54:23 -07:00
onnx_importer.py Bump Onnx Version to 1.16.1 (#3515) 2024-07-01 22:15:45 +05:30