add float16 (#45216)

f36c4da5 · feifei-111 · GitHub · c93451f4 · f36c4da5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/python/paddle/nn/functional/loss.py
+++ b/python/paddle/nn/functional/loss.py
@@ -2416,7 +2416,7 @@ def cross_entropy(input,
                out = paddle.squeeze(out, axis=axis)
            return out

-    check_variable_and_dtype(input, 'input', ['float32', 'float64'],
+    check_variable_and_dtype(input, 'input', ['float16', 'float32', 'float64'],
                             'softmax_cross_entropy')
    check_variable_and_dtype(
        label, 'label',