Sketch the conv1d op.

This commit is contained in:
laurent
2023-07-04 10:52:34 +01:00
parent e6b01d0c18
commit 3aac1047fe
7 changed files with 97 additions and 1 deletions

View File

@ -627,6 +627,17 @@ impl CpuStorage {
WCond(pred, layout).map(t, t_l, f, f_l)
}
pub(crate) fn conv1d(
&self,
_l: &Layout,
_kernel: &Self,
_kernel_l: &Layout,
_padding: usize,
_stride: usize,
) -> Result<Self> {
todo!()
}
pub(crate) fn embedding(&self, ids_l: &Layout, rhs: &Self, rhs_l: &Layout) -> Result<Self> {
let ids = self.as_slice::<u32>()?;
let (vocab_size, hidden_size) = rhs_l.shape().r2()?;