Mojo function
pack_matmul_b_shape_func
pack_matmul_b_shape_func[a_type: DType, c_type: DType, transpose_in_0: Bool](b_input: TileTensor[b_input.dtype, b_input.LayoutType, b_input.origin, linear_idx_type=b_input.linear_idx_type, element_size=b_input.element_size], kernel_type_m: Int = 0) -> IndexList[2]
TileTensor primary implementation of pack_matmul_b_shape_func.
Takes kernel_type_m directly instead of extracting it from a_shape
static shape params (0 = dynamic M).
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!