Mojo module
packing
Structs
-
BTileGenerator: Struct to encapsulate a tile of B that supports prepacking. -
PackMatrixCols: Pack columns from a matrix into the mlas packed layout and extract inner vectors of columns into the packed inner dimension, e.g. extracts [X, Y] and packs as [Yo][X][Yi]. -
PackMatrixRows: Pack rows from a matrix into the mlas packed layout and extract inner vectors of rows into the packed inner dimension, e.g. extract tile [X, Y] and pack into [Xo][Y][Xi].
Functions
-
pack_b: Utility function to pack the entire B matrix, such that each [tile_n // inner_size, tile_k, inner_size] tile of src is contiguous in dst. -
pack_b_ndbuffer: -
pack_matmul_b_shape_func: -
pack_transposed_b_ndbuffer:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!