mirror of https://github.com/llvm/torch-mlir
e0f301c890
This commit adds the ability to specify extra abstract interpretation functions in `torch_mlir.compile` to use during type refinement. This allows users to easily add custom ops without having to interact with MLIR or C++ directly. |
||
---|---|---|
.. | ||
importer/jit_ir | ||
custom_op_shape_dtype_fn.py | ||
lit.local.cfg | ||
smoketest.py |