diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 4f004fc094130a6e6ec4e7ddd021232539d73587..ce5cc78419fa72d651e990c17852ae207ff606cf 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -37,7 +37,6 @@ __all__ = [ "dropout", "indexing_one_hot", "leaky_relu", - "linear", "local_conv2d", "logsigmoid", "logsumexp", diff --git a/imperative/python/megengine/module/linear.py b/imperative/python/megengine/module/linear.py index fcb6ab15d47f50d14db1599b7892182d0ed8d6fa..8e7ebed7ecb7f9059c5a78502741cfee15e73f85 100644 --- a/imperative/python/megengine/module/linear.py +++ b/imperative/python/megengine/module/linear.py @@ -7,7 +7,7 @@ # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. import numpy as np -from ..functional import linear +from ..functional.nn import linear from ..tensor import Parameter from . import init from .module import Module diff --git a/imperative/python/megengine/module/quantized/linear.py b/imperative/python/megengine/module/quantized/linear.py index 41df57ff71e8d2f049b3594de6747e82ca32bf23..c01b2b492fed4879d6d318bbedfedece08f138fa 100644 --- a/imperative/python/megengine/module/quantized/linear.py +++ b/imperative/python/megengine/module/quantized/linear.py @@ -31,7 +31,7 @@ class Linear(QuantizedModule): inp_scale = dtype.get_scale(inp.dtype) w_scale = dtype.get_scale(self.weight.dtype) bias_dtype = dtype.qint32(inp_scale * w_scale) - return F.linear( + return F.nn.linear( inp, self.weight, None if self.bias is None else self.bias.astype(bias_dtype),