Searched refs:_get_device_dict_and_cores (Results 1 – 2 of 2) sorted by relevance
117 def _get_device_dict_and_cores(devices): member in TPUClusterResolver377 device_details = TPUClusterResolver._get_device_dict_and_cores(
566 device_details = resolver.TPUClusterResolver._get_device_dict_and_cores(591 resolver.TPUClusterResolver._get_device_dict_and_cores(device_list)