Searched refs:sess_config (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 77 sess_config=None): argument
78 sess_config = sess_config or config_pb2.ConfigProto()
96 sess_config = copy.deepcopy(sess_config)
97 sess_config = distribution.update_config_proto(sess_config)
99 return distribution, target, sess_config
121 sess_config=self._sess_config)
125 d, _, sess_config = self._get_test_objects(task_type, task_id, num_gpus)
128 config=sess_config) as sess, \
224 d, _, sess_config = self._get_test_objects(task_type, task_id, num_gpus)
229 config=sess_config) as sess, \
[all …]
|
| D | combinations.py | 578 sess_config = copy.deepcopy(context.context().config)
579 sess_config = strategy.update_config_proto(sess_config)
581 return session.Session(config=sess_config, target=target).as_default()
|
| D | collective_all_reduce_strategy_test.py | 73 sess_config = config_pb2.ConfigProto()
91 sess_config = strategy.update_config_proto(sess_config)
93 return strategy, target, sess_config
|