Searched refs:outer_selected_idx_tensor (Results 1 – 1 of 1) sorted by relevance
2451 const auto outer_selected_idx_tensor = at::from_blob( in index_select_sparse_cpu() local2459 ? 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()