Searched defs:rms_norm (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ | ||
| H A D | layer_norm.cpp | 266 Tensor rms_norm( in rms_norm() function |
| /aosp_15_r20/external/pytorch/torch/nn/ | ||
| H A D | functional.py | 2905 def rms_norm( function |