code: def init():
      gpu_thread_count = 2
      os.environ["TF_GPU_THREAD_MODE"] = "gpu_private"
      os.environ["TF_GPU_THREAD_COUNT"] = str(gpu_thread_count)
      os.environ["TF_USE_CUDNN_BATCHNORM_SPATIAL_PERSISTENT"] = "1"
      os.environ["TF_ENABLE_WINOGRAD_NONFUSED"] = "1"
      hvd.init()
      if hvd.rank() == 0:
          print("PY", sys.version)
          print("TF", tf.version.VERSION)
  
file_path: DALI/docs/examples/use_cases/tensorflow/resnet-n/nvutils/__init__.py
imports: from .runner import train
  from .runner_ctl import train_ctl
  from .runner import predict
  from .runner_ctl import predict_ctl
  from .cmdline import parse_cmdline
  import os, sys, random
  import tensorflow
  import horovod.tensorflow.keras
label_desc: Use image node_id 1 for calling a global function with example usage: init() without return types