Searched refs:num_tpus_per_host (Results 1 – 3 of 3) sorted by relevance
152 int* num_tpus_per_host, std::vector<std::vector<Device*>>* tpu_devices) { in GetTPUDevices() argument166 *num_tpus_per_host = 0; in GetTPUDevices()186 *num_tpus_per_host = host_tpu_devices.size(); in GetTPUDevices()187 } else if (*num_tpus_per_host != host_tpu_devices.size()) { in GetTPUDevices()191 "Mismatched number of TPU devices in cluster ", *num_tpus_per_host, in GetTPUDevices()
63 int* num_tpus_per_host,
137 int num_tpus_per_host = 0; in GetTPUDevices() local141 num_tpus_per_host = host_tpu_devices.size(); in GetTPUDevices()150 if (num_tpus_per_host != host_tpu_devices_size) in GetTPUDevices()153 num_tpus_per_host, ", got ", host_tpu_devices.size()); in GetTPUDevices()