code: def _run_config(device="cpu", device_id=0):
      return tf.estimator.RunConfig(
          model_dir="/tmp/tensorflow-checkpoints",
          device_fn=lambda op: "/{0}:{1}".format(device, device_id),
      )
  
imports: import nvidia.dali.plugin.tf
  import os
  from nvidia.dali.pipeline import Pipeline
  import nvidia.dali.fn
  import nvidia.dali.types
  from test_utils_tensorflow import num_available_gpus
  from shutil import rmtree
  import tensorflow
  import tensorflow.compat.v1
label_desc: Use image node_id 11 for calling a global function with example usage: _run_config(device, device_id) and returns: tf