class_docstr: Manages a TensorFlow model for training.
class_docstr_tok: ["Manages","a","TensorFlow","model","for","training","."]
code: def setup(self):
      import tensorflow as tf
  
      tf.config.threading.set_inter_op_parallelism_threads(
          self.inter_op_parallelism
      )
      tf.config.threading.set_intra_op_parallelism_threads(
          self.intra_op_parallelism
      )
      os.environ["KMP_BLOCKING_TIME"] = self.config.get(
          "KMP_BLOCKING_TIME", os.environ.get("KMP_BLOCKING_TIME", "0")
      )
  
imports: import logging
  import json
  import os
  import numpy
  import ray
  from contextlib import closing
  import logging
  import socket
  from zoo.orca.data.utils import ray_partitions_get_data_label
label_desc: Use image node_id 2 for calling the TFRunner obj's underlying member method code with example usage: obj.setup() without return types