mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 19:18:50 +00:00
Conv1d optimize (#392)
* Reorder the conv1d loops in the cpu backend. * Optimize the 1d convolution. * Conv1D optimize. * Fix some clippy lints.
This commit is contained in:
@ -40,6 +40,7 @@ pub mod backprop;
|
||||
mod conv;
|
||||
mod convert;
|
||||
pub mod cpu_backend;
|
||||
pub mod cpu_kernels;
|
||||
#[cfg(feature = "cuda")]
|
||||
pub mod cuda_backend;
|
||||
mod device;
|
||||
|
Reference in New Issue
Block a user