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
History
Vivek Khandelwal c681c3497a [MLIR][TORCH} Fix empty dim cases for the .dim ops 
This commit fixes the shape calculation for:
1.) aten.mean.dim
2.) aten.var.dim
3.) aten.sum.dim_IntList op

Also, it fixes the lowering of `aten.mean.dim` and
`aten.sum.dim_IntList` for handling the cases of empty dim list.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com
 		2022-07-29 11:08:57 +05:30
..
CAPI Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00
Conversion [MLIR][TORCH} Fix empty dim cases for the .dim ops 2022-07-29 11:08:57 +05:30
Dialect [MLIR][TORCH} Fix empty dim cases for the .dim ops 2022-07-29 11:08:57 +05:30
RefBackend llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
InitAll.cpp Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00