Clear out the outer name scope so the ops created inside `tf.while_loop`

don't get "while/" as name prefix.

PiperOrigin-RevId: 351689649

orbit/utils/loop_fns.py

@@ -112,6 +112,9 @@ def create_tf_while_loop_fn(step_fn):
           "cause unnecessary retracing when wrapped by `tf.function`.")
 
     for _ in tf.range(num_steps):
+      # Clear out the outer name scope so the ops created inside `tf.while_loop`
+      # don't get "while/" as name prefix.
+      with tf.name_scope(""):
         step_fn(iterator)
 
   return loop_fn
@@ -157,6 +160,9 @@ def create_tf_while_loop_fn_with_state(step_fn):
           "cause unnecessary retracing when wrapped by `tf.function`.")
 
     for _ in tf.range(num_steps):
+      # Clear out the outer name scope so the ops created inside `tf.while_loop`
+      # don't get "while/" as name prefix.
+      with tf.name_scope(""):
         # Relax the shapes within the loop, so the shape of `state` can change
         # across iterations. This is useful to aggregate outputs from each step
         # and concat to `state`.