Methodadd_rms_norm(
&self,
a: &Tensor,
b: &Tensor,
weight: &Tensor,
eps: f32,
)
cake-core/src/backends/vulkan/mod.rs:1405
Methodattention(&self, q: &Tensor, k: &Tensor, v: &Tensor, scale: f32, causal: bool)
cake-core/src/backends/rocm/mod.rs:380
Methodattention(&self, q: &Tensor, k: &Tensor, v: &Tensor, scale: f32, causal: bool)
cake-core/src/backends/metal/mod.rs:726
Methodattention(
&self,
q: &Tensor,
k: &Tensor,
v: &Tensor,
scale: f32,
causa
cake-core/src/backends/vulkan/mod.rs:1259