Elias Ellison f65eb09d6b [JIT] Move Shape Function definition to python ... Moves jit shape function registration to python. Like jit decompositions, a script must be run after adding new definitions which serializes them in a c++ file. This was a request so that torch-mlir could define functions in python and upstream their shape functions. cc @silvasean @makslevental Pull Request resolved: https://github.com/pytorch/pytorch/pull/75546 Approved by: https://github.com/davidberard98		2022-04-19 20:59:44 +00:00
..
gen_jit_shape_functions.py