MCPcopy Create free account
hub / github.com/evilsocket/cake / depthwise_conv1d_bias_ctx

Method depthwise_conv1d_bias_ctx

cake-core/src/backends/cpu/mod.rs:238–250  ·  view source on GitHub ↗
(
        &self,
        ctx: &Tensor,
        input: &Tensor,
        weight: &Tensor,
        bias: &Tensor,
        kernel_size: usize,
        channels: usize,
    )

Source from the content-addressed store, hash-verified

236 }
237
238 fn depthwise_conv1d_bias_ctx(
239 &self,
240 ctx: &Tensor,
241 input: &Tensor,
242 weight: &Tensor,
243 bias: &Tensor,
244 kernel_size: usize,
245 channels: usize,
246 ) -> Result<Tensor> {
247 // Concatenate [ctx, input] along time dimension, then run conv
248 let combined = Tensor::cat(&[ctx, input], 2)?;
249 self.depthwise_conv1d_bias(&combined, weight, bias, kernel_size, channels)
250 }
251
252 fn add3(&self, a: &Tensor, b: &Tensor, c: &Tensor) -> Result<Tensor> {
253 (a + b)? + c

Callers 6

forwardMethod · 0.45
forward_cachedMethod · 0.45
forwardMethod · 0.45
forward_cachedMethod · 0.45

Calls 1

depthwise_conv1d_biasMethod · 0.45