torch-mlir/frontends/pytorch/csrc/c10_dispatch
Stella Laurenzo e5433e314f Add capture function arguments.
* Adds at::Tensor -> MlirValue tracking.
* Adds conversions for tensor and scalar types to MLIR types.
* Adds npcomp C APIs for constructing custom types.
* Reworks pybind include so as to get Torch pybind helpers (needed to pass at::Tensor type from Python->C++).
2020-10-01 18:59:58 -07:00
..
CMakeLists.txt Add capture function arguments. 2020-10-01 18:59:58 -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 capture function arguments. 2020-10-01 18:59:58 -07:00
func_builder.cpp Add capture function arguments. 2020-10-01 18:59:58 -07:00
func_builder.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
module_builder.cpp Add capture function arguments. 2020-10-01 18:59:58 -07:00
module_builder.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
python_bindings.cpp Add capture function arguments. 2020-10-01 18:59:58 -07:00