code: def tree_map(
      fn: Callable[[T], Any], tree: tuple, leaf_type: Type[T]
  ) -> tuple:
      ...
  
file_path: transformers/src/transformers/models/esm/openfold_utils/tensor_utils.py
imports: from functools import partial
  from typing import Any, Callable, Dict, List, Type, TypeVar, Union, overload
  import torch
  import torch.nn
  import torch.types
label_desc: Use image node_id 13 for calling a global function with example usage: tree_map(fn, tree, leaf_type) without return types