parameters: (dequeues, dims)
return_statement: return nest.map_structure_up_to(
        dequeues, _tag_sharding_attribute_for_dequeued_tensor, dequeues, dims)
docstring: Tags appropriate XLA sharding attribute to the dequeued tensors.
  
    Args:
      dequeues: A list of dequeued tensors on TPU.
      dims: A list of integer describes how the tensor is partitioned.
  
    Returns:
      The same dequeues with appropriate xla_sharding attribute.
docstring_summary: Tags appropriate XLA sharding attribute to the dequeued tensors.
docstring_tokens: Tags appropriate XLA sharding attribute to the dequeued tensors .
function: def tag_sharding_attribute_for_dequeued_tensors(dequeues, dims):
    """Tags appropriate XLA sharding attribute to the dequeued tensors.
  
    Args:
      dequeues: A list of dequeued tensors on TPU.
      dims: A list of integer describes how the tensor is partitioned.
  
    Returns:
      The same dequeues with appropriate xla_sharding attribute.
    """
    nest.assert_shallow_structure(dequeues, dims)
    return nest.map_structure_up_to(
        dequeues, _tag_sharding_attribute_for_dequeued_tensor, dequeues, dims)
function_tokens: def tag_sharding_attribute_for_dequeued_tensors ( dequeues , dims ) : nest . assert_shallow_structure ( dequeues , dims ) return nest . map_structure_up_to ( dequeues , _tag_sharding_attribute_for_dequeued_tensor , dequeues , dims )