未验证 提交 74875921 编写于 作者: Z Zhong Hui 提交者: GitHub

[BF16] Normal support bf16 (#54564) (#54620)

上级 bd1a240b
...@@ -327,7 +327,7 @@ def gaussian(shape, mean=0.0, std=1.0, seed=0, dtype=None, name=None): ...@@ -327,7 +327,7 @@ def gaussian(shape, mean=0.0, std=1.0, seed=0, dtype=None, name=None):
distribution, with ``shape`` and ``dtype``. distribution, with ``shape`` and ``dtype``.
""" """
op_type_for_check = 'gaussian/standard_normal/randn/normal' op_type_for_check = 'gaussian/standard_normal/randn/normal'
supported_dtypes = ['float32', 'float64', 'float16', 'uint16'] supported_dtypes = ['float32', 'float64', 'float16', 'uint16', 'bfloat16']
if dtype is None: if dtype is None:
dtype = paddle.framework.get_default_dtype() dtype = paddle.framework.get_default_dtype()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册