diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 9e06f107a37585a7eb176d66984b5fef26fed9e8..194a3bc973e245e9b028c508481aad4062785a95 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -489,7 +489,8 @@ def xpu_places(device_ids=None): list of paddle.XPUPlace: Created XPU place list. Examples: .. code-block:: python - + # required: xpu + import paddle import paddle.static as static diff --git a/python/paddle/nn/functional/__init__.py b/python/paddle/nn/functional/__init__.py index ff18afa9d20282a6e1147cbca0d17580456c0c95..f3d9f9dde11a4e48bbabbdb03336021ebc3863ad 100644 --- a/python/paddle/nn/functional/__init__.py +++ b/python/paddle/nn/functional/__init__.py @@ -195,5 +195,8 @@ __all__ = [ #noqa 'gather_tree', 'one_hot', 'normalize', - 'temporal_shift' + 'temporal_shift', + 'batch_norm', + 'layer_norm', + 'instance_norm' ] diff --git a/python/paddle/static/__init__.py b/python/paddle/static/__init__.py index 93394f9b5afdef5cb8a4d4f355c36e59d974fd5a..93f34b2297943a70da929fab3ebf7278f5c2c770 100644 --- a/python/paddle/static/__init__.py +++ b/python/paddle/static/__init__.py @@ -96,6 +96,7 @@ __all__ = [ #noqa 'set_program_state', 'cpu_places', 'cuda_places', + 'xpu_places', 'Variable', 'create_global_var', 'accuracy',