mirror of https://github.com/llvm/torch-mlir
7256c9e395
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> |
||
|---|---|---|
| .. | ||
| AdjustCallingConventions.cpp | ||
| CMakeLists.txt | ||
| DecomposeComplexOps.cpp | ||
| DropShapeCalculations.cpp | ||
| GlobalizeObjectGraph.cpp | ||
| InlineGlobalSlots.cpp | ||
| MaximizeValueSemantics.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| PrepareForGlobalizeObjectGraph.cpp | ||
| ReduceOpVariants.cpp | ||
| RefinePublicReturn.cpp | ||
| RefineTypes.cpp | ||
| ReifyShapeCalculations.cpp | ||
| ShapeLibrary.cpp | ||
| SimplifyShapeCalculations.cpp | ||