code: def _add_tpu_host_options(data):
      options = tf.data.Options()
      options.threading.private_threadpool_size = 48
      options.threading.max_intra_op_parallelism = 1
      return data.with_options(options)
  
imports: import collections
  import dataclasses
  import functools
  import itertools
  from typing import Any, Callable, Dict, Iterator, Optional, Sequence, Union
  from absl import logging
  from flax.training import common_utils
  import jax
  import jax.numpy
  import numpy
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 21 for calling a global function with example usage: _add_tpu_host_options(data) and returns: data