llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test/python/fx_importer
History
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
..
v2.3 Representing Symbolic Shape Expressions in Torch Dialect (#3372) 2024-06-07 04:04:03 -07:00
basic_test.py Representing Symbolic Shape Expressions in Torch Dialect (#3372) 2024-06-07 04:04:03 -07:00
custom_op_test.py Allow custom ops to return an array of tensors (#3531) 2024-07-14 11:54:23 -07:00
sparse_test.py [torch-mlir][sparse] add decomposition features to sparse compiler (#3505) 2024-06-28 10:18:36 -07:00
symbolic_shape_expr_test.py [Symbolic Shapes] Test coverage for unbacked symint from data dependent ops (#3542) 2024-07-14 11:52:03 -07:00
sympy_to_affine_expr_test.py Representing Symbolic Shape Expressions in Torch Dialect (#3372) 2024-06-07 04:04:03 -07:00