o Ne5@sdddlmZddlmZddlmZddlZddlm Z gZ ddZ dd Z e d d Z d d ZdS))LayerHelperBase) convert_dtype)_dygraph_tracerN)contextmanagercCsdt|tr|tjtjtjfvr|j}ntd|j|dvr#t|}ntdt|t |dS)ak Set default dtype. The default dtype is initially float32. Args: d(string|np.dtype): the dtype to make the default. It only supports float16, float32 and float64. Returns: None. Examples: .. code-block:: python import paddle paddle.set_default_dtype("float32") zMset_default_dtype only supports [float16, float32, float64] , but received %s)float16float32float64rrrN) isinstancetypenprrr__name__ TypeErrorstrrset_default_dtype)drJD:\Projects\ConvertPro\env\Lib\site-packages\paddle/framework/framework.pyrs  rcCstS)a1 Get the current default dtype. The default dtype is initially float32. Args: None. Returns: String, this global dtype only supports float16, float32, float64. Examples: .. code-block:: python import paddle paddle.get_default_dtype() )rget_default_dtyperrrrrKsrccs>t}|r|j}||_z dVW||_dS||_wdVdS)a/ Create a context which enables or disables dygraph gradient calculation. Args: mode(bool): whether to enable (`True`), or disable (`False`) grad. Examples: .. code-block:: python import paddle x = paddle.ones([3, 2]) x.stop_gradient = False with paddle.set_grad_enabled(False): y = x * 2 with paddle.set_grad_enabled(True): z = x * 2 print(y.stop_gradient) # True print(z.stop_gradient) # False NrZ _has_grad)modetracerZ prev_moderrrset_grad_enabled]s rcCst}|r|jSdS)a+ Returns whether current dygraph gradient calculation mode is enabled. Returns: bool: True if current dygraph gradient calculation mode is enabled, otherwise false. Examples: .. code-block:: python import paddle # Dygraph gradient calculation mode is enabled by default. paddle.is_grad_enabled() # True with paddle.set_grad_enabled(False): paddle.is_grad_enabled() # False paddle.enable_static() paddle.is_grad_enabled() # False Fr)rrrris_grad_enabledsr)Zpaddle.fluid.layer_helper_baserZpaddle.fluid.data_feederrZpaddle.fluid.frameworkrnumpyr contextlibr__all__rrrrrrrrs    2  !