code: def _tf_data_yield_flat_up_to(shallow_tree, input_tree):
      """Yields elements `input_tree` partially flattened up to `shallow_tree`."""
      if _tf_data_is_nested(shallow_tree):
          for shallow_branch, input_branch in zip(
              _tf_data_yield_value(shallow_tree),
              _tf_data_yield_value(input_tree),
          ):
              for input_leaf in _tf_data_yield_flat_up_to(
                  shallow_branch, input_branch
              ):
                  yield input_leaf
      else:
          yield input_tree
  
imports: import collections
  import enum
  import wrapt
  from tensorflow.python import pywrap_tensorflow
  from tensorflow.python.platform import tf_logging
  from tensorflow.python.util import _pywrap_utils
  from tensorflow.python.util.compat import collections_abc
  from tensorflow.python.util.custom_nest_protocol import CustomNestProtocol
label_desc: Use image node_id 27 for calling a global function with example usage: _tf_data_yield_flat_up_to(shallow_tree, input_tree) without return types