Home
last modified time | relevance | path

Searched refs:num_hosts (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py646 if device_config.num_cores % device_config.num_hosts:
649 device_config.num_hosts))
650 self._num_hosts = device_config.num_hosts
665 self._num_hosts = tpu_system_metadata.num_hosts
824 config_proto.num_hosts = self._num_hosts
897 num_hosts=self._num_hosts,
1224 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
1242 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
1248 num_hosts=num_hosts,
1264 range(num_hosts), table_variables, accumulator_variables):
[all …]
Dtensor_tracer_report.py130 self.num_hosts = None
247 report.config.num_hosts = tt_config.num_hosts
334 self._write_report('%s %s\n'%(_FIELD_NAME_NUM_HOSTS, tt_config.num_hosts))
Dtpu_system_metadata.py134 num_hosts=len(device_dict),
142 logging.info('*** Num TPU Workers: %d', metadata.num_hosts)
Dtensor_tracer.py1608 num_hosts=None): argument
1640 self._tt_config.num_hosts = num_hosts
1644 if self._tt_config.num_hosts is None:
1645 self._tt_config.num_hosts = (
1690 self._tt_config.num_hosts = 1
Dtensor_tracer.proto37 int32 num_hosts = 5; field
/external/bcc/examples/networking/distributed_bridge/
Dmain.py32 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):
Dtunnel_mesh.py17 num_hosts = int(argv[1]) variable
65 for i in range(0, num_hosts):
/external/bcc/examples/networking/tunnel_monitor/
Dmain.py19 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/
Dtpu_embedding_configuration.proto40 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/
Dtpu_strategy.py153 "num_workers").set(self.extended.num_hosts)
196 "num_workers").set(self.extended.num_hosts)
661 def num_hosts(self): member in TPUExtended
663 return self._tpu_metadata.num_hosts
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v2_utils.py354 and strategy.extended.num_hosts > 1)