torch-mlir/frontends/pytorch/csrc/c10_dispatch
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
..
CMakeLists.txt Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00
acap_dispatch.cpp Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00
acap_dispatch.h Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00
python_bindings.cpp Add boilerplate to do device capture (pytorch 1.6). 2020-09-28 10:30:54 -07:00