mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 11:37:11 +00:00
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:
@ -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,
|
||||
|
Reference in New Issue
Block a user