Searched refs:tf_config (Results 1 – 19 of 19) sorted by relevance
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)}):115 tf_config = {122 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):127 tf_config = {[all …]
488 tf_config = {494 {'TF_CONFIG': json.dumps(tf_config)}):1004 tf_config = {1010 {'TF_CONFIG': json.dumps(tf_config)}):1459 tf_config = {1465 {'TF_CONFIG': json.dumps(tf_config)}):1633 tf_config = {1639 {'TF_CONFIG': json.dumps(tf_config)}):
1331 tf_config = {'cluster': {run_config.TaskType.PS: ['fake_ps_0']}}1333 'TF_CONFIG': json.dumps(tf_config)1360 tf_config = {'cluster': {run_config.TaskType.PS: ['fake_ps_0']}}1362 'TF_CONFIG': json.dumps(tf_config)1385 tf_config = {1395 'TF_CONFIG': json.dumps(tf_config)
375 def tf_config(self): member in RunConfig
943 tf_config = {949 {'TF_CONFIG': json.dumps(tf_config)}):1509 tf_config = {1515 {'TF_CONFIG': json.dumps(tf_config)}):
495 tf_config = {501 {'TF_CONFIG': json.dumps(tf_config)}):1640 tf_config = {1646 {'TF_CONFIG': json.dumps(tf_config)}):
29 def _set_tf_config_env_variable(tf_config): argument31 '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 = {126 with _set_tf_config_env_variable(tf_config):[all …]
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)
47 tf_config = _load_tf_config()48 return tf_config[key] if key in tf_config else default130 tf_config = _load_tf_config()131 if 'cluster' not in tf_config:133 return ClusterSpec(tf_config['cluster'])
348 tf_config = {354 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):363 tf_config = {369 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):383 tf_config = {"cluster": build_non_distributed_cluster_spec()}384 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):393 tf_config = {399 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):409 tf_config = {"cluster": build_distributed_cluster_spec()}410 with patch.dict("os.environ", {"TF_CONFIG": json.dumps(tf_config)}):
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 = {361 {'TF_CONFIG': json.dumps(tf_config)}), self.assertRaises(ValueError):833 tf_config = {841 {'TF_CONFIG': json.dumps(tf_config)}):
857 config=config.tf_config,
587 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))588 if "cluster" not in tf_config:590 cluster_spec = multi_worker_util.normalize_cluster_spec(tf_config["cluster"])591 if "task" not in tf_config:593 task_env = tf_config["task"]600 rpc_layer = tf_config.get("rpc_layer", "grpc")749 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))751 cluster_spec = tf_config.get("cluster", {})752 task_env = tf_config.get("task", {})761 rpc_layer = tf_config.get("rpc_layer", rpc_layer)[all …]
127 def init_run_config(config, tf_config): argument144 cluster_spec = server_lib.ClusterSpec(tf_config.get('cluster', {}))162 if (not tf_config or 'master' in cluster_spec.jobs or165 config._init_distributed_setting_from_environment_var(tf_config)171 assert tf_config
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(865 tf_config = {"cluster": cluster_spec}871 {"TF_CONFIG": json.dumps(tf_config)}):885 tf_config = {"cluster": cluster_spec}888 {"TF_CONFIG": json.dumps(tf_config)}):[all …]
392 def _task_thread(self, task_fn, tf_config, *args, **kwargs): argument394 os.environ['TF_CONFIG'] = json.dumps(tf_config)400 tf_config = {408 tf_config = {413 args=(task_fn, tf_config) + args,
1553 tf_config = {"cluster": cluster_spec}1555 {"TF_CONFIG": json.dumps(tf_config)}):1562 tf_config = {"cluster": self._cluster_spec}1564 {"TF_CONFIG": json.dumps(tf_config)}):
402 tf_config = {413 {'TF_CONFIG': json.dumps(tf_config)}):
5323 tf_config = json.loads(os.environ.get('TF_CONFIG', '{}'))5324 cluster_spec = server_lib.ClusterSpec(tf_config.get('cluster', {}))5325 return tf_config and 'master' not in cluster_spec.jobs