Home
last modified time | relevance | path

Searched refs:run_config_lib (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config_test.py24 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib unknown
40 return run_config_lib.RunConfig(
47 config = run_config_lib.RunConfig()
51 config = run_config_lib.RunConfig()
63 run_config_lib.TaskType.PS: ["host1:1", "host2:2"],
64 run_config_lib.TaskType.WORKER: ["host3:3", "host4:4", "host5:5"]
67 "type": run_config_lib.TaskType.WORKER,
72 config = run_config_lib.RunConfig()
79 self.assertEqual(config.task_type, run_config_lib.TaskType.WORKER)
85 run_config_lib.TaskType.PS: ["localhost:9990"],
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dlearn_runner_test.py29 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib unknown
131 run_config_lib.TaskType.PS: ["localhost:1234", "localhost:1235"],
132 run_config_lib.TaskType.WORKER: ["localhost:1236", "localhost:1237"],
133 run_config_lib.TaskType.MASTER: ["localhost:1238"],
168 run_config=run_config_lib.RunConfig())
210 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
218 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
227 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
235 run_config = run_config_lib.RunConfig(model_dir="")
242 run_config = run_config_lib.RunConfig()
[all …]
Dlearn_runner.py26 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib unknown
73 if not isinstance(run_config, run_config_lib.RunConfig):
288 if config.task_type == run_config_lib.TaskType.MASTER:
292 elif config.task_type == run_config_lib.TaskType.PS:
294 elif config.task_type == run_config_lib.TaskType.WORKER:
Dexperiment_test.py32 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib unknown
196 run_config_lib.TaskType.PS: ['host1:2222', 'host2:2222'],
197 run_config_lib.TaskType.WORKER:
285 'environment': run_config_lib.Environment.CLOUD,
287 'type': run_config_lib.TaskType.WORKER,
293 config = run_config_lib.RunConfig(
317 job_name=run_config_lib.TaskType.WORKER,
325 config = run_config_lib.RunConfig(master='host4:2222')
341 config = run_config_lib.RunConfig(master='')
354 'environment': run_config_lib.Environment.CLOUD,
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtpu_config_test.py24 from tensorflow.python.estimator import run_config as run_config_lib unknown
49 run_config_lib.TaskType.CHIEF: ['host3:3'],
50 run_config_lib.TaskType.WORKER: ['host3:4']
53 'type': run_config_lib.TaskType.CHIEF,
64 run_config_lib.TaskType.CHIEF: ['host3:3'],
65 run_config_lib.TaskType.WORKER: ['host3:4']
68 'type': run_config_lib.TaskType.CHIEF,
134 run_config_lib.TaskType.CHIEF: ['host0:0'],
137 'type': run_config_lib.TaskType.EVALUATOR,
Dtpu_config.py27 from tensorflow.python.estimator import run_config as run_config_lib unknown
32 _TF_CONFIG_ENV = run_config_lib._TF_CONFIG_ENV
33 _SERVICE_KEY = run_config_lib._SERVICE_KEY
188 class RunConfig(run_config_lib.RunConfig):
228 self.task_type != run_config_lib.TaskType.EVALUATOR):
/external/tensorflow/tensorflow/contrib/distribute/python/
Destimator_training_test.py43 from tensorflow.python.estimator import run_config as run_config_lib unknown
151 config=run_config_lib.RunConfig(
481 run_config_lib.RunConfig(
491 run_config_lib.RunConfig()))
499 run_config_lib.RunConfig()))
505 config_with_train_distribute = run_config_lib.RunConfig(
509 config_with_eval_distribute = run_config_lib.RunConfig(
523 config = run_config_lib.RunConfig(
531 run_config_lib.RunConfig(
537 run_config_lib.RunConfig(
[all …]
Dkeras_test.py32 from tensorflow.python.estimator import run_config as run_config_lib unknown
331 self._config = run_config_lib.RunConfig(
353 config = run_config_lib.RunConfig(tf_random_seed=_RANDOM_SEED,
383 config = run_config_lib.RunConfig(tf_random_seed=_RANDOM_SEED,
438 config = run_config_lib.RunConfig(
462 config = run_config_lib.RunConfig(tf_random_seed=_RANDOM_SEED,
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py29 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib unknown
404 run_config_lib.TaskType.PS: ['host1:1', 'host2:2'],
405 run_config_lib.TaskType.WORKER: ['host3:3', 'host4:4']
408 'type': run_config_lib.TaskType.WORKER,