Searched refs:_diag_len (Results 1 – 1 of 1) sorted by relevance
644 def _diag_len(dim1, dim2, offset): function655 diag_chunks.append(_diag_len(c1, c2, k))671 right_shape = (_diag_len(a.shape[axis1], a.shape[axis2], offset),)