torch-mlir/frontends/pytorch/csrc
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
..
c10_dispatch Add capture function arguments. 2020-10-01 18:59:58 -07:00
type_dispatch Add capture function arguments. 2020-10-01 18:59:58 -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 public API for constructing a module/function to capture PyTorch ops. 2020-09-29 14:23:22 -07:00
init_python_bindings.h Add capture function arguments. 2020-10-01 18:59:58 -07:00
pybind.h Add capture function arguments. 2020-10-01 18:59:58 -07:00