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?
Thank you! We'll create more content like this.
Thank you for helping us improve!