| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | step_intersection_test.cc | 58 PerHostStepDb CreateTestSteps(uint32 num_hosts, uint64 shift_ps) { in CreateTestSteps() argument
61 for (uint32 host_id = 0; host_id < num_hosts; host_id++) { in CreateTestSteps()
75 PerHostStepDb CreateNoStep(uint32 num_hosts) { in CreateNoStep() argument
77 for (uint32 host_id = 0; host_id < num_hosts; host_id++) { in CreateNoStep()
96 uint32 num_hosts = 4; in TEST() local
99 PerHostStepDb perhost_stepdb = CreateTestSteps(num_hosts, shift_ps); in TEST()
103 uint32 dst_num_steps = kNumStepsPerHost - num_hosts + 1; in TEST()
107 EXPECT_EQ(src_first_step_index, num_hosts - 1); in TEST()
115 uint32 num_hosts = 4; in TEST() local
118 PerHostStepDb perhost_stepdb = CreateTestSteps(num_hosts, shift_ps); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 1307 if device_config.num_cores % device_config.num_hosts:
1310 device_config.num_hosts))
1311 self._num_hosts = device_config.num_hosts
1326 self._num_hosts = tpu_system_metadata.num_hosts
1505 config_proto.num_hosts = self._num_hosts
1582 num_hosts=self._num_hosts,
2016 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
2030 def create_variables_and_ops(self, table, slot_variable_names, num_hosts, argument
2036 num_hosts=num_hosts,
2052 range(num_hosts), table_variables, accumulator_variables):
[all …]
|
| D | tpu_system_metadata.py | 65 def __new__(cls, num_cores, num_hosts, num_of_cores_per_host, topology, argument
68 cls).__new__(cls, num_cores, num_hosts, num_of_cores_per_host,
153 num_hosts=len(device_dict),
161 logging.info('*** Num TPU Workers: %d', metadata.num_hosts)
|
| D | tensor_tracer_report.py | 144 self.num_hosts = None
267 report.config.num_hosts = tt_config.num_hosts
356 self._write_report('%s %s\n'%(_FIELD_NAME_NUM_HOSTS, tt_config.num_hosts))
|
| D | tpu_embedding_v2.py | 484 self._strategy.extended.num_hosts)
508 config_proto.num_hosts = self._strategy.extended.num_hosts
1659 assert len(list_of_hosts) == strategy.extended.num_hosts
1715 num_hosts = len(hosts)
1720 partial_partition = rows % num_hosts
1721 full_rows_per_host = rows // num_hosts
1728 + [full_rows_per_host] * (num_hosts - partial_partition))
|
| D | tpu_embedding_v2_utils_test.py | 102 config.num_hosts = 2
|
| D | tensor_tracer.py | 1943 num_hosts=None): argument
1982 self._tt_config.num_hosts = num_hosts
1986 if self._tt_config.num_hosts is None:
1987 self._tt_config.num_hosts = (
2039 self._tt_config.num_hosts = 1
|
| D | tensor_tracer.proto | 41 int32 num_hosts = 5; field
|
| /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/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/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt | 15 name: "num_hosts"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.-t-p-u-system-metadata.pbtxt | 15 name: "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/core/tpu/kernels/ |
| D | tpu_configuration_ops.cc | 150 size_t num_hosts = ctx->num_inputs(); in Compute() local
214 params.num_hosts = num_hosts; in Compute()
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | xplane_to_memory_profile_test.cc | 91 EXPECT_EQ(memory_profile.num_hosts(), 1); in TEST()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 338 "num_workers").set(self.extended.num_hosts)
644 "num_workers").set(self.extended.num_hosts)
706 "num_workers").set(self.extended.num_hosts)
1336 def num_hosts(self): member in TPUExtended
1338 return self._tpu_metadata.num_hosts
|
| D | tpu_strategy_test.py | 1226 self.assertEqual(strategy.extended.num_hosts, 1)
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | memory_profile.proto | 121 int32 num_hosts = 2; field
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 227 size_t num_hosts; member
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training.py | 2777 strategy.extended.num_hosts > 1)
|