Skip to main content
Log in

Mojo function

get_partitioned_matmul_mojo_shape

get_partitioned_matmul_mojo_shape[b_type: DType, kernel_rows: Int, kernel_cols: Int, use_i8mm: Bool](m: Int, n: Int, k: Int, num_tasks: Int) -> Index[2]

Was this page helpful?