torch-mlir/lib/Conversion/TorchToTMTensor
Zachary Cetinic e7111d473b
[Torch Dialect] Scatter reduce lowering (#1884)
- Lowers the torch.scatter_reduce to linalg_on_tensors dialect.
- Includes support for "sum", "prod", "amax", "amin" and "mean".
2023-02-21 23:05:55 +00:00
..
CMakeLists.txt Bump LLVM (#958) 2022-06-22 22:23:46 -05:00
TorchToTMTensor.cpp [Torch Dialect] Scatter reduce lowering (#1884) 2023-02-21 23:05:55 +00:00