expand_active_nodes -> child_index_to_tag_mask
Use tabulate instead of transform to compute child tag masks