↓ 1 callersMethodrel_shiftRelative position shift: convert from (batch/heads, ?, seq, 2*seq-1) to (batch/heads, ?, seq, seq). For row i, extract columns [seq-1-i .. 2*seq-1-i],
cake-core/src/models/luxtts/rel_pos_attention.rs:115
↓ 1 callersMethodrocblas_gemm(&self, a: *const f32, b: *const f32, m: usize, k: usize, n: usize)
cake-core/src/backends/rocm/mod.rs:264
↓ 1 callersFunctionrun_zero_config_worker(
name: &str,
cluster_key: &str,
address: &str,
cache_dir: &std::path::Path,
_model: &str,
cake-mobile/src/lib.rs:329
Methodadaln_modulate(&self, x: &Tensor, nw: &Tensor, sc: &Tensor, sh: &Tensor, e: f32)
cake-core/src/backends/rocm/mod.rs:410
Methodadaln_modulate(&self, x: &Tensor, norm_weight: &Tensor, scale: &Tensor, shift: &Tensor, eps: f32)
cake-core/src/backends/metal/mod.rs:982
Methodadaln_modulate(
&self,
x: &Tensor,
nw: &Tensor,
scale: &Tensor,
shift: &Tensor,
cake-core/src/backends/vulkan/mod.rs:1472
Methodadaln_modulate(&self, x: &Tensor, norm_weight: &Tensor, scale: &Tensor, shift: &Tensor, eps: f32)
cake-core/src/backends/cuda/mod.rs:148