llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
Gaurav Shukla 4b911ada40 [LINALG] Add E2E support for `aten.mean.dim` op 
- This commit adds support for `aten.mean.dim` op.
- It also adds a new test script `stats.py` for statistics related ops.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 		2022-05-04 20:11:42 +05:30
..
IR Fix TupleIndex canonicalizer. 2022-05-03 09:08:49 -07:00
Transforms [LINALG] Add E2E support for `aten.mean.dim` op 2022-05-04 20:11:42 +05:30
Utils importer: add initial support for loading BFloat16 tensors (#761) 2022-04-29 09:01:49 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00