For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

selective_scan_update_shape

def selective_scan_update_shape[dtype: DType](state_in: ManagedTensorSlice[Input, static_spec=state_in.static_spec], x: ManagedTensorSlice[Input, static_spec=x.static_spec], dt: ManagedTensorSlice[Input, static_spec=dt.static_spec], A: ManagedTensorSlice[Input, static_spec=A.static_spec], B: ManagedTensorSlice[Input, static_spec=B.static_spec], C: ManagedTensorSlice[Input, static_spec=C.static_spec], D: ManagedTensorSlice[Input, static_spec=D.static_spec], z: ManagedTensorSlice[Input, static_spec=z.static_spec], dt_bias: ManagedTensorSlice[Input, static_spec=dt_bias.static_spec]) -> Tuple[IndexList[3], IndexList[2]]

Returns:

Tuple[IndexList[3], IndexList[2]]