Add the conv-transpose1d op. (#1251)

* Skeleton structure for conv-transpose1d.

* CPU implementation for conv-transpose1d.
This commit is contained in:
Laurent Mazare
2023-11-03 09:44:46 +01:00
committed by GitHub
parent 6975c65112
commit be4555c5a5
8 changed files with 221 additions and 0 deletions

View File

@ -39,6 +39,14 @@ pub trait BackendStorage: Sized {
_params: &crate::conv::ParamsConv1D,
) -> Result<Self>;
fn conv_transpose1d(
&self,
_l: &Layout,
_kernel: &Self,
_kernel_l: &Layout,
_params: &crate::conv::ParamsConvTranspose1D,
) -> Result<Self>;
fn conv2d(
&self,
_l: &Layout,