summaryrefslogtreecommitdiff
path: root/candle-nn/src/conv.rs
Commit message (Expand)AuthorAgeFilesLines
* Transposed conv1d in candle-nn. (#1252)Laurent Mazare2023-11-031-0/+94
* Add fuse-conv-bn method for Conv2d (#1196)jamjamjon2023-10-271-0/+21
* Add clone to various nn layers. (#910)Laurent Mazare2023-09-201-3/+3
* Remove the parameters for the Wuerstchen layer-norm. (#879)Laurent Mazare2023-09-171-1/+1
* Add weight, bias, and hidden_size methods (#816)Eric Buehler2023-09-111-0/+16
* TinyViT backbone for segment-anything. (#787)Laurent Mazare2023-09-091-13/+13
* More segment-anything. (#763)Laurent Mazare2023-09-071-0/+11
* Expose the conv2d-transpose layers. (#761)Laurent Mazare2023-09-071-0/+41
* Add a nn layer for conv-transpose2d. (#760)Laurent Mazare2023-09-071-1/+50
* Dilated convolutions (#657)Laurent Mazare2023-08-291-0/+6
* VarBuilder cleanup (#627)Laurent Mazare2023-08-271-5/+5
* Add Efficientnet (#572)Laurent Mazare2023-08-231-3/+17
* Add some group parameter to convolutions. (#566)Laurent Mazare2023-08-231-2/+16
* Add a couple functions required for yolo. (#527)Laurent Mazare2023-08-201-0/+16
* Add a simple Module trait and implement it for the various nn layers (#500)Laurent Mazare2023-08-181-2/+6
* Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)Laurent Mazare2023-08-071-2/+10
* Add a stable diffusion example (#328)Laurent Mazare2023-08-061-0/+81
* Rename the .r functions to .dims so as to be a bit more explicit. (#220)Laurent Mazare2023-07-221-1/+1
* Add some documentation and test to the linear layer. (#151)Laurent Mazare2023-07-121-0/+1
* Move the conv1d layer to candle_nn. (#117)Laurent Mazare2023-07-101-0/+49