Searched refs:_to_block_triangular_inplace (Results 1 – 1 of 1) sorted by relevance
3550 def _to_block_triangular_inplace(self, d, row_block, col_block): member in TestSparseCompressedTritonKernels3589 input = self._to_block_triangular_inplace(input, row_block, col_block)3758 attn_mask = self._to_block_triangular_inplace(attn_mask, block_size, block_size)