torch-mlir/lib/Dialect/Torch
Sean Silva 1b769f7841 Extend GlobalizeObjectGraph to handle torch.prim.GetAttr returning NnModuleType
This happens in practice. With this, we can globalize slots for the
non-trivial classifier layer obtained from
https://github.com/NVIDIA/NeMo/blob/main/tutorials/nlp/Joint_Intent_and_Slot_Classification.ipynb

This also adds support for tuple return types, which were needed by that
model.
2021-02-19 10:23:25 -08:00
..
IR Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
Transforms Extend GlobalizeObjectGraph to handle torch.prim.GetAttr returning NnModuleType 2021-02-19 10:23:25 -08:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00