| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator.py | 117 session_config=None, argument
141 self._session_config = session_config
245 session_config = copy.deepcopy(config)
246 session_config.MergeFrom(self._session_config)
248 session_config = self._session_config
255 config=session_config,
263 config=session_config,
267 def session_config(self): member in _WorkerContext
331 session_config, argument
336 session_config = copy.deepcopy(session_config)
[all …]
|
| D | distribute_coordinator_test.py | 106 session_config=None, argument
126 if session_config:
129 session_config.intra_op_parallelism_threads += 1
131 session_config.device_filters.extend(
134 session_config.inter_op_parallelism_threads += 1
135 session_config.device_filters.append("/job:somejob")
391 session_config=None, argument
796 session_config=None, argument
799 del cluster_spec, task_type, task_id, session_config, environment
826 session_config = kwargs.get("session_config", None)
[all …]
|
| D | estimator_training.py | 290 session_config=run_config.session_config)
341 session_config=run_config.session_config)
389 session_config=run_config.session_config)
|
| D | collective_all_reduce_strategy.py | 354 session_config=None, argument
383 if session_config:
384 session_config.CopyFrom(self._update_config_proto(session_config))
|
| D | parameter_server_strategy.py | 460 session_config=None, argument
490 if session_config:
491 session_config.CopyFrom(self._update_config_proto(session_config))
|
| D | tpu_strategy.py | 595 session_config=None, argument
600 if session_config:
601 session_config.CopyFrom(self._update_config_proto(session_config))
|
| D | distribute_lib.py | 568 session_config=None, argument
584 session_config, cluster_spec, task_type, task_id)
1332 session_config=None, argument
1337 del session_config, cluster_spec, task_type, task_id
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_config_test.py | 39 self.assertIsNone(run_config.session_config)
42 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
43 run_config = tpu_config_lib.RunConfig(session_config=session_config)
44 self.assertEqual(session_config, run_config.session_config)
59 self.assertIsNone(run_config.session_config)
73 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
74 run_config = tpu_config_lib.RunConfig(session_config=session_config)
75 self.assertEqual(session_config, run_config.session_config)
|
| D | tpu_strategy_util.py | 97 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
99 with session_lib.Session(config=session_config, target=master) as sess:
|
| D | tpu_context.py | 265 if (self._config.session_config and
266 self._config.session_config.cluster_def.job):
267 cluster_def = self._config.session_config.cluster_def
516 cluster_def = (run_config.session_config.cluster_def
517 if run_config.session_config else None)
|
| D | tpu_system_metadata.py | 167 session_config = get_session_config_with_timeout(
170 master_address, config=session_config) as sess:
|
| D | datasets_test.py | 56 session_config = config_pb2.ConfigProto(cluster_def=self._cluster_def)
58 self._sess = session.Session(self._worker.target, config=session_config)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib.i | 75 tensorflow::ConfigProto session_config; in ListDevices() local
76 return ListDevicesWithSessionConfig(session_config, out_status); in ListDevices()
101 def list_devices(session_config=None):
105 if session_config:
106 return ListDevicesWithSessionConfig(session_config.SerializeToString(),
|
| D | device_lib.py | 25 def list_local_devices(session_config=None): argument
41 for s in pywrap_tensorflow.list_devices(session_config=session_config)
|
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 37 session_config=None): argument
51 session_config=session_config)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | evaluation.py | 161 session_config=None, argument
217 config=session_config)
236 session_config=None, argument
308 config=session_config,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | model.py | 92 if config.session_config is not None:
93 session_config = config.session_config
94 session_config.allow_soft_placement = True
96 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
97 config = config.replace(session_config=session_config)
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | learning.py | 84 session_config=None, argument
188 session_config=session_config,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 298 def _CreateConverter(self, gdef, session_config, conversion_params): argument
304 session_config=session_config,
327 session_config = self._GetConfigProto(run_params, GraphState.CALIBRATE)
328 logging.info("Running calibration graph, config:\n%s", str(session_config))
330 converter = self._CreateConverter(gdef, session_config, conversion_params)
344 session_config = self._GetConfigProto(run_params, GraphState.INFERENCE)
346 str(session_config))
347 converter = self._CreateConverter(gdef, session_config, conversion_params)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 109 session_config=None): argument
155 self._session_config = session_config or config_pb2.ConfigProto()
563 session_config=None, argument
631 session_config=session_config)
795 session_config=None): argument
864 session_config=session_config,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | run_config.py | 246 session_config=None): argument
321 self._session_config = session_config
395 def session_config(self): member in RunConfig
|
| D | run_config_test.py | 299 session_config=config_pb2.ConfigProto(allow_soft_placement=True),
307 config.session_config)
316 session_config=config_pb2.ConfigProto(allow_soft_placement=False),
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | dist_session_debug_grpc_test.py | 86 cls.session_config = config_pb2.ConfigProto(
132 config=self.session_config, graph=graph,
199 config=self.session_config, graph=graph,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.estimator.-run-config.pbtxt | 78 name: "session_config"
99 …ry_steps\', \'save_checkpoints_steps\', \'save_checkpoints_secs\', \'session_config\', \'keep_chec…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.-run-config.pbtxt | 78 name: "session_config"
99 …ry_steps\', \'save_checkpoints_steps\', \'save_checkpoints_secs\', \'session_config\', \'keep_chec…
|