feat(mge): make Function available at autodiff.Function

GitOrigin-RevId: 7c9ab65351

imperative/python/megengine/autodiff/__init__.py

@@ -6,4 +6,5 @@
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+from ..core.autodiff.grad import Function
 from .grad_manager import GradManager

imperative/python/megengine/quantization/fake_quant.py

@@ -11,7 +11,7 @@ from typing import Iterable
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from ..core.tensor.dtype import _metadata_dict, get_quantized_dtype
 from ..module import Module
 from ..tensor import Parameter, Tensor

imperative/python/megengine/quantization/internal_fake_quant.py

@@ -12,7 +12,7 @@ from functools import partial
 import numpy as np
 
 from .. import functional as F
-from ..core.autodiff.grad import Function
+from ..autodiff import Function
 from .fake_quant import _FakeQuantize
 from .observer import MinMaxObserver
 from .qconfig import QConfig

imperative/python/megengine/quantization/utils.py

@@ -12,8 +12,8 @@ from typing import Dict
 import numpy as np
 
 from .. import functional as F
+from ..autodiff import Function
 from ..core._imperative_rt.core2 import apply
-from ..core.autodiff.grad import Function
 from ..core.ops import builtin
 from ..core.tensor import megbrain_graph
 from ..core.tensor.dtype import _metadata_dict

imperative/python/test/unit/core/test_function.py

@@ -15,7 +15,7 @@ import megengine.optimizer as optimizer
 from megengine import Parameter
 from megengine import Tensor as tensor
 from megengine import tensor
-from megengine.core.autodiff.grad import Function
+from megengine.autodiff import Function
 from megengine.module import Module