[subset] rename _collect_subset_layout to _collect_layout_indices. Better describes what the function does.