Skeleton for the avg-pool2d and upsample-nearest2d ops. (#337)

* Skeleton for the avg-pool2d and upsample-nearest2d ops.

* Preliminary conv2d support.
This commit is contained in:
Laurent Mazare
2023-08-07 17:15:38 +02:00
committed by GitHub
parent f53a333ea9
commit 2345b8ce3f
7 changed files with 88 additions and 17 deletions

View File

@ -266,6 +266,24 @@ impl Storage {
}
}
pub(crate) fn avg_pool2d(
&self,
_layout: &Layout,
_kernel_size: (usize, usize),
_stride: (usize, usize),
) -> Result<Self> {
todo!()
}
pub(crate) fn upsample_nearest2d(
&self,
_layout: &Layout,
_h: usize,
_w: usize,
) -> Result<Self> {
todo!()
}
pub(crate) fn where_cond(
&self,
layout: &Layout,