| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| D | sagemaker_cluster_resolver.py | 66 tf_config = {
75 return tf_config
79 tf_config = _load_tf_config(port)
80 return tf_config[key] if key in tf_config else default
166 tf_config = _load_tf_config(self._port)
167 if 'cluster' not in tf_config:
169 return ClusterSpec(tf_config['cluster'])
|
| D | tfconfig_cluster_resolver.py | 47 tf_config = _load_tf_config()
48 return tf_config[key] if key in tf_config else default
159 tf_config = _load_tf_config()
160 if 'cluster' not in tf_config:
162 return ClusterSpec(tf_config['cluster'])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_combinations.py | 131 tf_config = cluster_resolver.TFConfigClusterResolver()
132 master = tf_config.master()
133 if tf_config.rpc_layer:
135 master = master[len("%s://" % tf_config.rpc_layer):]
137 cluster_spec=tf_config.cluster_spec(),
138 task_type=tf_config.task_type,
139 task_id=tf_config.task_id,
141 environment=tf_config.environment,
143 rpc_layer=tf_config.rpc_layer or "grpc",
|
| D | distribute_coordinator.py | 590 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))
591 if "cluster" not in tf_config:
593 cluster_spec = multi_worker_util.normalize_cluster_spec(tf_config["cluster"])
594 if "task" not in tf_config:
596 task_env = tf_config["task"]
603 rpc_layer = tf_config.get("rpc_layer", "grpc")
752 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))
753 rpc_layer = tf_config.get("rpc_layer", rpc_layer)
754 environment = tf_config.get("environment", None)
757 cluster_spec = tf_config.get("cluster", {})
[all …]
|
| D | multi_worker_continuous_run_test.py | 94 tf_config = json.loads(os.environ['TF_CONFIG'])
95 worker_id = tf_config['task']['index']
127 tf_config = json.loads(os.environ['TF_CONFIG'])
128 worker_id = tf_config['task']['index']
|
| D | distribute_coordinator_test.py | 769 tf_config = {"cluster": cluster_spec, "environment": "google"}
788 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
797 tf_config = {"cluster": cluster_spec, "rpc_layer": "cake"}
813 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
854 tf_config = {"cluster": cluster_spec}
858 {"TF_CONFIG": json.dumps(tf_config)}), test.mock.patch.object(
873 tf_config = {"cluster": cluster_spec}
879 {"TF_CONFIG": json.dumps(tf_config)}):
893 tf_config = {"cluster": cluster_spec}
896 {"TF_CONFIG": json.dumps(tf_config)}):
[all …]
|
| D | estimator_training.py | 127 def init_run_config(config, tf_config): argument
144 cluster_spec = server_lib.ClusterSpec(tf_config.get('cluster', {}))
165 config._init_distributed_setting_from_environment_var(tf_config)
171 assert tf_config
|
| D | multi_worker_test_base.py | 681 def _task_thread(self, task_fn, tf_config, executing_eagerly, *args, argument
684 os.environ['TF_CONFIG'] = json.dumps(tf_config)
715 tf_config = kwargs.pop('tf_config', None)
716 if tf_config is None:
718 tf_config = {
726 tf_config = {
731 args=(task_fn, tf_config, context.executing_eagerly()) + args,
|
| D | collective_all_reduce_strategy_test.py | 44 from tensorflow.python.framework import config as tf_config unknown
536 gpus = tf_config.list_physical_devices('GPU')
537 tf_config.set_logical_device_configuration(gpus[-1], [
545 self.assertLen(tf_config.list_logical_devices('GPU'), len(gpus) + 1)
|
| D | mirrored_strategy_test.py | 1313 tf_config = {"cluster": cluster_spec}
1315 {"TF_CONFIG": json.dumps(tf_config)}):
1325 tf_config = {"cluster": self._cluster_spec}
1327 {"TF_CONFIG": json.dumps(tf_config)}):
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 150 tf_config = json.loads(os.environ.get("TF_CONFIG", "{}"))
151 if tf_config:
153 len(tf_config["cluster"].get("chief", [])) +
154 len(tf_config["cluster"].get("worker", [])))
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 23 from tensorflow.python.framework import config as tf_config unknown
140 if context.executing_eagerly() and tf_config.list_logical_devices('GPU'):
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 31 from tensorflow.python.framework import config as tf_config unknown
434 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):
543 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):
719 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):
925 if context.executing_eagerly() and tf_config.list_logical_devices("GPU"):
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | keras_test.py | 33 from tensorflow.python.framework import config as tf_config unknown
104 if tf_config.list_logical_devices('GPU'):
113 tf_config.list_logical_devices('GPU')) else ['cpu:0']
|
| D | loss_scale_optimizer_test.py | 30 from tensorflow.python.framework import config as tf_config unknown
57 if tf_config.list_logical_devices('GPU'):
|