Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp2424 const auto dim_indices_in_inner_loop = nnz >= index_len; in index_select_sparse_cpu() local
2426 if (dim_indices_in_inner_loop) { in index_select_sparse_cpu()
2458 return dim_indices_in_inner_loop in index_select_sparse_cpu()