[tvm] branch unity updated: [nn.Module] Add Missing Import nn.MultiLinear (#15784)

[ruihangl]

This is an automated email from the ASF dual-hosted git repository.

ruihangl pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/unity by this push:
     new 9fd45741e8 [nn.Module] Add Missing Import nn.MultiLinear (#15784)
9fd45741e8 is described below

commit 9fd45741e818ad0fa41bd0305fc1552c29649976
Author: Junru Shao <junrus...@apache.org>
AuthorDate: Wed Sep 20 06:45:33 2023 -0700

    [nn.Module] Add Missing Import nn.MultiLinear (#15784)
    
    Following #15662, this PR adds the missing import logic of
    `nn.MultiLinear` to the `tvm.relax.frontend.nn` namespace.
---
 python/tvm/relax/frontend/nn/__init__.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/python/tvm/relax/frontend/nn/__init__.py 
b/python/tvm/relax/frontend/nn/__init__.py
index 7e086abeee..f195e6be5c 100644
--- a/python/tvm/relax/frontend/nn/__init__.py
+++ b/python/tvm/relax/frontend/nn/__init__.py
@@ -16,15 +16,16 @@
 # under the License.
 """A PyTorch-like API to build IRModules."""
 from . import op, spec
-from .core import Effect, Module, ModuleList, Parameter, Tensor, ExternModule
+from .core import Effect, ExternModule, Module, ModuleList, Parameter, Tensor
 from .modules import (
+    Conv1D,
+    ConvTranspose1D,
     Embedding,
     IOEffect,
     KVCache,
-    Linear,
-    Conv1D,
-    ConvTranspose1D,
     LayerNorm,
+    Linear,
+    MultiLinear,
     RMSNorm,
 )
 from .op import *