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

Method silu_mul

cake-core/src/backends/cpu/mod.rs:87–89  ·  view source on GitHub ↗
(&self, gate: &Tensor, up: &Tensor)

Source from the content-addressed store, hash-verified

85 }
86
87 fn silu_mul(&self, gate: &Tensor, up: &Tensor) -> Result<Tensor> {
88 candle_nn::ops::silu(&gate.contiguous()?)? * up.contiguous()?
89 }
90
91 fn stable_softplus(&self, x: &Tensor) -> Result<Tensor> {
92 // ln(1 + exp(clamp(x, -inf, 88))) with max(x, result)

Callers 14

fused_silu_mul_cpuFunction · 0.45
fused_silu_mul_gpuFunction · 0.45
vulkan_dispatch_overheadFunction · 0.45
vulkan_silu_mulFunction · 0.45
vulkan_mlp_fullFunction · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45
forwardMethod · 0.45

Calls 1

siluFunction · 0.85

Tested by 3