torch-mlir/lib/Dialect/Torch
Ramiro Leal-Cavazos 51e267aa37
Combine maximize-value-semantics rewrite patterns into one pattern (#642)
This commit replaces the two rewrite patterns of
maximize-value-semantics with a single pattern that captures the
behavior of both as well as other edge cases previously not
supported. The new pattern works by first performing alias analysis on
a subgraph to see if pattern is applicable, then rewriting all
non-value tensors to value tensors in a single go.
2022-03-10 09:36:52 -08:00
..
IR LLVM bump 2022-02-16 15:28:13 -05:00
Transforms Combine maximize-value-semantics rewrite patterns into one pattern (#642) 2022-03-10 09:36:52 -08:00
Utils [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00