Home
last modified time | relevance | path

Searched refs:split_edges (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinearAlgebra.cpp2595 auto split_edges = at::cumsum(split_counts, /*dim=*/0) - 1; in compute_T18_scale_square() local
2596 auto unique_s = sorted_s.index_select(0, split_edges).clamp(/*min=*/0); in compute_T18_scale_square()