From 164d9cdf1f252ee7ca243214995377ed41d82304 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 9 Oct 2020 22:30:26 +0800
Subject: [PATCH] feat(mge): do not export F.nn.linear

GitOrigin-RevId: 589964729da1b933333f475ee39e4f0a40bbfdde
---
 imperative/python/megengine/functional/nn.py           | 1 -
 imperative/python/megengine/module/linear.py           | 2 +-
 imperative/python/megengine/module/quantized/linear.py | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 4f004fc09..ce5cc7841 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 fcb6ab15d..8e7ebed7e 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 41df57ff7..c01b2b492 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),
-- 
GitLab