Move MatchShapeCoveringDynamicIndex function to while_loop_unroller.h and update other usages to use the same method.
Move MatchShapeCoveringDynamicIndex function to while_loop_unroller.h and update other usages to use the same method.

Add support for matching ds and dus that are inside (nested)fusions within loop bodies.
