o p i @spddlmZddlmZddlZddlZddlmZddl m Z er*ddl m Z m Z gZdd d ZdddZdS)) annotations) TYPE_CHECKINGN) convert_dtype)LayerHelperBase) DTypeLike _DTypeLiteraldrreturnNonecCst|tr|tjtjtjfvr|j}n*td|jt|tj r-t |}|dkr+dn|}|dvr6t |}ntd|t |dS)a Set default dtype. The default dtype is initially float32. Args: d(string|paddle.dtype|np.dtype): the dtype to make the default. It only supports float16, bfloat16, float32 and float64. Returns: None. Examples: .. code-block:: python >>> import paddle >>> paddle.set_default_dtype("float32") zKset_default_dtype only supports [float16, float32, float64] , but received Zuint16bfloat16)float16float32float64r zUset_default_dtype only supports [float16, float32, float64, bfloat16] , but received N) isinstancetypenpr r r__name__ TypeErrorpaddleZdtyperstrrset_default_dtype)rra/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/framework/framework.pyrs&   rrcCstS)a6 Get the current default dtype. The default dtype is initially float32. Args: None. Returns: str, this global dtype only supports float16, float32, float64. Examples: .. code-block:: python >>> import paddle >>> paddle.get_default_dtype() )rget_default_dtyperrrrrPsr)rrr r )r r) __future__rtypingrnumpyrrZpaddle.base.data_feederrZpaddle.base.layer_helper_baserZpaddle._typing.dtype_likerr__all__rrrrrrs     1