torch-mlir/python/npcomp/torch/opdefs
Stella Laurenzo a74a98094b
Add a new python script to auto-generate ATen op ODS definitions. (#43)
* Add a new python script to auto-generate ATen op ODS definitions.

* There is still some work on some of the ops to annotate correct types.
* The ODS is not actually included into the dialect yet, but I'd like to commit it so that we can track changes.
* Will reconcile this with the ops produced by the existing script in a followup. Still need to do some more iteration to reach parity.
2020-09-16 16:21:24 -07:00
..
__init__.py Initial work on a torch op registry. 2020-08-28 15:20:55 -07:00
aten_ops.py Initial work on a torch op registry. 2020-08-28 15:20:55 -07:00
generate_ods.py Add a new python script to auto-generate ATen op ODS definitions. (#43) 2020-09-16 16:21:24 -07:00
registry.py Add a new python script to auto-generate ATen op ODS definitions. (#43) 2020-09-16 16:21:24 -07:00