code: def _get_all_ranks(st):
      """Get ranks of all submessages of a StructuredTensor."""
      fields = st.field_names()
      all_ranks = {(): st.rank}
      for k in fields:
          v = st.field_value(k)
          if isinstance(v, StructuredTensor):
              for k2, v2 in _get_all_ranks(v).items():
                  all_ranks[(k,) + k2] = v2
      return all_ranks
  
imports: from typing import Sequence
  from tensorflow.core.config import flags
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops.ragged import dynamic_ragged_shape
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.ops.ragged.row_partition import RowPartition
  from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
  from tensorflow.python.util import deprecation
  from tensorflow.python.util import dispatch
label_desc: Use image node_id 25 for calling a global function with example usage: _get_all_ranks(st) and returns: all_ranks