huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sketch the conv1d op.

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-core/src/op.rs
View File

@ -12,6 +12,14 @@ pub(crate) enum Op {
Embedding(Tensor, Tensor),
WhereCond(Tensor, Tensor, Tensor),
#[allow(dead_code)]
Conv1D {
arg: Tensor,
kernel: Tensor,
padding: usize,
stride: usize,
},
Cat(Vec<Tensor>, usize),
#[allow(dead_code)] // add is currently unused.