Searched refs:tpu_map (Results 1 – 1 of 1) sorted by relevance
40 def __init__(self, name, tpu_map): argument42 self._tpu_map = tpu_map53 def __init__(self, tpu_map): argument54 self._tpu_map = tpu_map118 def mock_service_client(self, tpu_map=None): argument120 if tpu_map is None:121 tpu_map = {}124 mock_locations.nodes.return_value = MockNodeClass(tpu_map)157 tpu_map = {170 service=self.mock_service_client(tpu_map=tpu_map),[all …]