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