Searched refs:_get_device_assignment (Results 1 – 1 of 1) sorted by relevance
279 def _get_device_assignment(self): member in _InternalTPUContext339 return (self._get_device_assignment()