class_docstr: Tests for ops in byteps.keras.
code: def __init__(self, *args, **kwargs):
      super(TfKerasTests, self).__init__(*args, **kwargs)
      warnings.simplefilter("module")
      bps.init()
  
      self.config = tf.ConfigProto()
      self.config.gpu_options.allow_growth = True
      self.config.gpu_options.visible_device_list = str(
          bps.local_rank()
      )
  
file_path: byteps/tests/test_tensorflow_keras.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import numpy
  import warnings
  from distutils.version import LooseVersion
  from tensorflow import keras
  from tensorflow.python.keras import backend
  import byteps.tensorflow.keras
label_desc: Use image node_id 1 to create a new TfKerasTests object  with example: obj = TfKerasTests()