torch-mlir/python
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
..
torch_mlir Allow custom ops to return an array of tensors (#3531) 2024-07-14 11:54:23 -07:00
CMakeLists.txt [Stablehlo] enable stablehlo's python extension binding (#3529) 2024-07-10 13:00:13 +08:00
TorchMLIRModule.cpp Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00