Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp2451 const auto outer_selected_idx_tensor = at::from_blob( in index_select_sparse_cpu() local
2459 ? make_output(inner_selected_idx_tensor, outer_selected_idx_tensor) in index_select_sparse_cpu()
2460 : make_output(outer_selected_idx_tensor, inner_selected_idx_tensor); in index_select_sparse_cpu()