torch-mlir/test/Dialect
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
..
ATen Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
Basicpy Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
Numpy Bump llvm-project to be7352c00d51f4358db3a23ed6a077f7cb48eafd 2021-01-21 11:16:55 -08:00
Refback [RefBackend] Use std.global_memref instead of homegrown thing 2020-11-13 18:43:50 -08:00
Refbackrt [RefBackend] Fix leaks related to ABI boundaries. 2020-11-25 13:09:58 -08:00
TCF Add TCF convolutional op with bias addition (#137) 2020-12-15 12:53:12 -08:00
TCP Fix TensorFromElementsOp reference 2021-01-28 12:01:35 -08:00
Torch Extend GlobalizeObjectGraph to handle torch.prim.GetAttr returning NnModuleType 2021-02-19 10:23:25 -08:00