mirror of https://github.com/llvm/torch-mlir
e7111d473b
- Lowers the torch.scatter_reduce to linalg_on_tensors dialect. - Includes support for "sum", "prod", "amax", "amin" and "mean". |
||
---|---|---|
.. | ||
CMakeLists.txt | ||
TorchToTMTensor.cpp |