torch-mlir/lib/Conversion
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
..
TorchToLinalg [TORCH][MLIR] Fix the return types of `aten.native_layer_norm`. 2022-03-17 12:08:32 +05:30
TorchToSCF Update external/llvm-project 2022-01-18 13:25:42 -05:00
TorchToStd LLVM bump 2022-02-16 15:28:13 -05:00
TorchToTMTensor [MLIR][TORCH] Add value tensor variant to aten::_index_put_impl_ 2022-03-16 22:02:02 +05:30
TorchToTosa [tosa] Support for Aten[Zeros|Ones|Fill_Scalar] ops (#604) 2022-02-16 09:53:51 -08:00
Utils [MLIR][TORCH] Move common helper functions to Utils.cpp 2022-03-08 22:52:34 +05:30
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30