Sean Silva 55c3cc6624 Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim ... Interestingly, TorchScript has its own op (`torch::jit::Operator`) registry separate from the dispatcher (it is a superset of the dispatcher). This is where the "prim" ops and some "aten" ops (that should probably be renamed to "prim") live. In particular, `aten::__is__` is in that latter category of "aten but really prim". This registry is also the source of truth for what the TorchScript interpreter calls into when it executes. The bulk of the "not part of the dispatcher" ops live in 09feb5f579/torch/csrc/jit/runtime/register_prim_ops.cpp (L82) And the registry itself lives in: 09feb5f579/torch/csrc/jit/runtime/operator.cpp (L196) This fold further reduces the IR of ResNet by folding away some more not-taken branches. These not-taken branches in ResNet require first-class handling of the list type which we don't yet have on any backend.		2021-04-30 10:57:02 -07:00
..
builder	Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim	2021-04-30 10:57:02 -07:00
CMakeLists.txt	Properly model "derefinement".	2021-03-03 15:09:44 -08:00
init_python_bindings.cpp	Make torch_mlir compatible with binary PyTorch installations.	2020-12-14 09:51:00 -08:00
init_python_bindings.h	Delete old PyTorch 1.3 type dispatch oriented code paths.	2020-11-12 22:27:05 -08:00
pybind.h	Add TorchScript graph importer.	2020-11-23 14:20:09 -08:00