Add more conv2d support. (#340)

* Add more conv2d support.

* Conv2d cpu work.

* Conv2d output shape.
This commit is contained in:
Laurent Mazare
2023-08-08 07:04:32 +02:00
committed by GitHub
parent d0d7010682
commit b5bb5e056d
7 changed files with 137 additions and 2 deletions

View File

@ -1381,6 +1381,16 @@ impl BackendStorage for CudaStorage {
Ok(Self { slice, device })
}
fn conv2d(
&self,
_l: &Layout,
_kernel: &Self,
_kernel_l: &Layout,
_params: &crate::conv::ParamsConv2D,
) -> Result<Self> {
todo!()
}
fn avg_pool2d(&self, _: &Layout, _: (usize, usize), _: (usize, usize)) -> Result<Self> {
todo!()
}