From 41416d23762e6cf21aa6e1c3adb6d457f15d7071 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Sat, 17 Feb 2024 18:50:55 +0100
Subject: Expose more conv1d functions/structs. (#1726)

---
 candle-nn/src/conv.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'candle-nn/src/conv.rs')

diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs
index b1168405..6734ab1f 100644
--- a/candle-nn/src/conv.rs
+++ b/candle-nn/src/conv.rs
@@ -302,6 +302,22 @@ pub fn conv1d(
     Ok(Conv1d::new(ws, Some(bs), cfg))
 }
 
+pub fn conv1d_no_bias(
+    in_channels: usize,
+    out_channels: usize,
+    kernel_size: usize,
+    cfg: Conv1dConfig,
+    vb: crate::VarBuilder,
+) -> Result<Conv1d> {
+    let init_ws = crate::init::DEFAULT_KAIMING_NORMAL;
+    let ws = vb.get_with_hints(
+        (out_channels, in_channels / cfg.groups, kernel_size),
+        "weight",
+        init_ws,
+    )?;
+    Ok(Conv1d::new(ws, None, cfg))
+}
+
 pub fn conv_transpose1d(
     in_channels: usize,
     out_channels: usize,
-- 
cgit v1.2.3