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
..
api	Extend jit::load to work on flatbuffer file; Take 2 (#75256)	2022-04-06 17:54:01 +00:00
c10d	Add cuda-11.3+clang9 build workflow (take 2)	2022-04-11 17:13:01 +00:00
common
dist_autograd
jit	[JIT] Move Shape Function definition to python	2022-04-19 20:59:44 +00:00
lazy	[LT] Support Tensor.is_alias_of	2022-04-14 07:28:03 +00:00
lite_interpreter_runtime	[PyTorch] Add codegen unboxing ability (#69881)	2022-03-01 23:28:13 +00:00
monitor	torch/monitor: merge Interval and FixedCount stats (#72009)	2022-01-30 23:21:59 +00:00
profiler	[RecordFunction] More effecient machinery to determine which callbacks to run. (#75807)	2022-04-19 20:46:16 +00:00
rpc	Remove ProcessGroup from TensorPipeAgent initialization (#68128)	2021-11-11 12:28:55 -08:00
tensorexpr	[nnc] Update bounds overlap analysis to identify non-overlaps even with symbolic bounds	2022-04-14 20:24:03 +00:00
__init__.py