torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 55c7e66aa7 [MLIR][TORCH] Fix mean and mean.dim op for large-sized inputs
This commit fixes the aten.mean and aten.mean.dim op decomposition
for supporting large-sized inputs.
This commit also fixes the formatting for the file stats.py

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-22 08:38:51 +05:30
..
IR [MLIR][TORCH] Add aten.sort.int op 2022-11-20 19:00:41 +05:30
Transforms [MLIR][TORCH] Fix mean and mean.dim op for large-sized inputs 2022-11-22 08:38:51 +05:30
Utils mlir: add missing dependency on TableGen targets (#1537) 2022-11-01 14:59:11 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00