mirror of https://github.com/llvm/torch-mlir
cfc8de36f8
This commit adds support for aten.native_layer_norm operation. Here the previous code for aten.layer_norm is tweaked a little bit to accomodate both mean and variance values alongwith the layer norm value. This commit also adds decomposition of aten.layer_norm into aten.native_layer_norm, which was previously getting lowered directly to linalg. Signed-Off-By: Prateek Gupta<prateek@nod-labs.com> |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| TorchToLinalg.cpp | ||