Searched refs:TpusPerHost (Results 1 – 3 of 3) sorted by relevance
135 static Status TpusPerHost(int* tpus);
153 Status TpuPlatform::TpusPerHost(int* tpus) { in TpusPerHost() function in tensorflow::tpu::TpuPlatform
408 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in ListPhysicalDevices()423 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in CreateDevices()