diff --git a/python/paddle/nn/functional/norm.py b/python/paddle/nn/functional/norm.py index 95e1ca2504cd976354725f51da6456cbf3a7a48a..8078cf237ce037a28927e0367e9e73c0a6d3494a 100644 --- 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 = {}