Searched defs:not_zero_mask (Results 1 – 1 of 1) sorted by relevance
909 Tensor not_zero_mask, in _not_zero_mask_to_col_row_indices()1054 …auto not_zero_mask = blocked_layout ? _batch_tile_tensor(self_mask, blocksize, dense_dim) : self_m… in dense_to_sparse_compressed() local