| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 345 self._num_hosts = self._tpu_system_metadata.num_hosts
476 config_proto.num_hosts = self._num_hosts
525 num_hosts=self._num_hosts,
796 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
814 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
820 num_hosts=num_hosts,
835 range(num_hosts), table_variables, accumulator_variables)):
842 num_shards=num_hosts,
855 range(num_hosts), table_variables, accumulator_variables)):
860 num_shards=num_hosts,
[all …]
|
| D | tpu_context.py | 92 total_invocation_count = (self._internal_ctx.num_hosts
99 total_invocation_count = self._internal_ctx.num_hosts
117 def num_hosts(self): member in TPUContext
119 return self._internal_ctx.num_hosts
316 self._eval_batch_size, self.num_hosts, self.num_cores, self.config)
356 return self.num_replicas // self.num_hosts
386 def num_hosts(self): member in _InternalTPUContext
388 return metadata.num_hosts
480 return global_batch_size // self.num_hosts
598 num_hosts = self.num_hosts
[all …]
|
| D | tpu_system_metadata.py | 142 num_hosts=len(device_dict),
150 logging.info('*** Num TPU Workers: %d', metadata.num_hosts)
|
| D | _tpu_estimator_embedding.py | 214 num_hosts, num_cores, run_config): argument
218 self._num_hosts = num_hosts
|
| D | tpu_estimator.py | 947 num_hosts): argument
989 for host_id in xrange(num_hosts):
1239 num_hosts = self._ctx.num_hosts
1248 for host_id in range(num_hosts):
1273 num_hosts))
1286 for host_id in range(num_hosts):
|
| D | tensor_tracer.py | 1542 num_hosts=None): argument
1574 self._num_hosts = num_hosts
|
| /external/bcc/examples/networking/distributed_bridge/ |
| D | main.py | 32 num_hosts = 3 variable
42 for i in range(0, num_hosts):
43 print("Launching host %i of %i" % (i + 1, num_hosts))
50 cmd = ["python", "tunnel_mesh.py", str(num_hosts), str(i), str(dhcp), str(gretap)]
72 for i in range(1, num_hosts):
|
| D | tunnel_mesh.py | 17 num_hosts = int(argv[1]) variable
65 for i in range(0, num_hosts):
|
| /external/autotest/scheduler/ |
| D | rdb_integration_tests.py | 541 def min_dut_test_helper(self, num_hosts, suite_settings): argument
555 for i in range (0, num_hosts):
579 num_hosts = 1
584 self.min_dut_test_helper(num_hosts, [suite1, suite2])
589 num_hosts = 4
594 self.min_dut_test_helper(num_hosts, [suite1, suite2])
599 num_hosts = 7
604 self.min_dut_test_helper(num_hosts, [suite1, suite2])
609 num_hosts = 10
614 self.min_dut_test_helper(num_hosts, [suite1, suite2])
[all …]
|
| D | host_scheduler.py | 138 num_hosts = self.suite_host_num.get(parent_id, 0)
139 self.suite_host_num[parent_id] = num_hosts + 1
|
| /external/bcc/examples/networking/tunnel_monitor/ |
| D | main.py | 19 num_hosts = 9 variable
32 for i in range(0, num_hosts):
33 print("Launching host %i of %i" % (i + 1, num_hosts))
41 print("Starting tunnel %i of %i" % (len(self.processes) + 1, num_hosts))
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 40 int32 num_hosts = 4; field
47 // "id % num_hosts". For instance, 13 ids are split across 5 hosts as:
53 // of hosts, each of the first "table_descriptor.vocabulary_size % num_hosts"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 527 def num_hosts(self): member in TPUExtended
529 return self._tpu_metadata.num_hosts
|