class_docstr: Tests for ops in horovod.tensorflow.keras.
class_docstr_tok: ["Tests","for","ops","in","horovod.tensorflow.keras","."]
code: def __init__(self, *args, **kwargs):
      super(TfKerasTests, self).__init__(*args, **kwargs)
      warnings.simplefilter("module")
      hvd.init()
  
      self.config = tf.compat.v1.ConfigProto()
      self.config.gpu_options.allow_growth = True
      self.config.gpu_options.visible_device_list = str(
          hvd.local_rank()
      )
  
imports: import math
  import numpy
  import os
  import pytest
  import sys
  import tensorflow
  import warnings
  from packaging import version
  from tensorflow import keras
  from horovod.common.util import is_version_greater_equal_than
  import horovod.tensorflow.keras
  from common import temppath
label_desc: Use image node_id 1 to create a new TfKerasTests object from inherited base classes: tf.test with example: obj = TfKerasTests()