def setUp(self):
    self._base_dir = os.path.join(self.get_temp_dir(), 'keras_mirrored_strategy_test')
    gfile.MakeDirs(self._base_dir)
    self._config = run_config_lib.RunConfig(tf_random_seed=_RANDOM_SEED, model_dir=self._base_dir)
    self._dist = mirrored_strategy.MirroredStrategy(devices=['/device:GPU:0', '/device:GPU:1'])
def setUp(self):
    self._base_dir = os.path.join(self.get_temp_dir(), 'keras_mirrored_strategy_test')
    gfile.MakeDirs(self._base_dir)
    self._config = run_config_lib.RunConfig(tf_random_seed=_RANDOM_SEED, model_dir=self._base_dir)
    self._dist = mirrored_strategy.MirroredStrategy(devices=['/device:GPU:0', '/device:GPU:1'])