diff --git a/python/paddle/fluid/dygraph/__init__.py b/python/paddle/fluid/dygraph/__init__.py index 551561428da72..1132ef393d552 100644 --- a/python/paddle/fluid/dygraph/__init__.py +++ b/python/paddle/fluid/dygraph/__init__.py @@ -51,8 +51,6 @@ from .math_op_patch import monkey_patch_math_varbase -from .inplace_utils import inplace_apis_in_dygraph_only - __all__ = [] __all__ += layers.__all__ __all__ += base.__all__ diff --git a/python/paddle/fluid/dygraph/nn.py b/python/paddle/fluid/dygraph/nn.py index 4dfb67ab4aa42..4390ef6a9b365 100644 --- a/python/paddle/fluid/dygraph/nn.py +++ b/python/paddle/fluid/dygraph/nn.py @@ -30,6 +30,7 @@ in_dygraph_mode, _in_legacy_dygraph, ) + from ..data_feeder import ( convert_dtype, check_variable_and_dtype, diff --git a/python/paddle/nn/functional/activation.py b/python/paddle/nn/functional/activation.py index 6bf32317a461f..ca855b057b3a5 100644 --- a/python/paddle/nn/functional/activation.py +++ b/python/paddle/nn/functional/activation.py @@ -15,9 +15,9 @@ import paddle from paddle import _C_ops, _legacy_C_ops, in_dynamic_mode from paddle.framework import core +from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only from ...fluid.data_feeder import check_dtype, check_variable_and_dtype -from ...fluid.dygraph.inplace_utils import inplace_apis_in_dygraph_only from ...fluid.framework import ( _in_legacy_dygraph, convert_np_dtype_to_dtype_, diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 8c47809d222a9..ef8d154714aec 100644 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -20,6 +20,7 @@ import paddle from paddle import _C_ops, _legacy_C_ops +from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only from ..common_ops_import import _varbase_creator, fill_constant from ..fluid.data_feeder import ( @@ -28,7 +29,6 @@ check_variable_and_dtype, convert_dtype, ) -from ..fluid.dygraph.inplace_utils import inplace_apis_in_dygraph_only from ..fluid.framework import _in_legacy_dygraph, _non_static_mode from ..fluid.layers import utils from ..framework import ( diff --git a/python/paddle/tensor/math.py b/python/paddle/tensor/math.py index efa8d7c453b80..9ab909fa2dfb2 100644 --- a/python/paddle/tensor/math.py +++ b/python/paddle/tensor/math.py @@ -14,7 +14,6 @@ """ math functions """ - # TODO: define math functions import numpy as np @@ -23,13 +22,15 @@ from paddle import _C_ops, _legacy_C_ops from paddle.common_ops_import import VarDesc, dygraph_only, dygraph_utils +# TODO: define math functions +from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only + from ..fluid.data_feeder import ( check_dtype, check_type, check_variable_and_dtype, convert_dtype, ) -from ..fluid.dygraph.inplace_utils import inplace_apis_in_dygraph_only from ..fluid.layers import elementwise_sub, utils from ..framework import ( LayerHelper, diff --git a/python/paddle/fluid/dygraph/inplace_utils.py b/python/paddle/utils/inplace_utils.py similarity index 89% rename from python/paddle/fluid/dygraph/inplace_utils.py rename to python/paddle/utils/inplace_utils.py index fb27a5674b7d8..2ee809facab4f 100644 --- a/python/paddle/fluid/dygraph/inplace_utils.py +++ b/python/paddle/utils/inplace_utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. +# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -12,11 +12,11 @@ # See the License for the specific language governing permissions and # limitations under the License. -from ..wrapped_decorator import wrap_decorator -from ..framework import _non_static_mode import warnings import paddle from paddle import _C_ops, _legacy_C_ops +from paddle.fluid.framework import _non_static_mode +from paddle.fluid.wrapped_decorator import wrap_decorator # NOTE(pangyoki): The Inplace APIs with underline(`_`) is only valid for the method of calling `_C_ops`