code: def __init__(self, method_name="runTest"):
      super(WhereOpTest, self).__init__(method_name)
      if config.list_logical_devices("TPU"):
          with self.session() as sess:
              sess.run(tpu.initialize_system())
  
file_path: tensorflow-upstream/tensorflow/compiler/tests/where_op_test.py
imports: from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import config
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import tpu
label_desc: Use image node_id 1 to create a new WhereOpTest object from inherited base classes: xla_test with example: obj = WhereOpTest(method_name)