torch-mlir/frontends/pytorch/csrc
Stella Laurenzo b5f010284f Add boilerplate to do device capture (pytorch 1.6).
* Uses the new dispatcher API.
* Just prints to the console for the moment when an op is captured.
* Executes the op through the existing implementation.
2020-09-28 10:30:54 -07:00
..
c10_dispatch Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00
type_dispatch Explicitly load aten and std dialects when constructing a context. (#47) 2020-09-16 23:06:22 -07:00
CMakeLists.txt Add initial python bindings for c10 dispatcher internals. (#55) 2020-09-24 16:26:29 -07:00
init_python_bindings.cpp Add initial python bindings for c10 dispatcher internals. (#55) 2020-09-24 16:26:29 -07:00
init_python_bindings.h Add initial python bindings for c10 dispatcher internals. (#55) 2020-09-24 16:26:29 -07:00