Searched refs:split_edges (Results 1 – 1 of 1) sorted by relevance
2595 auto split_edges = at::cumsum(split_counts, /*dim=*/0) - 1; in compute_T18_scale_square() local2596 auto unique_s = sorted_s.index_select(0, split_edges).clamp(/*min=*/0); in compute_T18_scale_square()