未验证 提交 883ccdd5 编写于 作者: L LoneRanger 提交者: GitHub

add the float16 and bfloat16 for poisson (#55824)

上级 92aa92fa
...@@ -102,7 +102,7 @@ def poisson(x, name=None): ...@@ -102,7 +102,7 @@ def poisson(x, name=None):
Args: Args:
x(Tensor): A tensor with rate parameter of poisson Distribution. The data type x(Tensor): A tensor with rate parameter of poisson Distribution. The data type
should be float32, float64. should be bfloat16, float16, float32, float64.
name(str, optional): The default value is None. Normally there is no name(str, optional): The default value is None. Normally there is no
need for user to set this property. For more information, please need for user to set this property. For more information, please
refer to :ref:`api_guide_Name`. refer to :ref:`api_guide_Name`.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册