Mojo function
dispatch_materialized_mask
dispatch_materialized_mask[dtype: DType, layout: Layout, //, callback_fn: callback_fn_type](mask_nd: LayoutTensor[dtype, layout, mask_nd.origin], start_pos_nd: OptionalReg[LayoutTensor[DType.uint32, Layout.row_major(VariadicList(-1)), ImmutAnyOrigin]] = None)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!