language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-09-29 04:24:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add torch type declaration

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-12-09 12:05:07 +09:00
parent 480c9e4f83
commit 1aa61cd6a6
24 changed files with 84 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

0
crates/erg_compiler/lib/external/torch.d/utils.d/__init__.d.er vendored Normal file
View file

2
crates/erg_compiler/lib/external/torch.d/utils.d/data.d/__init__.d.er vendored Normal file
View file

@ -0,0 +1,2 @@
{.DataLoader;} = pyimport "./dataloader"
{.Dataset;} = pyimport "./dataset"

9
crates/erg_compiler/lib/external/torch.d/utils.d/data.d/dataloader.d.er vendored Normal file
View file

@ -0,0 +1,9 @@
dataset = pyimport "./dataset"
.DataLoader: ClassType
.DataLoader.
__call__: (
dataset: dataset.Dataset,
batch_size := Nat,
shuffle := Bool,
) -> .DataLoader

1
crates/erg_compiler/lib/external/torch.d/utils.d/data.d/dataset.d.er vendored Normal file
View file

@ -0,0 +1 @@
.Dataset: ClassType

28
crates/erg_compiler/lib/external/torch.d/utils.d/optim.d/__init__.d.er vendored Normal file
View file

@ -0,0 +1,28 @@
.Optimizer: ClassType
.ASGD: ClassType
.ASGD <: .Optimizer
.Adadelta: ClassType
.Adadelta <: .Optimizer
.Adagrad: ClassType
.Adagrad <: .Optimizer
.Adam: ClassType
.Adam <: .Optimizer
.AdamW: ClassType
.AdamW <: .Optimizer
.Adamax: ClassType
.Adamax <: .Optimizer
.LBFGS: ClassType
.LBFGS <: .Optimizer
.NAdam: ClassType
.NAdam <: .Optimizer
.RAdam: ClassType
.RAdam <: .Optimizer
.RMSprop: ClassType
.RMSprop <: .Optimizer
.Rprop: ClassType
.Rprop <: .Optimizer
.SGD: ClassType
.SGD <: .Optimizer
.SparseAdam: ClassType
.SparseAdam <: .Optimizer