Skip to main content

DispatchStftFn

Type Alias DispatchStftFn 

Source
pub type DispatchStftFn = fn(usize, &DeviceTensor, &DeviceTensor, &DeviceTensor) -> TractResult<()>;
Expand description

Per-backend STFT kernel launcher: (stride, input, window, output). input is interleaved-complex f32 [lead.., T, 2], window the pre-padded real window [frame], output [lead.., frames, frame, 2]. The kernel reads the frame length from the output shape (output[axis + 1]).