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:
Laurent Mazare
2023-08-10 16:23:52 +02:00
committed by GitHub
parent 0b0fa56978
commit c8039579a5
6 changed files with 62 additions and 22 deletions

View File

@ -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;