I think ```__debug_shape_info``` and ```__debug_shape_worker``` should be called ```__debug_tree_shape_xxx```. I know they become long, but the additional context adds value.