For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
gated_delta_conv1d_verify_fwd_shape
def gated_delta_conv1d_verify_fwd_shape(qkv_input_ragged: T, conv_weight: T, conv_state: T, slot_idx: T, input_row_offsets: T, verify_width: T) -> IndexList[Int(2)]
Computes the output shape for gated_delta_conv1d_verify_fwd.
Args:
- qkv_input_ragged (
T): Ragged QKV input,[total_seq_len, conv_dim]. - conv_weight (
T): Convolution filter,[conv_dim, kernel_size]. - conv_state (
T): Conv-state pool,[max_slots, conv_dim, kernel_size-1]. - slot_idx (
T): Pool row per batch item,[batch_size]. - input_row_offsets (
T): Ragged offsets,[batch_size + 1]. - verify_width (
T):[K], read for its shape only.
Returns: