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
Rob Suderman 73b6df9007
[torch] Fix DecomposeAtenInstanceNorm decomposition (#2960) 
The decomposition only suports a NCHW lowering however the operation can
support arbitrary spatial dimensions. Updated the lowering to better
support spatial dimensions.
 		2024-02-28 10:27:19 -08:00
..
IR [MLIR][TORCH] Extend support for OnnxToLinalg lowering for Dropout and Div op (#2938) 2024-02-27 11:02:05 +05:30
Transforms [torch] Fix DecomposeAtenInstanceNorm decomposition (#2960) 2024-02-28 10:27:19 -08:00
Utils Clang format refresh (#2812) 2024-01-29 12:59:33 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00