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
Prateek Gupta 7256c9e395 [TORCH][MLIR] Fix the return types of `aten.native_layer_norm`. 
This commit fixes the 2nd and 3rd return types of the `aten.native_layer_norm`.
Previously the mean and rSTD were returned with reduction dims removed.
This commit fixes this and keeps the reduction dims of the results.

Signed-Off-By: Prateek Gupta <prateek@nord-labs.com>
 		2022-03-17 12:08:32 +05:30
..
CAPI [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
Conversion [TORCH][MLIR] Fix the return types of `aten.native_layer_norm`. 2022-03-17 12:08:32 +05:30
Dialect [TORCH][MLIR] Fix the return types of `aten.native_layer_norm`. 2022-03-17 12:08:32 +05:30
RefBackend LLVM bump 2022-02-16 15:28:13 -05: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