Skip to main content

Mojo function

dot_at_b_impl

dot_at_b_impl(c: LayoutTensor[DType.float32, Layout.row_major(VariadicList(16, 16)), MutAnyOrigin], a: LayoutTensor[DType.float32, Layout.row_major(VariadicList(16, 16)), ImmutAnyOrigin], b: LayoutTensor[DType.float32, Layout.row_major(VariadicList(16, 16)), ImmutAnyOrigin])

dot_at_b_impl(c: LayoutTensor[DType.float16, Layout.row_major(VariadicList(32, 32)), MutAnyOrigin], a: LayoutTensor[DType.float16, Layout.row_major(VariadicList(32, 32)), ImmutAnyOrigin], b: LayoutTensor[DType.float16, Layout.row_major(VariadicList(32, 32)), ImmutAnyOrigin])

Was this page helpful?