Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dprefetch_to_device_test.py56 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
57 with self.test_session(config=worker_config):
81 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
82 with self.test_session(config=worker_config):
105 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
106 with self.test_session(config=worker_config):
132 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
133 with self.test_session(config=worker_config):
181 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
182 with self.test_session(config=worker_config):
Dcopy_to_device_test.py60 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
61 with self.test_session(config=worker_config):
85 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
86 with self.test_session(config=worker_config):
109 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
110 with self.test_session(config=worker_config):
134 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
135 with self.test_session(config=worker_config):
159 worker_config = config_pb2.ConfigProto(device_count={"CPU": 2})
160 with self.test_session(config=worker_config):
[all …]
Dreplicate_test.py339 worker_config = config_pb2.ConfigProto()
340 worker_config.device_count["CPU"] = 2
342 3, 0, worker_config=worker_config)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py47 worker_config = config_pb2.ConfigProto()
48 worker_config.device_count["CPU"] = 2
50 1, 1, worker_config=worker_config)
103 worker_config = config_pb2.ConfigProto()
104 worker_config.device_count["CPU"] = 2
106 1, 1, worker_config=worker_config)
Diterator_test.py601 worker_config = config_pb2.ConfigProto()
602 worker_config.device_count["CPU"] = 3
624 with self.session(config=worker_config) as sess:
/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_test_base.py93 worker_config=None, argument
123 config=worker_config,
141 config=worker_config,
164 worker_config = config_pb2.ConfigProto()
165 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
170 worker_config.inter_op_parallelism_threads = 4
176 worker_config.experimental.collective_group_leader = (
179 worker_config.experimental.collective_group_leader = (
204 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/tools/api/golden/v1/
Dtensorflow.test.pbtxt37 …argspec: "args=[\'num_workers\', \'num_ps\', \'protocol\', \'worker_config\', \'ps_config\'], vara…
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Drecording_tpu_driver.cc562 TpuDriverConfig worker_config; in RegisterRecordingTpuDriver() local
563 worker_config.set_worker(worker); in RegisterRecordingTpuDriver()
565 auto driver_status = TpuDriverRegistry::Open(worker_config); in RegisterRecordingTpuDriver()
Dpod_tpu_driver.cc261 TpuDriverConfig worker_config(config_); in PodTpuDriver() local
262 *(worker_config.mutable_worker()) = absl::StrCat("grpc://", worker); in PodTpuDriver()
264 CreateGrpcTpuDriver(worker_config, creds_).ConsumeValueOrDie(); in PodTpuDriver()
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py409 worker_config = config_pb2.ConfigProto()
410 worker_config.device_count["CPU"] = 2
412 1, 1, worker_config=worker_config)
436 worker_config = config_pb2.ConfigProto()
437 worker_config.device_count["CPU"] = 2
454 with self.test_session(config=worker_config) as sess:
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py3291 worker_config=None, argument
3360 config=worker_config,
/external/tensorflow/
DRELEASE.md442 …arguments. Usages should be updated to `tf.data.experimental.service.WorkerServer(worker_config)`.