torch-mlir/include/npcomp/Dialect/Torch
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
..
IR Add Conv2D Torchscript Import Support (#167) 2021-02-25 12:14:00 -08:00
Transforms Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00