| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config_test.py | 39 with patch.dict("os.environ", {"TF_CONFIG": tf_config_str}):
61 tf_config = {
71 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):
78 self.assertEqual(config.cluster_spec.as_dict(), tf_config["cluster"])
88 tf_config = {
95 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):
109 tf_config = {"cluster": {run_config_lib.TaskType.WORKER: ["host1:1"]}}
110 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)})
[all...] |
| linear_test.py | 488 tf_config = {
494 {'TF_CONFIG': json.dumps(tf_config)}):
[all...] |
| run_config.py | 86 `TF_CONFIG` environment variable, if the pertinent information is
87 present. The `TF_CONFIG` environment variable is a JSON object with
98 the task types in `cluster`. When `TF_CONFIG` contains said information, the
101 * `task_type` is set to `TF_CONFIG['task']['type']`. Defaults to `None`.
102 * `task_id` is set to `TF_CONFIG['task']['index']`. Defaults to 0.
103 * `cluster_spec` is parsed from `TF_CONFIG['cluster']`. Defaults to {}.
117 os.environ['TF_CONFIG'] = json.dumps(
134 # If not explicitly specified in the constructor and the TF_CONFIG
135 # environment variable is present, load cluster_spec from TF_CONFIG.
136 config = json.loads(os.environ.get('TF_CONFIG') or '{}'
375 def tf_config(self): member in class:RunConfig
[all...] |
| estimator_test.py | [all...] |
| dnn_linear_combined_test.py | 495 tf_config = {
501 {'TF_CONFIG': json.dumps(tf_config)}):
[all...] |
| dnn_test.py | [all...] |
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu_config_test.py | 29 def _set_tf_config_env_variable(tf_config):
31 'TF_CONFIG': json.dumps(tf_config)
47 tf_config = {
57 with _set_tf_config_env_variable(tf_config):
62 tf_config = {
72 with _set_tf_config_env_variable(tf_config):
113 tf_config = {
117 with _set_tf_config_env_variable(tf_config):
123 tf_config =
[all...] |
| tpu_config.py | 230 # TF_CONFIG. In that case, we respect the configuration in TF_CONFIG.
286 """Extracts the TPU job name from TF_CONFIG env variable."""
287 # TODO(xiejw): Extends this to support both TF_CONFIG env variable and cluster
289 tf_config = json.loads(os.environ.get(_TF_CONFIG_ENV, '{}'))
290 tpu_job_name = tf_config.get(_SERVICE_KEY, {}).get(_TPU_WORKER_JOB_NAME)
292 logging.info('Load TPU job name from TF_CONFIG: %s', tpu_job_name)
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| tfconfig_cluster_resolver.py | 15 """Implementation of Cluster Resolvers for TF_CONFIG Environment Variables."""
29 _TF_CONFIG_ENV = 'TF_CONFIG'
47 tf_config = _load_tf_config()
48 return tf_config[key] if key in tf_config else default
53 """Implementation of a ClusterResolver which reads the TF_CONFIG EnvVar."""
64 TF_CONFIG environment variable.
66 TF_CONFIG environment variable.
125 """Returns a ClusterSpec based on the TF_CONFIG environment variable.
128 A ClusterSpec with information from the TF_CONFIG environment variable
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| learn_runner_test.py | 145 # Ensure the TF_CONFIG environment variable is unset for all tests.
146 os.environ.pop("TF_CONFIG", None)
206 # Ensure the TF_CONFIG environment variable is unset for all tests.
207 os.environ.pop("TF_CONFIG", None)
329 # Ensure the TF_CONFIG environment variable is unset for all tests.
330 os.environ.pop("TF_CONFIG", None)
333 os.environ["TF_CONFIG"] = json.dumps({
339 # RunConfig constructor will set job_name from TF_CONFIG.
348 tf_config = {
354 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)})
[all...] |
| experiment_test.py | 266 tf_config = {'task': {'index': task_id}}
268 {'TF_CONFIG': json.dumps(tf_config)}):
283 tf_config = {
292 {'TF_CONFIG': json.dumps(tf_config)}):
338 tf_config = {'cluster': self._cluster_spec()}
340 {'TF_CONFIG': json.dumps(tf_config)}):
352 tf_config =
[all...] |
| experiment.py | [all...] |
| /external/tensorflow/tensorflow/python/distribute/ |
| distribute_coordinator.py | 555 This method parses configurations from "TF_CONFIG" environment variable and
556 starts a TensorFlow server. The "TF_CONFIG" is typically a json string and
560 TF_CONFIG='{
568 This "TF_CONFIG" specifies there are 3 workers and 2 ps tasks in the cluster
585 ValueError: if the "TF_CONFIG" environment is not complete.
587 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))
588 if "cluster" not in tf_config:
589 raise ValueError("\"cluster\" is not found in TF_CONFIG.")
590 cluster_spec = multi_worker_util.normalize_cluster_spec(tf_config["cluster"]
[all...] |
| estimator_training.py | 127 def init_run_config(config, tf_config):
144 cluster_spec = server_lib.ClusterSpec(tf_config.get('cluster', {}))
152 raise ValueError('Cannot set both "cluster_spec" of TF_CONFIG and '
162 if (not tf_config or 'master' in cluster_spec.jobs or
165 config._init_distributed_setting_from_environment_var(tf_config)
171 assert tf_config
279 # The cluster_spec comes from TF_CONFIG environment variable if it is
|
| distribute_coordinator_test.py | 761 tf_config = {"cluster": cluster_spec, "environment": "google"}
780 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
789 tf_config = {"cluster": cluster_spec, "rpc_layer": "cake"}
805 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
846 tf_config = {"cluster": cluster_spec}
850 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| multi_worker_test_base.py | 316 """A class that allows per-thread TF_CONFIG."""
326 if key == 'TF_CONFIG':
334 if key == 'TF_CONFIG':
342 if key == 'TF_CONFIG':
392 def _task_thread(self, task_fn, tf_config, *args, **kwargs):
394 os.environ['TF_CONFIG'] = json.dumps(tf_config)
400 tf_config = {
408 tf_config = {
413 args=(task_fn, tf_config) + args
[all...] |
| mirrored_strategy_multigpu_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| wals_test.py | 402 tf_config = {
413 {'TF_CONFIG': json.dumps(tf_config)}):
|
| /external/tensorflow/tensorflow/python/keras/ |
| backend.py | [all...] |
