pymllm.nn.functional¶

Attributes¶

`matmul_impl_default`
`matmul_impl_gguf`
`matmul_impl_blas`
`matmul_impl_mllmblas`

Functions¶

matmul(lhs, rhs[, transpose_lhs, transpose_rhs, impl])

Module Contents¶

pymllm.nn.functional.matmul_impl_default: int¶

pymllm.nn.functional.matmul_impl_gguf: int¶

pymllm.nn.functional.matmul_impl_blas: int¶

pymllm.nn.functional.matmul_impl_mllmblas: int¶

pymllm.nn.functional.matmul(lhs, rhs, transpose_lhs=False, transpose_rhs=False, impl=matmul_impl_default)¶

Parameters:

lhs (pymllm.ffi.Tensor)
rhs (pymllm.ffi.Tensor)
transpose_lhs (bool)
transpose_rhs (bool)
impl (int)

Return type:

pymllm.ffi.Tensor