summaryrefslogtreecommitdiff
path: root/torch/nn
AgeCommit message (Expand)AuthorFilesLines
2018-12-10Improve the docs of interpolate(align_corners=) (#14806)Yuxin Wu1-6/+14
2018-12-06Fixed ConvT docstring (#14876)Jon Crall1-12/+13
2018-12-06Consistent formatting in losses' docsRicardo Cuenca1-21/+21
2018-12-04Don't DCE PythonOpAdam Paszke1-0/+3
2018-12-04support conv transpose in scriptElias Ellison1-31/+57
2018-12-04Making dist.get_default_group private for PT1 release (#14767)Teng Li1-3/+7
2018-12-04Enable testing on Loss modules (#14778)David Riazati2-3/+3
2018-12-04Add tests for dropout/batchnorm train/eval, remove training constants (#14780)Wanchao Liang4-5/+5
2018-12-04BatchNorm support not tracking statsDavid Riazati1-5/+7
2018-12-04Add resnet test, convert more modules (#14437)Wanchao Liang2-11/+11
2018-12-04Loss (#14720)Elias Ellison1-3/+68
2018-12-04Add new reduction mode in kl_div (#14457)Ailing Zhang2-34/+47
2018-12-04interpolate (#14123)Elias Ellison1-8/+8
2018-12-03Add Pooling modules to Script (#14527)David Riazati2-6/+49
2018-12-03Add fractional_max_pool2d to standard libDavid Riazati2-13/+35
2018-12-03Add GroupNorm to standard library (#14722)David Riazati1-0/+5
2018-12-03enable NoneValue parameter assignment for WeakScriptModule (#14715)Wanchao Liang3-2/+6
2018-12-03Add linear + pixelshuffle modules to standard libDavid Riazati2-0/+9
2018-12-03Fixed DistributedDataParallel state pickling for multi-gpus (#14690)Teng Li1-6/+17
2018-12-02Fixed DistributedDataParallel cannot kick off all-reduce in a corner case (#1...Teng Li1-5/+52
2018-11-30Revert existing no_grad_embedding_renorm_ from aten (#14639)Elias Ellison1-2/+8
2018-11-29Add binary cross entropy to standard libDavid Riazati1-2/+5
2018-11-29Add InstanceNorm, Distance modules to ScriptDavid Riazati6-48/+91
2018-11-29Revert D13268293: [pytorch][PR] [jit] Add InstanceNorm, Distance modules to S...David Riazati6-92/+48
2018-11-29Add InstanceNorm, Distance modules to ScriptDavid Riazati6-48/+92
2018-11-29Add `List` to annotations (#14482)David Riazati1-2/+1
2018-11-29Add broadcast list default arg support (#14361)David Riazati1-27/+45
2018-11-29Fixed typo for BCEWithLogitLoss doc comments (#14532)Matthew Heidemann1-1/+1
2018-11-29typo in Module docstringRyan Moore1-1/+1
2018-11-28Use nn module tests in test_jit (#14238)David Riazati2-6/+41
2018-11-28Support Embedding + EmbeddingBag in Script + (Ignore flakey test) (#14509)Elias Ellison2-15/+40
2018-11-28pointwise_loss (#14134)Elias Ellison1-6/+31
2018-11-28Update Tensor doc (#14339)albanD1-0/+10
2018-11-28Revert D13219647: [pytorch][PR] Support Embedding + EmbeddingBag in ScriptEdward Yang2-40/+15
2018-11-28Support Embedding + EmbeddingBag in Script (#14415)Elias Ellison2-15/+40
2018-11-28Revert D13192230: [pytorch][PR] [jit] Use nn module tests in test_jitDavid Riazati2-41/+6
2018-11-27Use nn module tests in test_jit (#14238)David Riazati2-6/+41
2018-11-27Add poisson_nll_loss to scriptDavid Riazati1-0/+1
2018-11-27Add boolean dispatch for function overloading (#14425)David Riazati1-25/+133
2018-11-27Move Affine grid to C++ (#14392)Elias Ellison2-76/+8
2018-11-27Revert D13192228: [pytorch][PR] [jit] Add boolean dispatch for function overl...David Riazati1-133/+25
2018-11-27Add boolean dispatch for function overloading (#14081)David Riazati1-25/+133
2018-11-21Convert gumbel_softmax, lp pooling weak functions and modules (#14232)Wanchao Liang2-9/+32
2018-11-21Removed redundant allreduce options in DDP (#14208)Teng Li1-3/+0
2018-11-21Add list inequality operator (#14129)David Riazati1-1/+3
2018-11-20Allow undefined tensors as constants (#14120)David Riazati1-18/+38
2018-11-20Export BatchNorm functional and module, add necessary JIT support (#14016)Wanchao Liang2-5/+27
2018-11-16Keep `ModuleList` consistent with python `list` in `__setitem__` function. (#...lyuwenyu1-7/+12
2018-11-15Fixed THD DistributedDataParallel not picklable (#14051)Teng Li1-2/+3
2018-11-15Convert more weak functions (#14003)David Riazati1-39/+85
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-29 13:36:51 +0000