Searched defs:num_tpus_per_host (Results 1 – 2 of 2) sorted by relevance
152 int* num_tpus_per_host, std::vector<std::vector<Device*>>* tpu_devices) { in GetTPUDevices()
137 int num_tpus_per_host = 0; in GetTPUDevices() local