Skip to content

Commit 4305533

Browse files
spike-zhuspike-zhu
committed
feat: add infinicore.nn.Module referencing torch.nn.Module and test case
1 parent 2c168b6 commit 4305533

4 files changed

Lines changed: 863 additions & 0 deletions

File tree

‎python/infinicore/nn/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .modules import *

‎python/infinicore/nn/modules/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .module import InfiniCoreModule as Module

0 commit comments

Comments
 (0)