Home
last modified time | relevance | path

Searched refs:worker_config (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py44 worker_config = config_pb2.ConfigProto()
45 worker_config.device_count["CPU"] = 2
47 1, 1, worker_config=worker_config)
100 worker_config = config_pb2.ConfigProto()
101 worker_config.device_count["CPU"] = 2
103 1, 1, worker_config=worker_config)
Diterator_test.py598 worker_config = config_pb2.ConfigProto()
599 worker_config.device_count["CPU"] = 3
621 with self.session(config=worker_config) as sess:
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py57 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
58 with self.test_session(config=worker_config):
82 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
83 with self.test_session(config=worker_config):
106 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
107 with self.test_session(config=worker_config):
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
132 with self.test_session(config=worker_config):
156 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
157 with self.test_session(config=worker_config):
[all …]
Dprefetch_to_device_test.py53 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
54 with self.test_session(config=worker_config):
103 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
104 with self.test_session(config=worker_config):
131 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
132 with self.test_session(config=worker_config):
179 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
180 with self.test_session(config=worker_config):
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py74 worker_config=None, argument
100 config=worker_config,
118 config=worker_config,
127 config=worker_config,
140 worker_config = config_pb2.ConfigProto()
141 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
147 worker_config.experimental.collective_group_leader = (
150 worker_config.experimental.collective_group_leader = (
170 worker_config=worker_config,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.test.pbtxt25 …argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], vara…
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py408 worker_config = config_pb2.ConfigProto()
409 worker_config.device_count["CPU"] = 2
411 1, 1, worker_config=worker_config)
435 worker_config = config_pb2.ConfigProto()
436 worker_config.device_count["CPU"] = 2
453 with self.test_session(config=worker_config) as sess:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.test.pbtxt37 …argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], vara…
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2638 worker_config=None, argument
2693 config=worker_config,