add_n api support float16 (#42801)

353ede5a · WangXi · GitHub · 78d5cf7b · 353ede5a
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/tensor/math.py python/paddle/tensor/math.py +2 -2

未找到文件。
--- a/python/paddle/tensor/math.py
+++ b/python/paddle/tensor/math.py
@@ -1388,10 +1388,10 @@ def add_n(inputs, name=None):
        if len(inputs) > 0:
            for input in inputs:
                check_variable_and_dtype(input, "inputs", \
-                   ['float32', 'float64', 'int32', 'int64'], 'add_n')
+                   ['float16', 'float32', 'float64', 'int32', 'int64'], 'add_n')
    else:
        check_variable_and_dtype(inputs, "inputs", \
-                ['float32', 'float64', 'int32', 'int64'], 'add_n')
+                ['float16', 'float32', 'float64', 'int32', 'int64'], 'add_n')


    out = helper.create_variable_for_type_inference(