code: def _track_provenance_set(x, provenance: frozenset):
      return type(x)(track_provenance(part, provenance) for part in x)
  
file_path: pyro/pyro/ops/provenance.py
imports: from functools import partial, singledispatch
  from typing import Tuple
  import torch
  from torch.utils._pytree import tree_flatten, tree_map, tree_unflatten
label_desc: Use image node_id 2 for calling a global function with example usage: _track_provenance_set(x, provenance) and returns: type