| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_config_test.py | 24 from tensorflow.python.estimator import run_config as run_config_lib
38 run_config = tpu_config_lib.RunConfig()
39 self.assertIsNone(run_config.session_config)
43 run_config = tpu_config_lib.RunConfig(session_config=session_config)
44 self.assertEqual(session_config, run_config.session_config)
58 run_config = tpu_config_lib.RunConfig()
59 self.assertIsNone(run_config.session_config)
74 run_config = tpu_config_lib.RunConfig(session_config=session_config)
75 self.assertEqual(session_config, run_config.session_config)
97 run_config = tpu_config_lib.RunConfig()
[all …]
|
| D | tpu_context.py | 505 run_config = self._config
507 if run_config.tpu_config.tpu_job_name:
508 return run_config.tpu_config.tpu_job_name
514 run_config.evaluation_master
515 if mode == model_fn_lib.ModeKeys.EVAL else run_config.master)
516 cluster_def = (run_config.session_config.cluster_def
517 if run_config.session_config else None)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | learn_runner_test.py | 29 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib
31 from tensorflow.python.estimator import run_config as core_run_config_lib
109 def build_experiment_fn_for_output_dir(run_config=None): argument
112 return TestExperiment(config=run_config, model_dir=output_dir)
116 def build_experiment_for_run_config(run_config, hparams): argument
119 return TestExperiment(config=run_config)
168 run_config=run_config_lib.RunConfig())
210 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
214 run_config=run_config,
218 run_config = run_config_lib.RunConfig(model_dir=_MODIR_DIR)
[all …]
|
| D | learn_runner.py | 26 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib
71 def wrapped_experiment_fn(run_config, hparams): argument
73 if not isinstance(run_config, run_config_lib.RunConfig):
76 if not run_config.model_dir:
84 expected_uid = run_config.uid()
85 experiment = experiment_fn(run_config, hparams)
109 def run(experiment_fn, output_dir=None, schedule=None, run_config=None, argument
195 if output_dir is not None and run_config is not None:
198 if output_dir is None and run_config is None:
206 if run_config is not None:
[all …]
|
| D | experiment.py | 38 from tensorflow.contrib.learn.python.learn.estimators import run_config
349 if isinstance(config, run_config.RunConfig):
351 config.environment == run_config.Environment.LOCAL):
355 if (config.environment != run_config.Environment.LOCAL and
356 config.environment != run_config.Environment.GOOGLE and
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | estimator_training.py | 216 run_config = estimator.config
217 if not run_config._distribute_coordinator_mode: # pylint: disable=protected-access
239 if (run_config._distribute_coordinator_mode == # pylint: disable=protected-access
274 if (run_config._distribute_coordinator_mode ==
276 cluster_spec = run_config.cluster_spec
285 run_config.train_distribute,
287 run_config.eval_distribute,
288 mode=run_config._distribute_coordinator_mode,
290 session_config=run_config.session_config)
298 run_config = estimator._config
[all …]
|
| /external/tensorflow/tensorflow/python/estimator/ |
| D | run_config.py | 26 from tensorflow_estimator.python.estimator import run_config
30 run_config.__all__ = [s for s in dir(run_config) if not s.startswith('__')]
32 from tensorflow_estimator.python.estimator.run_config import *
|
| D | BUILD | 30 ":run_config",
79 ":run_config",
86 name = "run_config",
87 srcs = ["run_config.py"],
212 ":run_config",
326 ":run_config",
406 ":run_config",
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_test.py | 37 from tensorflow.contrib.learn.python.learn.estimators import run_config
210 config=run_config.RunConfig(tf_random_seed=1))
273 config = run_config.RunConfig(tf_random_seed=1)
301 config=run_config.RunConfig(tf_random_seed=1))
325 config=run_config.RunConfig(tf_random_seed=1))
340 config=run_config.RunConfig(tf_random_seed=1))
393 config=run_config.RunConfig(tf_random_seed=1))
438 config=run_config.RunConfig(tf_random_seed=1))
485 config=run_config.RunConfig(tf_random_seed=1))
511 config=run_config.RunConfig(tf_random_seed=1))
[all …]
|
| D | nonlinear_test.py | 26 from tensorflow.contrib.learn.python.learn.estimators import run_config
45 config=run_config.RunConfig(tf_random_seed=1))
70 config=run_config.RunConfig(tf_random_seed=1))
102 config=run_config.RunConfig(tf_random_seed=1))
114 config=run_config.RunConfig(tf_random_seed=1))
126 config=run_config.RunConfig(tf_random_seed=1))
|
| D | debug_test.py | 35 from tensorflow.contrib.learn.python.learn.estimators import run_config
223 config=run_config.RunConfig(tf_random_seed=1))
241 config=run_config.RunConfig(tf_random_seed=1))
252 config=run_config.RunConfig(tf_random_seed=1))
425 config=run_config.RunConfig(tf_random_seed=1))
484 config=run_config.RunConfig(tf_random_seed=1))
548 config=run_config.RunConfig(tf_random_seed=1))
558 config=run_config.RunConfig(tf_random_seed=1))
581 config=run_config.RunConfig(tf_random_seed=1))
629 config=run_config.RunConfig(tf_random_seed=1))
[all …]
|
| D | dnn_linear_combined_test.py | 35 from tensorflow.contrib.learn.python.learn.estimators import run_config
334 config = run_config.RunConfig(tf_random_seed=1)
497 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
502 config = run_config.RunConfig()
611 config=run_config.RunConfig(tf_random_seed=1))
648 config=run_config.RunConfig(tf_random_seed=1))
684 config=run_config.RunConfig(tf_random_seed=1))
1217 config=run_config.RunConfig(tf_random_seed=1))
1237 config=run_config.RunConfig(tf_random_seed=1))
1256 config=run_config.RunConfig(tf_random_seed=1))
[all …]
|
| D | __init__.py | 334 from tensorflow.contrib.learn.python.learn.estimators.run_config import ClusterConfig
335 from tensorflow.contrib.learn.python.learn.estimators.run_config import Environment
336 from tensorflow.contrib.learn.python.learn.estimators.run_config import RunConfig
337 from tensorflow.contrib.learn.python.learn.estimators.run_config import TaskType
|
| D | run_config_test.py | 24 from tensorflow.contrib.learn.python.learn.estimators import run_config as run_config_lib
26 from tensorflow.python.estimator import run_config as core_run_config
237 run_config = run_config_lib.RunConfig()
238 self.assertEqual(TEST_DIR, run_config.model_dir)
243 run_config = run_config_lib.RunConfig(model_dir=TEST_DIR)
244 self.assertEqual(TEST_DIR, run_config.model_dir)
|
| D | linear_test.py | 35 from tensorflow.contrib.learn.python.learn.estimators import run_config
457 feature_columns=[age], config=run_config.RunConfig(tf_random_seed=1))
490 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
495 config = run_config.RunConfig()
576 config=run_config.RunConfig(tf_random_seed=3))
1006 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1011 config = run_config.RunConfig()
1107 config=run_config.RunConfig(tf_random_seed=1))
1140 config=run_config.RunConfig(tf_random_seed=1))
1159 config=run_config.RunConfig(tf_random_seed=1))
[all …]
|
| D | stability_test.py | 28 from tensorflow.contrib.learn.python.learn.estimators import run_config
84 config = run_config.RunConfig(tf_random_seed=my_seed)
124 config = run_config.RunConfig(tf_random_seed=my_seed)
|
| D | estimator_test.py | 47 from tensorflow.contrib.learn.python.learn.estimators import run_config
334 expected_config = run_config.RunConfig()
360 expected_config = run_config.RunConfig()
561 config = run_config.RunConfig(tf_random_seed=test_random_seed)
569 config = run_config.RunConfig(model_dir='test_dir')
575 config = run_config.RunConfig(model_dir='test_dir')
587 config = run_config.RunConfig()
1331 tf_config = {'cluster': {run_config.TaskType.PS: ['fake_ps_0']}}
1335 config = run_config.RunConfig()
1349 estimator._get_replica_device_setter(run_config.RunConfig())):
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | estimator_test.py | 26 from tensorflow.contrib.learn.python.learn.estimators import run_config
149 config = run_config.RunConfig()
168 config = run_config.RunConfig()
190 config = run_config.RunConfig()
215 config = run_config.RunConfig()
235 config = run_config.RunConfig()
255 config = run_config.RunConfig()
283 config = run_config.RunConfig()
303 config = run_config.RunConfig()
326 config = run_config.RunConfig()
[all …]
|
| D | dnn_tree_combined_estimator_test.py | 27 from tensorflow.contrib.learn.python.learn.estimators import run_config
93 config = run_config.RunConfig()
116 config = run_config.RunConfig()
150 config = run_config.RunConfig()
181 config = run_config.RunConfig()
210 config = run_config.RunConfig()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm_test.py | 25 from tensorflow.contrib.learn.python.learn.estimators import run_config
91 config=run_config.RunConfig(tf_random_seed=2))
101 config=run_config.RunConfig(tf_random_seed=2))
110 config=run_config.RunConfig(tf_random_seed=2))
123 config=run_config.RunConfig(tf_random_seed=2))
154 config=run_config.RunConfig(tf_random_seed=2))
188 config=run_config.RunConfig(tf_random_seed=2))
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | latent_gan_estimator_test.py | 27 from tensorflow.python.estimator import run_config as run_config unknown
92 config = run_config.RunConfig(model_dir=tmp_dir,
|
| /external/tensorflow/tensorflow/contrib/tpu/ |
| D | README.md | 32 run_config = tpu_config.RunConfig(
39 config=run_config,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/examples/ |
| D | mnist.py | 84 run_config = tf.contrib.learn.RunConfig(save_checkpoints_secs=300)
97 config=run_config)
|
| D | binary_mnist.py | 86 run_config = tf.contrib.learn.RunConfig(save_checkpoints_secs=300)
95 config=run_config)
|
| D | boston_combined.py | 60 run_config = tf.contrib.learn.RunConfig(save_summary_steps=1)
72 config=run_config,
|