add uint16 for bfloat16 dtype check in layer_norm under static mode (#52845)

5a864270 · Guoxia Wang · GitHub · edd578a1 · 5a864270
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/nn/functional/norm.py python/paddle/nn/functional/norm.py +1 -1

未找到文件。
--- a/python/paddle/nn/functional/norm.py
+++ b/python/paddle/nn/functional/norm.py
@@ -337,7 +337,7 @@ def layer_norm(

    else:
        check_variable_and_dtype(
-            x, 'input', ['float16', 'float32', 'float64'], 'LayerNorm'
+            x, 'input', ['uint16', 'float16', 'float32', 'float64'], 'LayerNorm'
        )

        inputs = {}