torch-mlir/frontends/pytorch/csrc
Bryce Arden 27a4515de2
Add Conv2D Torchscript Import Support (#167)
Adds support for lowering a torch.nn.Conv2d module to the Torch Dialect through TorchScript import.
Generated IR can be viewed here:
https://gist.github.com/brycearden/6c0f790115c4577249372ef82768e6fd

Required implementing support for tuple in the ivalue importer and list in the node importer.
2021-02-25 12:14:00 -08:00
..
builder Add Conv2D Torchscript Import Support (#167) 2021-02-25 12:14:00 -08:00
CMakeLists.txt Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -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