[NPU] support fp16 of input for api pow (#31871)

4a823c5f · liym27 · GitHub · f354e1d6 · 4a823c5f
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +2 -2

未找到文件。
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -9525,8 +9525,8 @@ def pow(x, factor=1.0, name=None):
            y_2 = fluid.layers.pow(x, factor=factor_tensor)
            # y_2 is x^{3.0}
    """
-    check_variable_and_dtype(x, 'x', ['int32', 'int64', 'float32', 'float64'],
+    check_variable_and_dtype(
-                             'pow')
+        x, 'x', ['int32', 'int64', 'float16', 'float32', 'float64'], 'pow')
    helper = LayerHelper('pow', **locals())
    inputs = {'X': x}