class_docstr: Manages a TensorFlow estimator for training.
class_docstr_tok: ["Manages","a","TensorFlow","estimator","for","training","."]
code: def setup(self) -> None:
      import tensorflow.compat.v1 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")
      )
  
file_path: BigDL/python/orca/src/bigdl/orca/learn/tf/tf_runner.py
imports: from contextlib import closing
  import json
  import logging
  import os
  import socket
  import ray
  from bigdl.dllib.utils.log4Error import invalidInputError
  from typing import TYPE_CHECKING, Any
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