num_k_dims
tile_k
tile_n
max_block_n
A @ B^T
(B @ A^T)^T
B
M == 1
transpose_fn
packer::pack
n
2 * N
transpose
pack_b
should_pack_b
tile_k == 1
M
m
define_dot
A
E2B
E4B
26B-A4B
31B
u32