Searched refs:LinearBlockSparsePattern (Results 1 – 2 of 2) sorted by relevance
20 class LinearBlockSparsePattern: class29 LinearBlockSparsePattern.rlock.acquire()30 LinearBlockSparsePattern.prev_row_block_size = (31 LinearBlockSparsePattern.row_block_size33 LinearBlockSparsePattern.prev_col_block_size = (34 LinearBlockSparsePattern.col_block_size36 LinearBlockSparsePattern.row_block_size = row_block_size37 LinearBlockSparsePattern.col_block_size = col_block_size43 LinearBlockSparsePattern.row_block_size = (44 LinearBlockSparsePattern.prev_row_block_size[all …]
11 from torch.ao.nn.sparse.quantized.utils import LinearBlockSparsePattern179 row_block_size, col_block_size = LinearBlockSparsePattern.block_size()