MCPcopy Create free account

hub / github.com/MzeroMiko/mamba-mini / functions

Functions35 in github.com/MzeroMiko/mamba-mini

⨍ Functions 35◇ Types & classes 3↳ Endpoints 2
↓ 2 callersMethodbackward
(ctx, doys: torch.Tensor, *args)
test_selective_scan_easy.py:184
↓ 2 callersMethodbackward
(ctx, doys: torch.Tensor, *args)
test_selective_scan.py:184
↓ 2 callersMethodsave_for_backward
(ctx, *args)
test_selective_scan_easy.py:108
↓ 2 callersFunctionss_chunk
(us, dts, As, Bs, Cs, h0, mask)
test_selective_scan_easy.py:396
↓ 1 callersFunctionbuild_api_selective_scan
(chunksize=64)
test_selective_scan_easy.py:844
↓ 1 callersFunctionbuild_api_selective_scan
(chunksize=64)
test_selective_scan.py:381
↓ 1 callersFunctioncut_chunk
(us, dts, Bs, Cs, chunksize=chunksize)
test_selective_scan_easy.py:488
↓ 1 callersMethodforward
(ctx, us, dts, As, Bs, Cs, Ds, delta_bias=None, delta_softplus=False, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:113
↓ 1 callersMethodforward
(ctx, us, dts, As, Bs, Cs, Ds, delta_bias=None, delta_softplus=False, return_last_state=False, chunksize=64)
test_selective_scan.py:113
↓ 1 callersMethodsave_for_backward
(ctx, *args)
test_selective_scan.py:108
↓ 1 callersFunctionselective_scan_chunk
partial(h) / partial(t) = Ah + Bu; y = Ch + Du; => partial(h*exp(-At)) / partial(t) = Bu*exp(-At); => h_t = h_0 + sum_{0}_{t}
test_selective_scan_easy.py:26
↓ 1 callersFunctionselective_scan_chunk
partial(h) / partial(t) = Ah + Bu; y = Ch + Du; => partial(h*exp(-At)) / partial(t) = Bu*exp(-At); => h_t = h_0 + sum_{0}_{t}
test_selective_scan.py:26
↓ 1 callersFunctionselective_scan_easy
# B: batch_size, G: groups, D: dim, N: state dim, L: seqlen us: B, G * D, L dts: B, G * D, L As: G * D, N Bs: B, G, N, L Cs:
test_selective_scan_easy.py:14
↓ 1 callersFunctionselective_scan_easy
# B: batch_size, G: groups, D: dim, N: state dim, L: seqlen us: B, G * D, L dts: B, G * D, L As: G * D, N Bs: B, G, N, L Cs:
test_selective_scan.py:14
↓ 1 callersFunctionselective_scan_fn
(u, delta, A, B, C, D, z=None, delta_bias=None, delta_softplus=None, return_last_state=False)
test_selective_scan_easy.py:845
↓ 1 callersFunctionselective_scan_fn
(u, delta, A, B, C, D, z=None, delta_bias=None, delta_softplus=None, return_last_state=False)
test_selective_scan.py:384
↓ 1 callersFunctionselective_scan_ref
u: r(B D L) delta: r(B D L) A: c(D N) or r(D N) B: c(D N) or r(B N L) or r(B N 2L) or r(B G N L) or (B G N L) C: c(D N) or r(B N
test_selective_scan_easy.py:853
↓ 1 callersFunctionselective_scan_ref
u: r(B D L) delta: r(B D L) A: c(D N) or r(D N) B: c(D N) or r(B N L) or r(B N 2L) or r(B G N L) or (B G N L) C: c(D N) or r(B N
test_selective_scan.py:401
↓ 1 callersFunctionss_chunk_h
(cws, ht1)
test_selective_scan_easy.py:514
↓ 1 callersFunctionss_chunk_h1y1
(qs, ks, vs, ws=None, As=None, ts=None, dts=None, mask=_mask_tril, scale=1)
test_selective_scan_easy.py:497
↓ 1 callersFunctionss_chunk_y
(y1, hs, q_mul_w)
test_selective_scan_easy.py:524
↓ 1 callersFunctiontest_selective_scan
(is_variable_B, is_variable_C, varBC_groups, has_D, has_z, has_delta_bias, delta_softp
test_selective_scan_easy.py:941
↓ 1 callersFunctiontest_selective_scan
(is_variable_B, is_variable_C, varBC_groups, has_D, has_z, has_delta_bias, delta_softp
test_selective_scan.py:484
Methodbackward
(ctx, doys: torch.Tensor, *args)
test_selective_scan_easy.py:683
Methodforward
(ctx, us, dts, As, Bs, Cs, Ds, delta_bias=None, delta_softplus=False, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:596
Methodrev_comsum
(x)
test_selective_scan_easy.py:193
Methodrev_comsum
(x)
test_selective_scan.py:193
Methodrev_comsum_dim_2
(x)
test_selective_scan_easy.py:690
Functionselective_scan_easy_fwdbwd
(u, delta, A, B, C, D, delta_bias=None, delta_softplus=None, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:380
Functionselective_scan_easyv2
(us, dts, As, Bs, Cs, Ds, delta_bias=None, delta_softplus=False, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:391
Functionselective_scan_easyv2_fwdbwd
(u, delta, A, B, C, D, delta_bias=None, delta_softplus=None, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:831
Functionselective_scan_easyv3
(us, dts, As, Bs, Cs, Ds, delta_bias=None, delta_softplus=False, return_last_state=False, chunksize=64)
test_selective_scan_easy.py:467
Methodss_chunk
(us, dts, As, Bs, Cs, h0, mask)
test_selective_scan_easy.py:602
Functionss_chunk_h1y1_dk1
(qs, ks, vs, ws=None, As=None, ts=None, dts=None, mask=_mask_tril, scale=1)
test_selective_scan_easy.py:530
Functionss_chunk_y_dk1
(y1, hs, q_mul_w)
test_selective_scan_easy.py:549