class_docstr: Tests for scale & elementwise layers in TF-TRT.
code: def setUp(self):
      super().setUp()
      os.environ[
          "TF_TRT_ALLOW_ENGINE_NATIVE_SEGMENT_EXECUTION"
      ] = "True"
      gpus = config.list_physical_devices("GPU")
  
      logging.info("Found the following GPUs:")
      for gpu in gpus:
          logging.info(f"\t- {gpu}")
          config.set_memory_growth(gpu, True)
  
file_path: tensorflow/tensorflow/python/compiler/tensorrt/test/binary_tensor_weight_broadcast_test.py
imports: import os
  import numpy
  from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import gen_array_ops
  from tensorflow.python.platform import test
  from tensorflow.python.platform import tf_logging
label_desc: Use image node_id 5 for calling the BinaryTensorWeightBroadcastTest obj's underlying member method code with example usage: obj.setUp() without return types