Searched refs:TPUAssignment (Results 1 – 1 of 1) sorted by relevance
245 return TPUAssignment(worker_name=self._worker_name, num_cores=num_cores)248 class TPUAssignment(object): class