summaryrefslogtreecommitdiff
path: root/candle-core/src/conv.rs
Commit message (Collapse)AuthorAgeFilesLines
* 20241118 docs (#2629)zachcp2024-11-191-0/+2
| | | | | | | | | | | | | | | | | * module docs * varbuilder gguf docs * add a link to gguf files * small additonal mod doc titles * safetensor docs * more core docs * more module docs in canlde_core * 2 more link fixes
* Support for groups in conv-transpose1d. (#1731)Laurent Mazare2024-02-181-18/+41
| | | | | * Groups support in conv-transpose-1d. * Remove dangling file.
* Test for the transposed conv1d. (#1254)Laurent Mazare2023-11-031-1/+1
|
* Add the conv-transpose1d op. (#1251)Laurent Mazare2023-11-031-0/+70
| | | | | * Skeleton structure for conv-transpose1d. * CPU implementation for conv-transpose1d.
* Remove some unusued bits. (#1067)Laurent Mazare2023-10-091-2/+1
|
* Override the default cudnn heuristics. (#957)Laurent Mazare2023-09-251-0/+16
|
* Dilated convolutions (#657)Laurent Mazare2023-08-291-10/+17
| | | | | | | | | | | | | | | | | | | * Add the dilation parameter. * Restore the basic optimizer example. * Dilation support in cudnn. * Use the dilation parameter in the cpu backend. * More dilation support. * No support for dilation in transposed convolutions. * Add dilation to a test. * Remove a print. * Helper function.
* Backprop for conv2d. (#638)Laurent Mazare2023-08-281-8/+4
| | | | | | | | | | | * Start adding backprop for conv2d. * Backprop for conv2d. * Bugfix + start adding a conv2d test. * Conv2d backprop testing. * More conv fixes.
* Add conv-transpose. (#635)Laurent Mazare2023-08-281-0/+78
| | | | | | | | | | | * Add conv-transpose. * Return zeros for now. * Naive CPU implementation. * Add a conv-transpose test + fix the cpu implementation. * Add a second test.
* Add Efficientnet (#572)Laurent Mazare2023-08-231-6/+10
| | | | | | | | | * EfficientNet. * Complete the efficientnet implementation. * Improve group handling. * Get the efficientnet to work.
* Add some group parameter to convolutions. (#566)Laurent Mazare2023-08-231-0/+112
| | | | | | | | | | | | | * Add some group parameter to convolutions. * Avoid some unnecessary groups checks. * Move the tensor convolution bits. * Properh handling of groups. * Bump the crate version. * And add a changelog.
* Add a conv1d benchmark based on the whisper sizes. (#377)Laurent Mazare2023-08-091-5/+2
| | | | | * Add a conv1d benchmark based on the whisper sizes. * Enforce the batch-dim in conv1d.
* Add more conv2d support. (#340)Laurent Mazare2023-08-081-0/+29
| | | | | | | * Add more conv2d support. * Conv2d cpu work. * Conv2d output shape.
* Modular backends (#138)Laurent Mazare2023-07-111-1/+1
| | | | | * Add some trait to formalize backends. * Use the generic backend trait.
* Proper conv1d dispatch.laurent2023-07-041-4/+7
|
* Add more of the conv1d op.laurent2023-07-041-0/+24