Skip to main content

split_heads

forge::backend::wgpu::ops

Function split_heads 

Source 
pub fn split_heads(
    qkv: &WgpuStorage,
    t: usize,
    c: usize,
    h: usize,
) -> (WgpuStorage, WgpuStorage, WgpuStorage)