Rename __debug_shape_info, __debug_shape_worker to __debug_tree_shape_info,
__debug_tree_shape_worker to add context.