Skip to main content

Mojo function

apple_matmul

apple_matmul[*, transpose_b: Bool = False, elementwise_lambda_fn: Optional[elementwise_epilogue_type] = None](cblas_gemm_fn: cblas_gemm_type, c: NDBuffer[c.dtype, c.origin, c.shape, c.strides, alignment2=c.alignment2, address_space=c.address_space, exclusive=c.exclusive], a: NDBuffer[a.dtype, a.origin, a.shape, a.strides, alignment2=a.alignment2, address_space=a.address_space, exclusive=a.exclusive], b: NDBuffer[b.dtype, b.origin, b.shape, b.strides, alignment2=b.alignment2, address_space=b.address_space, exclusive=b.exclusive])

apple_matmul[*, transpose_b: Bool = False, elementwise_lambda_fn: Optional[elementwise_epilogue_type] = None](c: NDBuffer[c.dtype, c.origin, c.shape, c.strides, alignment2=c.alignment2, address_space=c.address_space, exclusive=c.exclusive], a: NDBuffer[a.dtype, a.origin, a.shape, a.strides, alignment2=a.alignment2, address_space=a.address_space, exclusive=a.exclusive], b: NDBuffer[b.dtype, b.origin, b.shape, b.strides, alignment2=b.alignment2, address_space=b.address_space, exclusive=b.exclusive])

Was this page helpful?